You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
329 lines
11 KiB
329 lines
11 KiB
package web_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
mock_config "github.com/analogj/scrutiny/webapp/backend/pkg/config/mock"
|
|
dbModels "github.com/analogj/scrutiny/webapp/backend/pkg/models/db"
|
|
"github.com/analogj/scrutiny/webapp/backend/pkg/web"
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/stretchr/testify/require"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestHealthRoute(t *testing.T) {
|
|
//setup
|
|
parentPath, _ := ioutil.TempDir("", "")
|
|
defer os.RemoveAll(parentPath)
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
fakeConfig := mock_config.NewMockInterface(mockCtrl)
|
|
fakeConfig.EXPECT().GetString("web.database.location").Return(path.Join(parentPath, "scrutiny_test.db")).AnyTimes()
|
|
fakeConfig.EXPECT().GetString("web.src.frontend.path").Return(parentPath).AnyTimes()
|
|
|
|
ae := web.AppEngine{
|
|
Config: fakeConfig,
|
|
}
|
|
|
|
router := ae.Setup(logrus.New())
|
|
|
|
//test
|
|
w := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("GET", "/api/health", nil)
|
|
router.ServeHTTP(w, req)
|
|
|
|
//assert
|
|
require.Equal(t, 200, w.Code)
|
|
require.Equal(t, "{\"success\":true}", w.Body.String())
|
|
}
|
|
|
|
func TestRegisterDevicesRoute(t *testing.T) {
|
|
//setup
|
|
parentPath, _ := ioutil.TempDir("", "")
|
|
defer os.RemoveAll(parentPath)
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
fakeConfig := mock_config.NewMockInterface(mockCtrl)
|
|
fakeConfig.EXPECT().GetString("web.database.location").Return(path.Join(parentPath, "scrutiny_test.db")).AnyTimes()
|
|
fakeConfig.EXPECT().GetString("web.src.frontend.path").Return(parentPath).AnyTimes()
|
|
ae := web.AppEngine{
|
|
Config: fakeConfig,
|
|
}
|
|
router := ae.Setup(logrus.New())
|
|
file, err := os.Open("testdata/register-devices-req.json")
|
|
require.NoError(t, err)
|
|
|
|
//test
|
|
w := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("POST", "/api/devices/register", file)
|
|
router.ServeHTTP(w, req)
|
|
|
|
//assert
|
|
require.Equal(t, 200, w.Code)
|
|
}
|
|
|
|
func TestUploadDeviceMetricsRoute(t *testing.T) {
|
|
//setup
|
|
parentPath, _ := ioutil.TempDir("", "")
|
|
defer os.RemoveAll(parentPath)
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
fakeConfig := mock_config.NewMockInterface(mockCtrl)
|
|
fakeConfig.EXPECT().GetString("web.database.location").AnyTimes().Return(path.Join(parentPath, "scrutiny_test.db"))
|
|
fakeConfig.EXPECT().GetString("web.src.frontend.path").AnyTimes().Return(parentPath)
|
|
ae := web.AppEngine{
|
|
Config: fakeConfig,
|
|
}
|
|
router := ae.Setup(logrus.New())
|
|
devicesfile, err := os.Open("testdata/register-devices-single-req.json")
|
|
require.NoError(t, err)
|
|
|
|
metricsfile, err := os.Open("testdata/upload-device-metrics-req.json")
|
|
require.NoError(t, err)
|
|
|
|
//test
|
|
wr := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("POST", "/api/devices/register", devicesfile)
|
|
router.ServeHTTP(wr, req)
|
|
require.Equal(t, 200, wr.Code)
|
|
|
|
mr := httptest.NewRecorder()
|
|
req, _ = http.NewRequest("POST", "/api/device/0x5000cca264eb01d7/smart", metricsfile)
|
|
router.ServeHTTP(mr, req)
|
|
require.Equal(t, 200, mr.Code)
|
|
|
|
//assert
|
|
}
|
|
|
|
func TestPopulateMultiple(t *testing.T) {
|
|
//setup
|
|
parentPath, _ := ioutil.TempDir("", "")
|
|
defer os.RemoveAll(parentPath)
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
fakeConfig := mock_config.NewMockInterface(mockCtrl)
|
|
//fakeConfig.EXPECT().GetString("web.database.location").AnyTimes().Return("testdata/scrutiny_test.db")
|
|
fakeConfig.EXPECT().GetStringSlice("notify.urls").Return([]string{}).AnyTimes()
|
|
fakeConfig.EXPECT().GetString("web.database.location").AnyTimes().Return(path.Join(parentPath, "scrutiny_test.db"))
|
|
fakeConfig.EXPECT().GetString("web.src.frontend.path").AnyTimes().Return(parentPath)
|
|
ae := web.AppEngine{
|
|
Config: fakeConfig,
|
|
}
|
|
router := ae.Setup(logrus.New())
|
|
devicesfile, err := os.Open("testdata/register-devices-req.json")
|
|
require.NoError(t, err)
|
|
|
|
metricsfile, err := os.Open("../models/testdata/smart-ata.json")
|
|
require.NoError(t, err)
|
|
failfile, err := os.Open("../models/testdata/smart-fail2.json")
|
|
require.NoError(t, err)
|
|
nvmefile, err := os.Open("../models/testdata/smart-nvme.json")
|
|
require.NoError(t, err)
|
|
scsifile, err := os.Open("../models/testdata/smart-scsi.json")
|
|
require.NoError(t, err)
|
|
scsi2file, err := os.Open("../models/testdata/smart-scsi2.json")
|
|
require.NoError(t, err)
|
|
|
|
//test
|
|
wr := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("POST", "/api/devices/register", devicesfile)
|
|
router.ServeHTTP(wr, req)
|
|
require.Equal(t, 200, wr.Code)
|
|
|
|
mr := httptest.NewRecorder()
|
|
req, _ = http.NewRequest("POST", "/api/device/0x5000cca264eb01d7/smart", metricsfile)
|
|
router.ServeHTTP(mr, req)
|
|
require.Equal(t, 200, mr.Code)
|
|
|
|
fr := httptest.NewRecorder()
|
|
req, _ = http.NewRequest("POST", "/api/device/0x5000cca264ec3183/smart", failfile)
|
|
router.ServeHTTP(fr, req)
|
|
require.Equal(t, 200, fr.Code)
|
|
|
|
nr := httptest.NewRecorder()
|
|
req, _ = http.NewRequest("POST", "/api/device/0x5002538e40a22954/smart", nvmefile)
|
|
router.ServeHTTP(nr, req)
|
|
require.Equal(t, 200, nr.Code)
|
|
|
|
sr := httptest.NewRecorder()
|
|
req, _ = http.NewRequest("POST", "/api/device/0x5000cca252c859cc/smart", scsifile)
|
|
router.ServeHTTP(sr, req)
|
|
require.Equal(t, 200, sr.Code)
|
|
|
|
s2r := httptest.NewRecorder()
|
|
req, _ = http.NewRequest("POST", "/api/device/0x5000cca264ebc248/smart", scsi2file)
|
|
router.ServeHTTP(s2r, req)
|
|
require.Equal(t, 200, s2r.Code)
|
|
|
|
//assert
|
|
}
|
|
|
|
//TODO: this test should use a recorded request/response playback.
|
|
//func TestSendTestNotificationRoute(t *testing.T) {
|
|
// //setup
|
|
// parentPath, _ := ioutil.TempDir("", "")
|
|
// defer os.RemoveAll(parentPath)
|
|
// mockCtrl := gomock.NewController(t)
|
|
// defer mockCtrl.Finish()
|
|
// fakeConfig := mock_config.NewMockInterface(mockCtrl)
|
|
// fakeConfig.EXPECT().GetString("web.database.location").AnyTimes().Return(path.Join(parentPath, "scrutiny_test.db"))
|
|
// fakeConfig.EXPECT().GetString("web.src.frontend.path").AnyTimes().Return(parentPath)
|
|
// fakeConfig.EXPECT().GetStringSlice("notify.urls").AnyTimes().Return([]string{"https://scrutiny.requestcatcher.com/test"})
|
|
// ae := web.AppEngine{
|
|
// Config: fakeConfig,
|
|
// }
|
|
// router := ae.Setup(logrus.New())
|
|
//
|
|
// //test
|
|
// wr := httptest.NewRecorder()
|
|
// req, _ := http.NewRequest("POST", "/api/health/notify", strings.NewReader("{}"))
|
|
// router.ServeHTTP(wr, req)
|
|
//
|
|
// //assert
|
|
// require.Equal(t, 200, wr.Code)
|
|
//}
|
|
|
|
func TestSendTestNotificationRoute_WebhookFailure(t *testing.T) {
|
|
//setup
|
|
parentPath, _ := ioutil.TempDir("", "")
|
|
defer os.RemoveAll(parentPath)
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
fakeConfig := mock_config.NewMockInterface(mockCtrl)
|
|
fakeConfig.EXPECT().GetString("web.database.location").AnyTimes().Return(path.Join(parentPath, "scrutiny_test.db"))
|
|
fakeConfig.EXPECT().GetString("web.src.frontend.path").AnyTimes().Return(parentPath)
|
|
fakeConfig.EXPECT().GetStringSlice("notify.urls").AnyTimes().Return([]string{"https://unroutable.domain.example.asdfghj"})
|
|
ae := web.AppEngine{
|
|
Config: fakeConfig,
|
|
}
|
|
router := ae.Setup(logrus.New())
|
|
|
|
//test
|
|
wr := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("POST", "/api/health/notify", strings.NewReader("{}"))
|
|
router.ServeHTTP(wr, req)
|
|
|
|
//assert
|
|
require.Equal(t, 500, wr.Code)
|
|
}
|
|
|
|
func TestSendTestNotificationRoute_ScriptFailure(t *testing.T) {
|
|
//setup
|
|
parentPath, _ := ioutil.TempDir("", "")
|
|
defer os.RemoveAll(parentPath)
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
fakeConfig := mock_config.NewMockInterface(mockCtrl)
|
|
fakeConfig.EXPECT().GetString("web.database.location").AnyTimes().Return(path.Join(parentPath, "scrutiny_test.db"))
|
|
fakeConfig.EXPECT().GetString("web.src.frontend.path").AnyTimes().Return(parentPath)
|
|
fakeConfig.EXPECT().GetStringSlice("notify.urls").AnyTimes().Return([]string{"script:///missing/path/on/disk"})
|
|
ae := web.AppEngine{
|
|
Config: fakeConfig,
|
|
}
|
|
router := ae.Setup(logrus.New())
|
|
|
|
//test
|
|
wr := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("POST", "/api/health/notify", strings.NewReader("{}"))
|
|
router.ServeHTTP(wr, req)
|
|
|
|
//assert
|
|
require.Equal(t, 500, wr.Code)
|
|
}
|
|
|
|
func TestSendTestNotificationRoute_ScriptSuccess(t *testing.T) {
|
|
//setup
|
|
parentPath, _ := ioutil.TempDir("", "")
|
|
defer os.RemoveAll(parentPath)
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
fakeConfig := mock_config.NewMockInterface(mockCtrl)
|
|
fakeConfig.EXPECT().GetString("web.database.location").AnyTimes().Return(path.Join(parentPath, "scrutiny_test.db"))
|
|
fakeConfig.EXPECT().GetString("web.src.frontend.path").AnyTimes().Return(parentPath)
|
|
fakeConfig.EXPECT().GetStringSlice("notify.urls").AnyTimes().Return([]string{"script:///usr/bin/env"})
|
|
ae := web.AppEngine{
|
|
Config: fakeConfig,
|
|
}
|
|
router := ae.Setup(logrus.New())
|
|
|
|
//test
|
|
wr := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("POST", "/api/health/notify", strings.NewReader("{}"))
|
|
router.ServeHTTP(wr, req)
|
|
|
|
//assert
|
|
require.Equal(t, 200, wr.Code)
|
|
}
|
|
|
|
func TestSendTestNotificationRoute_ShoutrrrFailure(t *testing.T) {
|
|
//setup
|
|
parentPath, _ := ioutil.TempDir("", "")
|
|
defer os.RemoveAll(parentPath)
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
fakeConfig := mock_config.NewMockInterface(mockCtrl)
|
|
fakeConfig.EXPECT().GetString("web.database.location").AnyTimes().Return(path.Join(parentPath, "scrutiny_test.db"))
|
|
fakeConfig.EXPECT().GetString("web.src.frontend.path").AnyTimes().Return(parentPath)
|
|
fakeConfig.EXPECT().GetStringSlice("notify.urls").AnyTimes().Return([]string{"discord://invalidtoken@channel"})
|
|
ae := web.AppEngine{
|
|
Config: fakeConfig,
|
|
}
|
|
router := ae.Setup(logrus.New())
|
|
|
|
//test
|
|
wr := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("POST", "/api/health/notify", strings.NewReader("{}"))
|
|
router.ServeHTTP(wr, req)
|
|
|
|
//assert
|
|
require.Equal(t, 500, wr.Code)
|
|
}
|
|
|
|
func TestGetDevicesSummaryRoute_Nvme(t *testing.T) {
|
|
//setup
|
|
parentPath, _ := ioutil.TempDir("", "")
|
|
defer os.RemoveAll(parentPath)
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
fakeConfig := mock_config.NewMockInterface(mockCtrl)
|
|
fakeConfig.EXPECT().GetString("web.database.location").AnyTimes().Return(path.Join(parentPath, "scrutiny_test.db"))
|
|
fakeConfig.EXPECT().GetString("web.src.frontend.path").AnyTimes().Return(parentPath)
|
|
ae := web.AppEngine{
|
|
Config: fakeConfig,
|
|
}
|
|
router := ae.Setup(logrus.New())
|
|
devicesfile, err := os.Open("testdata/register-devices-req-2.json")
|
|
require.NoError(t, err)
|
|
|
|
metricsfile, err := os.Open("../models/testdata/smart-nvme2.json")
|
|
require.NoError(t, err)
|
|
|
|
//test
|
|
wr := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("POST", "/api/devices/register", devicesfile)
|
|
router.ServeHTTP(wr, req)
|
|
require.Equal(t, 200, wr.Code)
|
|
|
|
mr := httptest.NewRecorder()
|
|
req, _ = http.NewRequest("POST", "/api/device/a4c8e8ed-11a0-4c97-9bba-306440f1b944/smart", metricsfile)
|
|
router.ServeHTTP(mr, req)
|
|
require.Equal(t, 200, mr.Code)
|
|
|
|
sr := httptest.NewRecorder()
|
|
req, _ = http.NewRequest("GET", "/api/summary", nil)
|
|
router.ServeHTTP(sr, req)
|
|
require.Equal(t, 200, sr.Code)
|
|
var device dbModels.DeviceWrapper
|
|
json.Unmarshal(sr.Body.Bytes(), &device)
|
|
|
|
//assert
|
|
require.Equal(t, "a4c8e8ed-11a0-4c97-9bba-306440f1b944", device.Data[0].WWN)
|
|
require.Equal(t, "passed", device.Data[0].SmartResults[0].SmartStatus)
|
|
}
|