Test: policy model

This commit is contained in:
HFO4
2019-11-14 14:45:06 +08:00
parent c867916c3a
commit ade2bf1812
4 changed files with 49 additions and 5 deletions

View File

@@ -21,6 +21,10 @@ func TestGetUserByID(t *testing.T) {
AddRow(1, "管理员", "[1]")
mock.ExpectQuery("^SELECT (.+)").WillReturnRows(groupRows)
policyRows := sqlmock.NewRows([]string{"id", "name"}).
AddRow(1, "默认上传策略")
mock.ExpectQuery("^SELECT (.+)").WillReturnRows(policyRows)
user, err := GetUserByID(1)
asserts.NoError(err)
asserts.Equal(User{