Browse Source

LIB-12: Dummy List manager prototype complete

LIB-12
heck 4 years ago
parent
commit
3079dff801
  1. 252
      test/test_sqlite3.cc

252
test/test_sqlite3.cc

@ -1,8 +1,27 @@
#include "../src/sqlite3.h"
#include "../src/pEpLog.hh"
#include <iostream>
#include <cstdio>
#include <exception>
#include <stdexcept>
//TODO: add const
using namespace std;
sqlite3 *db;
string db_path;
void print_exception(const exception &e, int level = 0)
{
cerr << string(level, ' ') << "exception: " << e.what() << endl;
try {
rethrow_if_nested(e);
} catch (const exception &e) {
print_exception(e, level + 1);
} catch (...) {
}
}
static int callback(void *NotUsed, int argc, char **argv, char **azColName)
{
@ -16,39 +35,240 @@ static int callback(void *NotUsed, int argc, char **argv, char **azColName)
void execute_sql(string stmt)
{
char *zErrMsg = 0;
// pEpLog("execute_sql(\"" + stmt + "\")");
char *zErrMsg = nullptr;
int rc = sqlite3_exec(db, stmt.c_str(), callback, 0, &zErrMsg);
if (rc != SQLITE_OK) {
cout << "SQL error: " << zErrMsg << endl;
runtime_error e{ string("execute_sql: " + string(sqlite3_errmsg(db)) + ":" + string(zErrMsg)) };
sqlite3_free(zErrMsg);
throw(e);
}
}
void open_db()
void create_or_open_db()
{
int rc = sqlite3_open("test.db", &db);
pEpLog("called");
int rc = sqlite3_open(db_path.c_str(), &db);
if (rc) {
cout << "Can't open database:" << sqlite3_errmsg(db) << endl;
} else {
cout << "Opened database successfully" << endl;
runtime_error e{ string("Can't open database (" + db_path + "):" + sqlite3_errmsg(db)) };
throw(e);
}
}
void delete_db()
{
pEpLog("called");
remove(db_path.c_str());
if (errno) {
cerr << "could not delete db (" + db_path + "): " << strerror(errno) << endl;
}
}
void db_config()
{
pEpLog("called");
try {
string sql;
sql = "PRAGMA foreign_keys=ON";
execute_sql(sql);
} catch (...) {
runtime_error e("db_config() - failed with exception");
throw_with_nested(e);
}
}
void create_tables()
{
pEpLog("called");
try {
string sql;
sql = "CREATE TABLE IF NOT EXISTS identities("
"id INTEGER PRIMARY KEY AUTOINCREMENT,"
"address TEXT NOT NULL,"
"UNIQUE (address));";
execute_sql(sql);
sql = "CREATE TABLE IF NOT EXISTS groups("
"id INTEGER PRIMARY KEY AUTOINCREMENT,"
"address TEXT NOT NULL,"
"manager_id INT NOT NULL,"
"UNIQUE (address),"
"FOREIGN KEY(manager_id) REFERENCES identities(id) ON DELETE RESTRICT);";
execute_sql(sql);
sql = "CREATE TABLE IF NOT EXISTS members("
"id INTEGER PRIMARY KEY AUTOINCREMENT,"
"group_id INT NOT NULL,"
"ident_id INT NOT NULL,"
"UNIQUE (group_id, ident_id),"
"FOREIGN KEY(group_id) REFERENCES groups(id) ON DELETE CASCADE,"
"FOREIGN KEY(ident_id) REFERENCES identities(id) ON DELETE RESTRICT);";
execute_sql(sql);
} catch (...) {
runtime_error e("create_tables() - failed with exception");
throw_with_nested(e);
}
}
void prune_identities()
{
pEpLog("called");
try {
string sql = "DELETE FROM identities WHERE identities.id NOT IN (SELECT ident_id FROM members) "
"AND identities.id NOT IN (SELECT manager_id FROM groups) ";
execute_sql(sql);
} catch (...) {
runtime_error e("prune_identities() - failed with exception");
throw_with_nested(e);
}
}
void insert_identity(string addr)
{
pEpLog("insert_identity(addr: \"" + addr + "\")");
try {
string sql;
sql = "INSERT OR IGNORE INTO identities (address) VALUES ('" + addr + "');";
execute_sql(sql);
} catch (...) {
runtime_error e("insert_identity(" + addr + ") - failed with exception");
throw_with_nested(e);
}
}
void delete_identity(string addr)
{
pEpLog("delete_identity(addr: \"" + addr + "\")");
try {
string sql;
sql = "DELETE FROM identities WHERE identities.address = '" + addr + "'";
execute_sql(sql);
} catch (...) {
throw_with_nested(runtime_error("delete_identity(" + addr + ") - failed with exception"));
}
}
void insert_group(string addr_grp, string addr_mgr)
{
pEpLog("insert_group(addr_grp: \"" + addr_grp + "\"\taddr_mgr: \"" + addr_mgr + "\")");
try {
insert_identity(addr_mgr);
string sql = "INSERT INTO groups(address, manager_id)"
"VALUES ('" +
addr_grp + "',(SELECT id FROM identities WHERE identities.address = '" +
addr_mgr + "'));";
execute_sql(sql);
} catch (...) {
prune_identities();
runtime_error e("insert_group(" + addr_grp + ", " + addr_mgr + ") - failed with exception");
throw_with_nested(e);
}
}
void delete_group(string addr)
{
pEpLog("delete_group(addr: \"" + addr + "\")");
try {
string sql;
sql = "DELETE FROM groups WHERE groups.address = '" + addr + "'";
execute_sql(sql);
// dont need to delete the group fk from members table since ON DELETE CASCADE
prune_identities();
} catch (...) {
prune_identities();
runtime_error e("delete_group(" + addr + ") - failed with exception");
throw_with_nested(e);
}
}
void insert_member(string addr_grp, string addr_member)
{
pEpLog("insert_member(addr_grp: \"" + addr_grp + "\"\taddr_member: \"" + addr_member + "\")");
try {
insert_identity(addr_member);
string sql = "INSERT INTO members(group_id, ident_id)"
"VALUES ("
"(SELECT id FROM groups WHERE groups.address = '" +
addr_grp +
"'),"
"(SELECT id FROM identities WHERE identities.address = '" +
addr_member + "'));";
execute_sql(sql);
} catch (...) {
prune_identities();
runtime_error e(
"insert_member(" + addr_grp + ", " + addr_member + ") - failed with exception");
throw_with_nested(e);
}
}
void delete_member(string addr_grp, string addr_member)
{
pEpLog("delete_member(addr_grp: \"" + addr_grp + "\",addr_member: '\"" + addr_member + "\")");
try {
string sql;
sql = "DELETE FROM members WHERE"
"(members.ident_id = (SELECT id FROM identities WHERE identities.address = '" +
addr_member +
"'))"
"AND (members.group_id = (SELECT id FROM groups WHERE groups.address = '" +
addr_grp + "'));";
execute_sql(sql);
prune_identities();
} catch (...) {
prune_identities();
runtime_error e(
"delete_member(" + addr_grp + ", " + addr_member + ") - failed with exception");
throw_with_nested(e);
}
}
int main(int argc, char *argv[])
{
string sql;
open_db();
sql = "CREATE TABLE GROUPS("
"ADDRESS TEXT PRIMARY KEY NOT NULL,"
"MANAGER_ADDR TEXT NOT NULL);";
pEp::Adapter::pEpLog::set_enabled(true);
db_path = "test.db";
string dummy_in;
try {
delete_db();
create_or_open_db();
db_config();
create_tables();
execute_sql(sql);
insert_group("grp1@peptest.org", "alice@peptest.org");
insert_group("grp2@peptest.org", "alice@peptest.org");
sql = "INSERT INTO GROUPS (ADDRESS, MANAGER_ADDR) "
"VALUES ('fdjklsf@fsffd.com', 'mgr1@fsd.com');";
try {
insert_group("grp1@peptest.org", "bob@peptest.org");
assert(false);
} catch (const exception &e) {
print_exception(e);
}
execute_sql(sql);
cin >> dummy_in;
insert_member("grp1@peptest.org", "bob@peptest.org");
insert_member("grp1@peptest.org", "carol@peptest.org");
insert_member("grp1@peptest.org", "joe@peptest.org");
cin >> dummy_in;
delete_group("grp1@peptest.org");
cin >> dummy_in;
try {
delete_identity("alice@peptest.org");
assert(false);
} catch (const exception &e) {
print_exception(e);
}
} catch (const exception &e) {
print_exception(e);
}
sqlite3_close(db);
}
Loading…
Cancel
Save