TEST

Author: tnull

src/test/utils.rs

@@ -1,4 +1,5 @@
 use crate::builder::NodeBuilder;
+use crate::io::sqlite_store::{self, SqliteStore};
 use crate::io::test_utils::TestSyncStore;
 use crate::{Config, Node};
 use lightning::util::logger::{Level, Logger, Record};
@@ -176,7 +177,7 @@ pub fn setup_bitcoind_and_electrsd() -> (BitcoinD, ElectrsD) {
 
 pub(crate) fn setup_two_nodes(
 	electrsd: &ElectrsD, allow_0conf: bool,
-) -> (Node<TestSyncStore>, Node<TestSyncStore>) {
+) -> (Node<SqliteStore>, Node<SqliteStore>) {
 	println!("== Node A ==");
 	let config_a = random_config();
 	let node_a = setup_node(electrsd, config_a);
@@ -190,12 +191,22 @@ pub(crate) fn setup_two_nodes(
 	(node_a, node_b)
 }
 
-pub(crate) fn setup_node(electrsd: &ElectrsD, config: Config) -> Node<TestSyncStore> {
+pub(crate) fn setup_node(electrsd: &ElectrsD, config: Config) -> Node<SqliteStore> {
 	let esplora_url = format!("http://{}", electrsd.esplora_url.as_ref().unwrap());
 	let mut builder = NodeBuilder::from_config(config.clone());
 	builder.set_esplora_server(esplora_url.clone());
-	let test_sync_store = Arc::new(TestSyncStore::new(config.storage_dir_path.into()));
-	let node = builder.build_with_store(test_sync_store).unwrap();
+
+	let mut sql_dir = PathBuf::from(config.storage_dir_path);
+	sql_dir.push("sqlite_store");
+	let sqlite_store = Arc::new(
+		SqliteStore::new(
+			sql_dir,
+			Some("test_sync_db".to_string()),
+			Some("test_sync_table".to_string()),
+		)
+		.unwrap(),
+	);
+	let node = builder.build_with_store(sqlite_store).unwrap();
 	node.start().unwrap();
 	node
 }