lobo/ddblocal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add ability for tests to retain the in memory db

Browse Source
This commit is contained in:
Emil Lerch 2024-02-01 08:43:43 -08:00
parent 17dea476e3
commit e94184419e
Signed by: lobo
GPG Key ID: A7B62D657EF764F8
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/Account.zig
View File

@ -29,10 +29,15 @@ pub fn deinit(self: Self) void {
} }
pub var data_dir: []const u8 = ""; pub var data_dir: []const u8 = "";
pub var test_retain_db: bool = false;
var test_db: ?sqlite.Db = null;
/// Gets the database for this account. If under test, a memory database is used /// Gets the database for this account. If under test, a memory database is used
/// instead. Will initialize the database with appropriate metadata tables /// instead. Will initialize the database with appropriate metadata tables
pub fn dbForAccount(allocator: std.mem.Allocator, account_id: []const u8) !sqlite.Db { pub fn dbForAccount(allocator: std.mem.Allocator, account_id: []const u8) !sqlite.Db {
const builtin = @import("builtin");
if (builtin.is_test and test_retain_db)
if (test_db) |db| return db;
// TODO: Need to move this function somewhere central // TODO: Need to move this function somewhere central
// TODO: Need configuration for what directory to use // TODO: Need configuration for what directory to use
// TODO: Should this be a pool, and if so, how would we know when to close? // TODO: Should this be a pool, and if so, how would we know when to close?
@ -40,7 +45,7 @@ pub fn dbForAccount(allocator: std.mem.Allocator, account_id: []const u8) !sqlit
defer allocator.free(file_without_path); defer allocator.free(file_without_path);
const db_file_name = try std.fs.path.joinZ(allocator, &[_][]const u8{ data_dir, file_without_path }); const db_file_name = try std.fs.path.joinZ(allocator, &[_][]const u8{ data_dir, file_without_path });
defer allocator.free(db_file_name); defer allocator.free(db_file_name);
const mode = if (@import("builtin").is_test) sqlite.Db.Mode.Memory else sqlite.Db.Mode{ .File = db_file_name }; const mode = if (builtin.is_test) sqlite.Db.Mode.Memory else sqlite.Db.Mode{ .File = db_file_name };
const new = mode == .Memory or (std.fs.cwd().statFile(file_without_path) catch null == null); const new = mode == .Memory or (std.fs.cwd().statFile(file_without_path) catch null == null);
var db = try sqlite.Db.init(.{ var db = try sqlite.Db.init(.{
.mode = mode, .mode = mode,
@ -68,5 +73,6 @@ pub fn dbForAccount(allocator: std.mem.Allocator, account_id: []const u8) !sqlit
\\ PayPerRequestDateTime INTEGER DEFAULT 0, \\ PayPerRequestDateTime INTEGER DEFAULT 0,
\\ PRIMARY KEY(TableName)) \\ PRIMARY KEY(TableName))
, .{}, .{}); , .{}, .{});
if (builtin.is_test and test_retain_db) test_db = db;
return db; return db;
} }