2
0
mirror of https://github.com/acepanel/panel.git synced 2026-02-04 17:17:13 +08:00

fix: tests

This commit is contained in:
耗子
2023-07-07 19:52:17 +08:00
parent f0c08d36dd
commit 88061496d8
2 changed files with 7 additions and 7 deletions

View File

@@ -25,12 +25,12 @@ func (s *SettingTestSuite) SetupTest() {
func (s *SettingTestSuite) TestGet() {
mockOrm, mockDb, _, _ := mock.Orm()
mockOrm.On("Query").Return(mockDb).Twice()
mockDb.On("Where", "key", "test").Return(mockDb).Twice()
mockDb.On("FirstOrFail", &models.Setting{}).Return(nil).Once()
mockOrm.On("Query").Return(mockDb)
mockDb.On("Where", "key", "test").Return(mockDb)
mockDb.On("FirstOrFail", &models.Setting{}).Return(nil)
a := s.setting.Get("test")
s.Equal("", a)
mockDb.On("FirstOrFail", &models.Setting{}).Return(nil).Once()
mockDb.On("FirstOrFail", &models.Setting{}).Return(nil)
b := s.setting.Get("test", "test")
s.Equal("test", b)
mockOrm.AssertExpectations(s.T())
@@ -40,8 +40,8 @@ func (s *SettingTestSuite) TestGet() {
func (s *SettingTestSuite) TestSet() {
mockOrm, mockDb, _, _ := mock.Orm()
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()
mockDb.On("Where", "key", "test").Return(mockDb)
mockDb.On("UpdateOrCreate", &models.Setting{}, models.Setting{Key: "test"}, models.Setting{Value: "test"}).Return(nil)
err := s.setting.Set("test", "test")
s.Nil(err)
mockOrm.AssertExpectations(s.T())

View File

@@ -31,7 +31,7 @@ func (s *UserTestSuite) TestCreate() {
mockDb.On("Create", &models.User{
Username: "haozi",
Password: "123456",
}).Return(nil).Once()
}).Return(nil)
user, err := s.user.Create("haozi", "123456")
s.Nil(err)
s.Equal("haozi", user.Username)