cuberite/SQLiteCpp
1
0
Fork 0
mirror of https://github.com/cuberite/SQLiteCpp.git synced 2025-08-04 17:56:13 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add unit test for Database::createFunction()

Browse Source 
using an example from
https://stackoverflow.com/a/8283265/1163698 How can I create a user-defined function in SQLite?
This commit is contained in:
Sébastien Rombauts 2019-12-29 22:03:11 +01:00
parent 41cf3c5035
commit dc3f1ac271
3 changed files with 48 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
include/SQLiteCpp/Database.h
View File

@ -336,9 +336,9 @@ public:
bool abDeterministic, bool abDeterministic,
void* apApp, void* apApp,
void (*apFunc)(sqlite3_context *, int, sqlite3_value **), void (*apFunc)(sqlite3_context *, int, sqlite3_value **),
void (*apStep)(sqlite3_context *, int, sqlite3_value **), void (*apStep)(sqlite3_context *, int, sqlite3_value **) = nullptr,
void (*apFinal)(sqlite3_context *), // NOLINT(readability/casting) void (*apFinal)(sqlite3_context *) = nullptr, // NOLINT(readability/casting)
void (*apDestroy)(void *)); void (*apDestroy)(void *) = nullptr);
/** /**
* @brief Create or redefine a SQL function or aggregate in the sqlite database. * @brief Create or redefine a SQL function or aggregate in the sqlite database.
@ -364,12 +364,12 @@ public:
bool abDeterministic, bool abDeterministic,
void* apApp, void* apApp,
void (*apFunc)(sqlite3_context *, int, sqlite3_value **), void (*apFunc)(sqlite3_context *, int, sqlite3_value **),
void (*apStep)(sqlite3_context *, int, sqlite3_value **), void (*apStep)(sqlite3_context *, int, sqlite3_value **) = nullptr,
void (*apFinal)(sqlite3_context *), // NOLINT(readability/casting) void (*apFinal)(sqlite3_context *) = nullptr,
void (*apDestroy)(void *)) void (*apDestroy)(void *) = nullptr)
{ {
return createFunction(aFuncName.c_str(), aNbArg, abDeterministic, createFunction(aFuncName.c_str(), aNbArg, abDeterministic,
apApp, apFunc, apStep, apFinal, apDestroy); apApp, apFunc, apStep, apFinal, apDestroy);
} }
/** /**

12
src/Database.cpp
View File

@ -192,17 +192,17 @@ void Database::createFunction(const char* apFuncName,
bool abDeterministic, bool abDeterministic,
void* apApp, void* apApp,
void (*apFunc)(sqlite3_context *, int, sqlite3_value **), void (*apFunc)(sqlite3_context *, int, sqlite3_value **),
void (*apStep)(sqlite3_context *, int, sqlite3_value **), void (*apStep)(sqlite3_context *, int, sqlite3_value **) /* = nullptr */,
void (*apFinal)(sqlite3_context *), // NOLINT(readability/casting) void (*apFinal)(sqlite3_context *) /* = nullptr */, // NOLINT(readability/casting)
void (*apDestroy)(void *)) void (*apDestroy)(void *) /* = nullptr */)
{ {
int TextRep = SQLITE_UTF8; int textRep = SQLITE_UTF8;
// optimization if deterministic function (e.g. of nondeterministic function random()) // optimization if deterministic function (e.g. of nondeterministic function random())
if (abDeterministic) if (abDeterministic)
{ {
TextRep = TextRep|SQLITE_DETERMINISTIC; textRep = textRep | SQLITE_DETERMINISTIC;
} }
const int ret = sqlite3_create_function_v2(mpSQLite, apFuncName, aNbArg, TextRep, const int ret = sqlite3_create_function_v2(mpSQLite, apFuncName, aNbArg, textRep,
apApp, apFunc, apStep, apFinal, apDestroy); apApp, apFunc, apStep, apFinal, apDestroy);
check(ret); check(ret);
} }

35
tests/Database_test.cpp
View File

@ -311,7 +311,40 @@ TEST(Database, execException)
EXPECT_STREQ("table test has 3 columns but 4 values were supplied", db.getErrorMsg()); EXPECT_STREQ("table test has 3 columns but 4 values were supplied", db.getErrorMsg());
} }
// TODO: test Database::createFunction() // From https://stackoverflow.com/a/8283265/1163698 How can I create a user-defined function in SQLite?
static void firstchar(sqlite3_context *context, int argc, sqlite3_value **argv)
{
if (argc == 1)
{
const unsigned char *text = sqlite3_value_text(argv[0]);
if (text && text[0])
{
char result[2];
result[0] = text[0]; result[1] = '\0';
sqlite3_result_text(context, result, -1, SQLITE_TRANSIENT);
return;
}
}
sqlite3_result_null(context);
}
TEST(Database, createFunction)
{
// Create a new database
SQLite::Database db(":memory:", SQLite::OPEN_READWRITE);
db.exec("CREATE TABLE test (id INTEGER PRIMARY KEY, value TEXT)");
EXPECT_EQ(1, db.exec("INSERT INTO test VALUES (NULL, \"first\")"));
EXPECT_EQ(1, db.exec("INSERT INTO test VALUES (NULL, \"second\")"));
// exception with SQL error: "no such function: firstchar"
EXPECT_THROW(db.exec("SELECT firstchar(value) FROM test WHERE id=1"), SQLite::Exception);
db.createFunction("firstchar", 1, true, nullptr, &firstchar, nullptr, nullptr, nullptr);
EXPECT_EQ(1, db.exec("SELECT firstchar(value) FROM test WHERE id=1"));
}
// TODO: test Database::loadExtension() // TODO: test Database::loadExtension()
#ifdef SQLITE_HAS_CODEC #ifdef SQLITE_HAS_CODEC