From 36b570c93b392e71564ea4e7149c108bfafee5d3 Mon Sep 17 00:00:00 2001
From: Timothy Quilling <teqed@shatteredsky.net>
Date: Tue, 4 Nov 2025 00:33:43 -0500
Subject: [PATCH] fix: tests

---
 consumer/src/lib.rs          | 8 ++++----
 consumer/tests/common/mod.rs | 5 +++--
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/consumer/src/lib.rs b/consumer/src/lib.rs
index 04716b66..39b20bba 100644
--- a/consumer/src/lib.rs
+++ b/consumer/src/lib.rs
@@ -6,8 +6,8 @@
 
 mod cmd;
 mod config;
-mod database_writer;
-mod db;
+pub mod database_writer;
+pub mod db;
 mod error;
 mod events;
 mod external;
@@ -15,9 +15,9 @@ mod indexer;
 // mod label_indexer; // Disabled - will be reimplemented with new cursor system
 mod relay;
 mod sources;
-mod types;
+pub mod types;
 mod utils;
-mod workers;
+pub mod workers;
 
 // Re-export key types and functions for binary use
 pub use cmd::{parse, Cli};
diff --git a/consumer/tests/common/mod.rs b/consumer/tests/common/mod.rs
index b96d1d18..cc9c84c6 100644
--- a/consumer/tests/common/mod.rs
+++ b/consumer/tests/common/mod.rs
@@ -202,8 +202,9 @@ pub fn unwrap_query<T>(result: Result<T, tokio_postgres::Error>, operation: &str
 }
 
 /// Helper to unwrap SQL operations with detailed error logging and context
-pub fn unwrap_sql_result<T>(
-    result: Result<T, tokio_postgres::Error>,
+/// Now supports both tokio_postgres::Error and eyre::Report
+pub fn unwrap_sql_result<T, E: std::fmt::Debug>(
+    result: Result<T, E>,
     operation: &str,
     context: &str,
 ) -> T {
-- 
2.51.2