fix test

anthias-labs · Feb 27, 2024 · 93ff285 · 93ff285
1 parent 135dc59
commit 93ff285
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 1 deletion.
diff --git a/core/src/environment/mod.rs b/core/src/environment/mod.rs
@@ -569,7 +569,7 @@ impl Environment {
                                 let to_block = U256::from(
                                     filter
                                         .block_option
-                                        .get_from_block()
+                                        .get_to_block()
                                         .ok_or(ArbiterCoreError::MissingDataError)?
                                         .as_number()
                                         .ok_or(ArbiterCoreError::MissingDataError)?

diff --git a/core/tests/middleware_integration.rs b/core/tests/middleware_integration.rs
@@ -625,3 +625,39 @@ async fn stream_with_meta() {
     assert_eq!(format!("{:?}", stream.next().await), "Some(Ok((ApprovalFilter(ApprovalFilter { owner: 0x2efdc9eecfee3a776209fcb8e9a83a6b221d74f5, spender: 0x2efdc9eecfee3a776209fcb8e9a83a6b221d74f5, amount: 1 }), LogMeta { address: 0x067ea9e44c76a2620f10b39a1b51d5124a299192, block_number: 0, block_hash: 0x0000000000000000000000000000000000000000000000000000000000000000, transaction_hash: 0x0000000000000000000000000000000000000000000000000000000000000000, transaction_index: 2, log_index: 0 })))");
     assert_eq!(format!("{:?}", stream.next().await), "Some(Ok((ApprovalFilter(ApprovalFilter { owner: 0x2efdc9eecfee3a776209fcb8e9a83a6b221d74f5, spender: 0x2efdc9eecfee3a776209fcb8e9a83a6b221d74f5, amount: 1 }), LogMeta { address: 0x067ea9e44c76a2620f10b39a1b51d5124a299192, block_number: 1, block_hash: 0x0000000000000000000000000000000000000000000000000000000000000000, transaction_hash: 0x0000000000000000000000000000000000000000000000000000000000000000, transaction_index: 0, log_index: 0 })))");
 }
+
+#[tokio::test]
+async fn get_logs() {
+    let (_environment, client) = startup();
+
+    let arbx = deploy_arbx(client.clone()).await;
+
+    for _ in 0..2 {
+        arbx.approve(client.address(), eU256::from(1))
+            .send()
+            .await
+            .unwrap()
+            .await
+            .unwrap();
+    }
+
+    client.update_block(1, 1).unwrap();
+
+    arbx.approve(client.address(), eU256::from(1))
+        .send()
+        .await
+        .unwrap()
+        .await
+        .unwrap();
+
+    let filter = arbx
+        .approval_filter()
+        .filter
+        .from_block(0)
+        .to_block(1)
+        .address(arbx.address());
+    println!("filter: {:#?}", filter);
+    let logs = client.get_logs(&filter).await.unwrap();
+    println!("logs: {:#?}", logs);
+    assert_eq!(logs.len(), 3);
+}