fix server test

AnalogJ · Feb 4, 2024 · 09f4b34 · 09f4b34
1 parent f24abf2
commit 09f4b34
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/webapp/backend/pkg/web/server_test.go b/webapp/backend/pkg/web/server_test.go
@@ -4,6 +4,16 @@ import (
  "bytes"
  "encoding/json"
  "fmt"
+ "io"
+ "io/ioutil"
+ "net/http"
+ "net/http/httptest"
+ "os"
+ "path"
+ "strings"
+ "testing"
+ "time"
+
  "github.com/analogj/scrutiny/webapp/backend/pkg"
  "github.com/analogj/scrutiny/webapp/backend/pkg/config"
  mock_config "github.com/analogj/scrutiny/webapp/backend/pkg/config/mock"
@@ -14,15 +24,6 @@ import (
  "github.com/sirupsen/logrus"
  "github.com/stretchr/testify/require"
  "github.com/stretchr/testify/suite"
- "io"
- "io/ioutil"
- "net/http"
- "net/http/httptest"
- "os"
- "path"
- "strings"
- "testing"
- "time"
 )
 
 /*
@@ -189,6 +190,7 @@ func (suite *ServerTestSuite) TestUploadDeviceMetricsRoute() {
  fakeConfig.EXPECT().GetString("web.influxdb.token").Return("my-super-secret-auth-token").AnyTimes()
  fakeConfig.EXPECT().GetString("web.influxdb.org").Return("scrutiny").AnyTimes()
  fakeConfig.EXPECT().GetString("web.influxdb.bucket").Return("metrics").AnyTimes()
+ fakeConfig.EXPECT().GetBool("user.metrics.repeat_notifications").Return(true).AnyTimes()
  fakeConfig.EXPECT().GetBool("web.influxdb.tls.insecure_skip_verify").Return(false).AnyTimes()
  fakeConfig.EXPECT().GetBool("web.influxdb.retention_policy").Return(false).AnyTimes()
  if _, isGithubActions := os.LookupEnv("GITHUB_ACTIONS"); isGithubActions {
@@ -247,6 +249,7 @@ func (suite *ServerTestSuite) TestPopulateMultiple() {
  fakeConfig.EXPECT().GetString("web.influxdb.token").Return("my-super-secret-auth-token").AnyTimes()
  fakeConfig.EXPECT().GetString("web.influxdb.org").Return("scrutiny").AnyTimes()
  fakeConfig.EXPECT().GetString("web.influxdb.bucket").Return("metrics").AnyTimes()
+ fakeConfig.EXPECT().GetBool("user.metrics.repeat_notifications").Return(true).AnyTimes()
  fakeConfig.EXPECT().GetBool("web.influxdb.tls.insecure_skip_verify").Return(false).AnyTimes()
  fakeConfig.EXPECT().GetBool("web.influxdb.retention_policy").Return(false).AnyTimes()
  if _, isGithubActions := os.LookupEnv("GITHUB_ACTIONS"); isGithubActions {
@@ -529,6 +532,7 @@ func (suite *ServerTestSuite) TestGetDevicesSummaryRoute_Nvme() {
  fakeConfig.EXPECT().GetString("web.influxdb.token").Return("my-super-secret-auth-token").AnyTimes()
  fakeConfig.EXPECT().GetString("web.influxdb.org").Return("scrutiny").AnyTimes()
  fakeConfig.EXPECT().GetString("web.influxdb.bucket").Return("metrics").AnyTimes()
+ fakeConfig.EXPECT().GetBool("user.metrics.repeat_notifications").Return(true).AnyTimes()
  fakeConfig.EXPECT().GetBool("web.influxdb.tls.insecure_skip_verify").Return(false).AnyTimes()
  fakeConfig.EXPECT().GetBool("web.influxdb.retention_policy").Return(false).AnyTimes()
  fakeConfig.EXPECT().GetStringSlice("notify.urls").AnyTimes().Return([]string{})