ik_llama.cpp/common/sqlite_modern_cpp/tests/sqlcipher.cc

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <catch.hpp>

#include <sqlite_modern_cpp/sqlcipher.h>
using namespace sqlite;
using namespace std;

struct TmpFile
{
    string fname;

    TmpFile(): fname("./sqlcipher.db") { }
    ~TmpFile() { remove(fname.c_str()); }
};

TEST_CASE("sqlcipher works", "[sqlcipher]") {
    TmpFile file;
    sqlcipher_config config;
    {
        config.key = "DebugKey";
        sqlcipher_database db(file.fname, config);

        db << "CREATE TABLE foo (a integer, b string);";
        db << "INSERT INTO foo VALUES (?, ?)" << 1 << "hello";
        db << "INSERT INTO foo VALUES (?, ?)" << 2 << "world";

        string str;
        db << "SELECT b from FOO where a=?;" << 2 >> str;

        REQUIRE(str == "world");
    }

    bool failed = false;
    try {
        config.key = "DebugKey2";
        sqlcipher_database db(file.fname, config);
        db << "INSERT INTO foo VALUES (?, ?)" << 3 << "fail";
    } catch(const errors::notadb&) {
        failed = true;
        // Expected, wrong key
    }
    REQUIRE(failed == true);

    {
        config.key = "DebugKey";
        sqlcipher_database db(file.fname, config);
        db.rekey("DebugKey2");
    }
    {
        config.key = "DebugKey2";
        sqlcipher_database db(file.fname, config);
        db << "INSERT INTO foo VALUES (?, ?)" << 3 << "fail";
    }
}