status-go/appdatabase/test_helpers.go

package appdatabase

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

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

const kdfIterationsNumberForTests = 3200

// SetupTestSQLDB creates a temporary sqlite database file, initialises and then returns with a teardown func
func SetupTestSQLDB(prefix string) (*sql.DB, func() error, error) {
	tmpfile, err := ioutil.TempFile("", prefix)
	if err != nil {
		return nil, nil, err
	}
	db, err := InitializeDB(tmpfile.Name(), prefix, kdfIterationsNumberForTests)
	if err != nil {
		return nil, nil, err
	}

	return db, func() error {
		err := db.Close()
		if err != nil {
			return err
		}
		return os.Remove(tmpfile.Name())
	}, nil
}

func SetupTestMemorySQLDB(prefix string) (*sql.DB, error) {
	db, err := InitializeDB(sqlite.InMemoryPath, prefix, kdfIterationsNumberForTests)
	if err != nil {
		return nil, err
	}

	return db, nil
}