lightning/wallet/db_tests.c

#include "db.c"

#include "wallet/test_utils.h"

#include <stdio.h>
#include <unistd.h>

static struct db *create_test_db(const char *testname)
{
	struct db *db;
	char filename[] = "/tmp/ldb-XXXXXX";

	int fd = mkstemp(filename);
	if (fd == -1)
		return NULL;
	close(fd);

	db = db_open(NULL, filename);
	return db;
}

static bool test_empty_db_migrate(void)
{
	struct db *db = create_test_db(__func__);
	CHECK(db);
	CHECK(db_get_version(db) == -1);
	db_migrate(db);
	CHECK(db_get_version(db) == db_migration_count());

	tal_free(db);
	return true;
}

static bool test_primitives(void)
{
	struct db *db = create_test_db(__func__);
	CHECK_MSG(db_begin_transaction(db), "Starting a new transaction");
	CHECK(db->in_transaction);
	CHECK_MSG(db_commit_transaction(db), "Committing a transaction");
	CHECK(!db->in_transaction);
	CHECK_MSG(db_begin_transaction(db), "Starting a transaction after commit");
	CHECK(db_rollback_transaction(db));

	CHECK_MSG(db_exec(__func__, db, "SELECT name FROM sqlite_master WHERE type='table';"), "Simple correct SQL command");
	CHECK_MSG(!db_exec(__func__, db, "not a valid SQL statement"), "Failing SQL command");
	CHECK(!db->in_transaction);
	CHECK_MSG(db_begin_transaction(db), "Starting a transaction after a failed transaction");
	tal_free(db);

	return true;
}

static bool test_vars(void)
{
	struct db *db = create_test_db(__func__);
	char *varname = "testvar";
	CHECK(db);
	db_migrate(db);

	/* Check default behavior */
	CHECK(db_get_intvar(db, varname, 42) == 42);

	/* Check setting and getting */
	CHECK(db_set_intvar(db, varname, 1));
	CHECK(db_get_intvar(db, varname, 42) == 1);

	/* Check updating */
	CHECK(db_set_intvar(db, varname, 2));
	CHECK(db_get_intvar(db, varname, 42) == 2);

	tal_free(db);
	return true;
}

int main(void)
{
	bool ok = true;

	ok &= test_empty_db_migrate();
	ok &= test_vars();
	ok &= test_primitives();

	return !ok;
}