package sqlite

import (
	"io/ioutil"
	"os"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestOpen(t *testing.T) {
	dir, err := ioutil.TempDir("", "test-open")
	require.NoError(t, err)
	defer os.Remove(dir)

	dbPath := filepath.Join(dir, "db.sql")

	// Open the db for the first time.
	db, err := openAndMigrate(dbPath, "some-key", reducedKdfIterationsNumber)
	require.NoError(t, err)

	// Insert some data.
	_, err = db.Exec("CREATE TABLE test(name TEXT)")
	require.NoError(t, err)
	_, err = db.Exec(`INSERT INTO test (name) VALUES ("abc")`)
	require.NoError(t, err)
	db.Close()

	// Open again with different key should fail
	// because the file already exists and it should not
	// be recreated.
	_, err = openAndMigrate(dbPath, "different-key", reducedKdfIterationsNumber)
	require.Error(t, err)
}

// TestCommunitiesMigrationDirty tests the communities migration when
// dirty flag has been set to true.
// We first make it fail, then clean up so that it can be replayed, and
// then execute again, and we should be all migrated.
func TestCommunitiesMigrationDirty(t *testing.T) {
	// Open the db for the first time.
	db, err := open(InMemoryPath, "some-key", reducedKdfIterationsNumber)
	require.NoError(t, err)

	// Create a communities table, so that migration will fail
	_, err = db.Exec(`CREATE TABLE communities_communities (a varchar);`)
	require.NoError(t, err)

	// Migrate the database, this should fail
	err = Migrate(db)
	require.Error(t, err)

	// Version and dirty should be true and set to communities migration
	var version uint
	var dirty bool

	err = db.QueryRow(`SELECT version, dirty FROM `+migrationsTable).Scan(&version, &dirty)
	require.NoError(t, err)

	require.True(t, dirty)
	require.Equal(t, communitiesMigrationVersion, version)

	// Drop communities table and re-run migrations

	_, err = db.Exec(`DROP TABLE communities_communities`)

	require.NoError(t, err)

	// Migrate the database, this should work
	err = Migrate(db)
	require.NoError(t, err)

	// Make sure communities table is present

	var name string
	err = db.QueryRow(`SELECT name FROM sqlite_master WHERE type='table' AND name='communities_communities'`).Scan(&name)

	require.NoError(t, err)
	require.Equal(t, "communities_communities", name)

}

// TestCommunitiesMigrationNotDirty tests the communities migration when
// dirty flag has been set to false, and the communities migration has
// effectively been skipped.
// We first make it fail, then clean up so that it can be replayed, set
// dirty to false and then execute again, and we should be all migrated.
func TestCommunitiesMigrationNotDirty(t *testing.T) {
	// Open the db for the first time.
	db, err := open(InMemoryPath, "some-key", reducedKdfIterationsNumber)
	require.NoError(t, err)

	// Create a communities table, so that migration will fail
	_, err = db.Exec(`CREATE TABLE communities_communities (a varchar);`)
	require.NoError(t, err)

	// Migrate the database, this should fail
	err = Migrate(db)
	require.Error(t, err)

	// Set dirty to false
	_, err = db.Exec(`UPDATE ` + migrationsTable + ` SET dirty = 0`)
	require.NoError(t, err)

	// Version and dirty should be true and set to communities migration
	var version uint
	var dirty bool

	err = db.QueryRow(`SELECT version, dirty FROM `+migrationsTable).Scan(&version, &dirty)
	require.NoError(t, err)

	require.False(t, dirty)
	require.Equal(t, communitiesMigrationVersion, version)

	// Drop communities table and re-run migrations
	_, err = db.Exec(`DROP TABLE communities_communities`)

	require.NoError(t, err)

	// Migrate the database, this should work
	err = Migrate(db)
	require.NoError(t, err)

	// Make sure communities table is present

	var name string
	err = db.QueryRow(`SELECT name FROM sqlite_master WHERE type='table' AND name='communities_communities'`).Scan(&name)

	require.NoError(t, err)
	require.Equal(t, "communities_communities", name)

}