p90524618
/
answer
mirror of https://gitee.com/answerdev/answer.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix: repo test set notification userID to correct

This commit is contained in:
LinkinStar 2022-11-18 16:52:05 +08:00
parent 0c1a77d52c
commit 4075739fec
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/repo/repo_test/notification_repo_test.go
View File

@ -81,7 +81,7 @@ func Test_notificationRepo_GetNotificationPage(t *testing.T) {
err := notificationRepo.AddNotification(context.TODO(), ent) err := notificationRepo.AddNotification(context.TODO(), ent)
assert.NoError(t, err) assert.NoError(t, err)
notificationPage, total, err := notificationRepo.GetNotificationPage(context.TODO(), &schema.NotificationSearch{UserID: userID}) notificationPage, total, err := notificationRepo.GetNotificationPage(context.TODO(), &schema.NotificationSearch{UserID: ent.UserID})
assert.NoError(t, err) assert.NoError(t, err)
assert.True(t, total > 0) assert.True(t, total > 0)
assert.Equal(t, notificationPage[0].UserID, ent.UserID) assert.Equal(t, notificationPage[0].UserID, ent.UserID)

3
internal/repo/repo_test/repo_main_test.go
View File

@ -52,7 +52,8 @@ var (
func TestMain(t *testing.M) { func TestMain(t *testing.M) {
dbSetting, ok := dbSettingMapping[os.Getenv("TEST_DB_DRIVER")] dbSetting, ok := dbSettingMapping[os.Getenv("TEST_DB_DRIVER")]
if !ok { if !ok {
dbSetting = dbSettingMapping[string(schemas.MYSQL)] // Use sqlite3 to test.
dbSetting = dbSettingMapping[string(schemas.SQLITE)]
} }
defer func() { defer func() {
if tearDown != nil { if tearDown != nil {