Infrastructure/Cloudreve
0
Fork 0
Code Issues Pull requests Activity

Test: newly added sb models

Browse source
This commit is contained in:
HFO4 2021-11-09 19:29:56 +08:00
parent 3064ed60f3
commit eeee43d569
5 changed files with 31 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
models/node_test.go
View file

@ -48,7 +48,7 @@ func TestNode_BeforeSave(t *testing.T) {
node.Aria2OptionsSerialized.Timeout = 1 node.Aria2OptionsSerialized.Timeout = 1
a.NoError(node.BeforeSave()) a.NoError(node.BeforeSave())
a.Contains("1", node.Aria2Options) a.Contains(node.Aria2Options, "1")
} }
func TestNode_SetStatus(t *testing.T) { func TestNode_SetStatus(t *testing.T) {

15
models/policy_test.go
View file

@ -327,3 +327,18 @@ func TestPolicy_IsThumbExist(t *testing.T) {
asserts.Equal(testCase.expect, policy.IsThumbExist(testCase.name)) asserts.Equal(testCase.expect, policy.IsThumbExist(testCase.name))
} }
} }
func TestPolicy_UpdateAccessKeyAndClearCache(t *testing.T) {
a := assert.New(t)
cache.Set("policy_1331", Policy{}, 3600)
p := &Policy{}
p.ID = 1331
mock.ExpectBegin()
mock.ExpectExec("UPDATE(.+)").WithArgs("ak", sqlmock.AnyArg()).WillReturnResult(sqlmock.NewResult(1, 1))
mock.ExpectCommit()
a.NoError(p.UpdateAccessKeyAndClearCache("ak"))
a.NoError(mock.ExpectationsWereMet())
_, ok := cache.Get("policy_1331")
a.False(ok)
}

6
models/tag_test.go
View file

@ -55,9 +55,9 @@ func TestGetTagsByUID(t *testing.T) {
func TestGetTagsByID(t *testing.T) { func TestGetTagsByID(t *testing.T) {
asserts := assert.New(t) asserts := assert.New(t)
mock.ExpectQuery("SELECT(.+)").WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(1)) mock.ExpectQuery("SELECT(.+)").WillReturnRows(sqlmock.NewRows([]string{"name"}).AddRow("tag"))
res, err := GetTasksByID(1) res, err := GetTagsByUID(1)
asserts.NoError(mock.ExpectationsWereMet()) asserts.NoError(mock.ExpectationsWereMet())
asserts.NoError(err) asserts.NoError(err)
asserts.EqualValues(1, res.ID) asserts.EqualValues("tag", res[0].Name)
} }

11
models/task_test.go
View file

@ -91,3 +91,14 @@ func TestListTasks(t *testing.T) {
asserts.EqualValues(5, total) asserts.EqualValues(5, total)
asserts.Len(res, 1) asserts.Len(res, 1)
} }
func TestGetTasksByStatus(t *testing.T) {
a := assert.New(t)
mock.ExpectQuery("SELECT(.+)").
WithArgs(1, 2).
WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(1))
res := GetTasksByStatus(1, 2)
a.NoError(mock.ExpectationsWereMet())
a.Len(res, 1)
}

2
models/webdav_test.go
View file

@ -55,6 +55,6 @@ func TestDeleteWebDAVAccountByID(t *testing.T) {
mock.ExpectBegin() mock.ExpectBegin()
mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1, 1)) mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1, 1))
mock.ExpectCommit() mock.ExpectCommit()
asserts.NoError(DeleteTagByID(1, 1)) DeleteWebDAVAccountByID(1, 1)
asserts.NoError(mock.ExpectationsWereMet()) asserts.NoError(mock.ExpectationsWereMet())
} }