diff --git a/app/services/setting_test.go b/app/services/setting_test.go index 6709a41b..9d03b8d1 100644 --- a/app/services/setting_test.go +++ b/app/services/setting_test.go @@ -33,16 +33,12 @@ func (s *SettingTestSuite) TestGet() { b := s.setting.Get("test", "test") s.Equal("", a) s.Equal("test", b) - mockOrm.AssertExpectations(s.T()) - mockDb.AssertExpectations(s.T()) } func (s *SettingTestSuite) TestSet() { - /*mockOrm, mockDb, _, _ := mock.Orm() + mockOrm, mockDb, _, _ := mock.Orm() mockOrm.On("Query").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") s.Nil(err) - mockOrm.AssertExpectations(s.T()) - mockDb.AssertExpectations(s.T())*/ } diff --git a/app/services/user_test.go b/app/services/user_test.go index 8acf85e7..410a4ec6 100644 --- a/app/services/user_test.go +++ b/app/services/user_test.go @@ -35,6 +35,4 @@ func (s *UserTestSuite) TestCreate() { user, err := s.user.Create("haozi", "123456") s.Nil(err) s.Equal("haozi", user.Username) - mockOrm.AssertExpectations(s.T()) - mockDb.AssertExpectations(s.T()) }