fix: unit test

graphprotocol · Apr 22, 2024 · 8389340 · 8389340
1 parent bf38303
commit 8389340
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/availability-oracle/src/test.rs b/availability-oracle/src/test.rs
@@ -1,4 +1,5 @@
 use crate::contract;
+use crate::epoch_block_oracle_subgraph::*;
 use crate::ipfs::*;
 use crate::network_subgraph::*;
 use crate::util::bytes32_to_cid_v0;
@@ -51,7 +52,7 @@ async fn test_reconcile() {
         Arc::new(MockSubgraph),
         0,
         Duration::default(),
-        &vec!["mainnet".into()],
+        Arc::new(MockEBOSubgraph),
         &vec![
             "ethereum".into(),
             "ethereum/contract".into(),
@@ -95,6 +96,14 @@ impl NetworkSubgraph for MockSubgraph {
     }
 }
 
+struct MockEBOSubgraph;
+
+impl EpochBlockOracleSubgraph for MockEBOSubgraph {
+    fn supported_networks(self: Arc<Self>) -> Pin<Box<dyn Stream<Item = Result<String, Error>>>> {
+        futures::stream::iter(vec![Ok("mainnet".to_string())]).boxed()
+    }
+}
+
 struct MockIpfs;
 
 #[async_trait]