From f0c08d36dd4cb01c32ef09485acc488374fd40c9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=80=97=E5=AD=90?= Date: Fri, 7 Jul 2023 19:48:25 +0800 Subject: [PATCH] fix: tests --- app/services/setting_test.go | 2 +- app/services/user_test.go | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/app/services/setting_test.go b/app/services/setting_test.go index 7ffe0e2f..495fbe46 100644 --- a/app/services/setting_test.go +++ b/app/services/setting_test.go @@ -39,7 +39,7 @@ func (s *SettingTestSuite) TestGet() { func (s *SettingTestSuite) TestSet() { mockOrm, mockDb, _, _ := mock.Orm() - mockOrm.On("Query").Return(mockDb).Twice() + mockOrm.On("Query").Return(mockDb) mockDb.On("Where", "key", "test").Return(mockDb).Once() mockDb.On("UpdateOrCreate", &models.Setting{}, models.Setting{Key: "test"}, models.Setting{Value: "test"}).Return(nil).Once() err := s.setting.Set("test", "test") diff --git a/app/services/user_test.go b/app/services/user_test.go index 8acf85e7..e9bdb08d 100644 --- a/app/services/user_test.go +++ b/app/services/user_test.go @@ -27,7 +27,7 @@ func (s *UserTestSuite) SetupTest() { func (s *UserTestSuite) TestCreate() { mockOrm, mockDb, _, _ := mock.Orm() - mockOrm.On("Query").Return(mockDb).Once() + mockOrm.On("Query").Return(mockDb) mockDb.On("Create", &models.User{ Username: "haozi", Password: "123456",