f Update TestSyncStore

Author: tnull

lightning-persister/src/test_utils.rs

@@ -142,10 +142,10 @@ impl TestSyncStore {
 }
 
 impl KVStore for TestSyncStore {
-	fn read(&self, namespace: &str, key: &str) -> std::io::Result<Vec<u8>> {
-		let fs_res = self.fs_store.read(namespace, key);
-		let sqlite_res = self.sqlite_store.read(namespace, key);
-		let test_res = self.test_store.read(namespace, key);
+	fn read(&self, namespace: &str, sub_namespace: &str, key: &str) -> std::io::Result<Vec<u8>> {
+		let fs_res = self.fs_store.read(namespace, sub_namespace, key);
+		let sqlite_res = self.sqlite_store.read(namespace, sub_namespace, key);
+		let test_res = self.test_store.read(namespace, sub_namespace, key);
 
 		match fs_res {
 			Ok(read) => {
@@ -163,12 +163,12 @@ impl KVStore for TestSyncStore {
 		}
 	}
 
-	fn write(&self, namespace: &str, key: &str, buf: &[u8]) -> std::io::Result<()> {
-		let fs_res = self.fs_store.write(namespace, key, buf);
-		let sqlite_res = self.sqlite_store.write(namespace, key, buf);
-		let test_res = self.test_store.write(namespace, key, buf);
+	fn write(&self, namespace: &str, sub_namespace: &str, key: &str, buf: &[u8]) -> std::io::Result<()> {
+		let fs_res = self.fs_store.write(namespace, sub_namespace, key, buf);
+		let sqlite_res = self.sqlite_store.write(namespace, sub_namespace, key, buf);
+		let test_res = self.test_store.write(namespace, sub_namespace, key, buf);
 
-		assert!(self.list(namespace).unwrap().contains(&key.to_string()));
+		assert!(self.list(namespace, sub_namespace).unwrap().contains(&key.to_string()));
 
 		match fs_res {
 			Ok(()) => {
@@ -184,12 +184,12 @@ impl KVStore for TestSyncStore {
 		}
 	}
 
-	fn remove(&self, namespace: &str, key: &str) -> std::io::Result<()> {
-		let fs_res = self.fs_store.remove(namespace, key);
-		let sqlite_res = self.sqlite_store.remove(namespace, key);
-		let test_res = self.test_store.remove(namespace, key);
+	fn remove(&self, namespace: &str, sub_namespace: &str, key: &str, lazy: bool) -> std::io::Result<()> {
+		let fs_res = self.fs_store.remove(namespace, sub_namespace, key, lazy);
+		let sqlite_res = self.sqlite_store.remove(namespace, sub_namespace, key, lazy);
+		let test_res = self.test_store.remove(namespace, sub_namespace, key, lazy);
 
-		assert!(!self.list(namespace).unwrap().contains(&key.to_string()));
+		assert!(!self.list(namespace, sub_namespace).unwrap().contains(&key.to_string()));
 
 		match fs_res {
 			Ok(()) => {
@@ -205,10 +205,10 @@ impl KVStore for TestSyncStore {
 		}
 	}
 
-	fn list(&self, namespace: &str) -> std::io::Result<Vec<String>> {
-		let fs_res = self.fs_store.list(namespace);
-		let sqlite_res = self.sqlite_store.list(namespace);
-		let test_res = self.test_store.list(namespace);
+	fn list(&self, namespace: &str, sub_namespace: &str) -> std::io::Result<Vec<String>> {
+		let fs_res = self.fs_store.list(namespace, sub_namespace);
+		let sqlite_res = self.sqlite_store.list(namespace, sub_namespace);
+		let test_res = self.test_store.list(namespace, sub_namespace);
 
 		match fs_res {
 			Ok(mut list) => {