package localnotifications

import (
	"database/sql"
	"io/ioutil"
	"os"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/status-im/status-go/appdatabase"
	"github.com/status-im/status-go/sqlite"
)

func setupAppTestDb(t *testing.T) (*sql.DB, func()) {
	tmpfile, err := ioutil.TempFile("", "local-notifications-tests-")
	require.NoError(t, err)
	db, err := appdatabase.InitializeDB(tmpfile.Name(), "local-notifications-tests", sqlite.ReducedKDFIterationsNumber)
	require.NoError(t, err)
	return db, func() {
		require.NoError(t, os.Remove(tmpfile.Name()))
	}
}

func setupTestDB(t *testing.T, db *sql.DB) (*Database, func()) {
	return NewDB(db, 1777), func() {
		require.NoError(t, db.Close())
	}
}

func TestWalletPreferences(t *testing.T) {
	appDB, appStop := setupAppTestDb(t)
	defer appStop()

	db, stop := setupTestDB(t, appDB)
	defer stop()

	enabled := true
	require.NoError(t, db.ChangeWalletPreference(enabled))
	rst, err := db.GetWalletPreference()
	require.NoError(t, err)
	require.Equal(t, enabled, rst.Enabled)

	enabled = false
	require.NoError(t, db.ChangeWalletPreference(enabled))
	rst, err = db.GetWalletPreference()
	require.Equal(t, enabled, rst.Enabled)

	require.NoError(t, err)
}

func TestPreferences(t *testing.T) {
	appDB, appStop := setupAppTestDb(t)
	defer appStop()

	db, stop := setupTestDB(t, appDB)
	defer stop()

	enabled := true

	require.NoError(t, db.ChangeWalletPreference(enabled))
	rst, err := db.GetPreferences()

	require.Equal(t, 1, len(rst))
	require.Equal(t, enabled, rst[0].Enabled)

	require.NoError(t, err)
}