package database

import (
	"testing"

	"github.com/gotify/server/model"
	"github.com/gotify/server/test"
	"github.com/jinzhu/gorm"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/suite"
)

func TestMigration(t *testing.T) {
	suite.Run(t, &MigrationSuite{})
}

type MigrationSuite struct {
	suite.Suite
	tmpDir test.TmpDir
}

func (s *MigrationSuite) BeforeTest(suiteName, testName string) {
	s.tmpDir = test.NewTmpDir("gotify_migrationsuite")
	db, err := gorm.Open("sqlite3", s.tmpDir.Path("test_obsolete.db"))
	assert.Nil(s.T(), err)
	defer db.Close()

	assert.Nil(s.T(), db.CreateTable(new(model.User)).Error)
	assert.Nil(s.T(), db.Create(&model.User{
		Name:  "test_user",
		Admin: true,
	}).Error)

	// we should not be able to create applications by now
	assert.False(s.T(), db.HasTable(new(model.Application)))
}

func (s *MigrationSuite) AfterTest(suiteName, testName string) {
	assert.Nil(s.T(), s.tmpDir.Clean())
}

func (s *MigrationSuite) TestMigration() {
	db, err := New("sqlite3", s.tmpDir.Path("test_obsolete.db"), "admin", "admin", 6, true)
	assert.Nil(s.T(), err)
	defer db.Close()

	assert.True(s.T(), db.DB.HasTable(new(model.Application)))

	// a user already exist, not adding a new user
	assert.Nil(s.T(), db.GetUserByName("admin"))

	// the old user should persist
	assert.Equal(s.T(), true, db.GetUserByName("test_user").Admin)

	// we should be able to create applications
	assert.Nil(s.T(), db.CreateApplication(&model.Application{
		Token:       "A1234",
		UserID:      db.GetUserByName("test_user").ID,
		Description: "this is a test application",
		Name:        "test application",
	}))
	assert.Equal(s.T(), "test application", db.GetApplicationByToken("A1234").Name)
}