mirror of
https://github.com/gotify/server.git
synced 2026-05-07 05:48:41 +08:00
c912bb8cba
sqlite3 doesn't support adding a foreign key via gorm.DB#AddForeignKey this would mean, that we have some hacky solutions for having foreign keys for sqlite and the other databases. Therefore manually deleting the referencing models seems to be the best solution. We already have interfaces for database capsuling, so no api must be adjusted.
63 lines
1.9 KiB
Go
63 lines
1.9 KiB
Go
package database
|
|
|
|
import (
|
|
"github.com/gotify/server/model"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func (s *DatabaseSuite) TestApplication() {
|
|
assert.Nil(s.T(), s.db.GetApplicationByToken("asdasdf"), "not existing app")
|
|
assert.Nil(s.T(), s.db.GetApplicationByID(uint(1)), "not existing app")
|
|
|
|
user := &model.User{Name: "test", Pass: []byte{1}}
|
|
s.db.CreateUser(user)
|
|
assert.NotEqual(s.T(), 0, user.ID)
|
|
|
|
apps := s.db.GetApplicationsByUser(user.ID)
|
|
assert.Empty(s.T(), apps)
|
|
|
|
app := &model.Application{UserID: user.ID, Token: "C0000000000", Name: "backupserver"}
|
|
s.db.CreateApplication(app)
|
|
|
|
apps = s.db.GetApplicationsByUser(user.ID)
|
|
assert.Len(s.T(), apps, 1)
|
|
assert.Contains(s.T(), apps, app)
|
|
|
|
newApp := s.db.GetApplicationByToken(app.Token)
|
|
assert.Equal(s.T(), app, newApp)
|
|
|
|
newApp = s.db.GetApplicationByID(app.ID)
|
|
assert.Equal(s.T(), app, newApp)
|
|
|
|
newApp.Image = "asdasd"
|
|
s.db.UpdateApplication(newApp)
|
|
|
|
newApp = s.db.GetApplicationByID(app.ID)
|
|
assert.Equal(s.T(), "asdasd", newApp.Image)
|
|
|
|
s.db.DeleteApplicationByID(app.ID)
|
|
|
|
apps = s.db.GetApplicationsByUser(user.ID)
|
|
assert.Empty(s.T(), apps)
|
|
|
|
assert.Nil(s.T(), s.db.GetApplicationByID(app.ID))
|
|
}
|
|
|
|
func (s *DatabaseSuite) TestDeleteAppDeletesMessages() {
|
|
s.db.CreateApplication(&model.Application{ID: 55, Token: "token"})
|
|
s.db.CreateApplication(&model.Application{ID: 66, Token: "token2"})
|
|
s.db.CreateMessage(&model.Message{ID: 12, ApplicationID: 55})
|
|
s.db.CreateMessage(&model.Message{ID: 13, ApplicationID: 66})
|
|
s.db.CreateMessage(&model.Message{ID: 14, ApplicationID: 55})
|
|
s.db.CreateMessage(&model.Message{ID: 15, ApplicationID: 55})
|
|
|
|
s.db.DeleteApplicationByID(55)
|
|
|
|
assert.Nil(s.T(), s.db.GetMessageByID(12))
|
|
assert.NotNil(s.T(), s.db.GetMessageByID(13))
|
|
assert.Nil(s.T(), s.db.GetMessageByID(14))
|
|
assert.Nil(s.T(), s.db.GetMessageByID(15))
|
|
assert.Empty(s.T(), s.db.GetMessagesByApplication(55))
|
|
assert.NotEmpty(s.T(), s.db.GetMessagesByApplication(66))
|
|
}
|