refactor: use utils module and add new but simple test

Author: neithanmo

integration-tests/src/main.rs

@@ -2,16 +2,17 @@
 // SPDX-License-Identifier: Apache-2.0
 
 use anyhow::Result;
-use rav_tests::test_tap_rav_v1;
+use rav_tests::{test_invalid_chain_id, test_tap_rav_v1};
 
 mod metrics;
 mod rav_tests;
-mod receipt;
+mod utils;
+
 use metrics::MetricsChecker;
-use receipt::create_tap_receipt;
 
 #[tokio::main]
 async fn main() -> Result<()> {
     // Run the TAP receipt test
+    test_invalid_chain_id().await?;
     test_tap_rav_v1().await
 }

integration-tests/src/rav_tests.rs

@@ -8,12 +8,22 @@ use std::str::FromStr;
 use std::sync::Arc;
 use std::time::Duration;
 use thegraph_core::alloy::primitives::Address;
+use thegraph_core::alloy::signers::local::PrivateKeySigner;
 
+use crate::utils::{create_request, create_tap_receipt, find_allocation};
 use crate::MetricsChecker;
 
-// TODO: Would be nice to read this values from:
-// contrib/tap-agent/config.toml
-// and contrib/local-network/.env
+const INDEXER_URL: &str = "http://localhost:7601";
+// Taken from .env
+// this is the key gateway uses
+const ACCOUNT0_SECRET: &str = "ac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80";
+
+// The deployed gateway and indexer
+// use this verifier contract
+// which must be part of the eip712 domain
+const TAP_VERIFIER_CONTRACT: &str = "0x8198f5d8F8CfFE8f9C413d98a0A55aEB8ab9FbB7";
+const CHAIN_ID: u64 = 1337;
+
 const GATEWAY_URL: &str = "http://localhost:7700";
 const SUBGRAPH_ID: &str = "BFr2mx7FgkJ36Y6pE5BiXs1KmNUmVDCnL82KUSdcLW1g";
 const GATEWAY_API_KEY: &str = "deadbeefdeadbeefdeadbeefdeadbeef";
@@ -30,42 +40,20 @@ const NUM_RECEIPTS: u32 = 3;
 const BATCHES: u32 = 2;
 const MAX_TRIGGERS: usize = 100;
 
+const GRT_DECIMALS: u8 = 18;
+const GRT_BASE: u128 = 10u128.pow(GRT_DECIMALS as u32);
+
+const MAX_RECEIPT_VALUE: u128 = GRT_BASE / 10_000;
+
 // Function to test the tap RAV generation
 pub async fn test_tap_rav_v1() -> Result<()> {
     // Setup HTTP client
     let http_client = Arc::new(Client::new());
 
     // Query the network subgraph to find active allocations
-    println!("Querying for active allocations...");
-    let response = http_client
-        .post(GRAPH_URL)
-        .json(&json!({
-            "query": "{ allocations(where: { status: Active }) { id indexer { id } subgraphDeployment { id } } }"
-        }))
-        .send()
-        .await?;
-
-    if !response.status().is_success() {
-        return Err(anyhow::anyhow!(
-            "Network subgraph request failed with status: {}",
-            response.status()
-        ));
-    }
-
-    // Try to find a valid allocation
-    let response_text = response.text().await?;
-
-    let json_value = serde_json::from_str::<serde_json::Value>(&response_text)?;
-    let allocation_id = json_value
-        .get("data")
-        .and_then(|d| d.get("allocations"))
-        .and_then(|a| a.as_array())
-        .filter(|arr| !arr.is_empty())
-        .and_then(|arr| arr[0].get("id"))
-        .and_then(|id| id.as_str())
-        .ok_or_else(|| anyhow::anyhow!("No valid allocation ID found"))?;
+    let allocation_id = find_allocation(http_client.clone(), GRAPH_URL).await?;
 
-    let allocation_id = Address::from_str(allocation_id)?;
+    let allocation_id = Address::from_str(&allocation_id)?;
 
     // Create a metrics checker
     let metrics_checker =
@@ -205,3 +193,41 @@ pub async fn test_tap_rav_v1() -> Result<()> {
     println!("❌ TEST FAILED: No RAV generation detected");
     Err(anyhow::anyhow!("Failed to detect RAV generation"))
 }
+
+pub async fn test_invalid_chain_id() -> Result<()> {
+    let wallet: PrivateKeySigner = ACCOUNT0_SECRET.parse().unwrap();
+
+    let http_client = Arc::new(Client::new());
+
+    let allocation_id = find_allocation(http_client.clone(), GRAPH_URL).await?;
+
+    let allocation_id = Address::from_str(&allocation_id)?;
+    println!("Found allocation ID: {}", allocation_id);
+
+    let receipt = create_tap_receipt(
+        MAX_RECEIPT_VALUE,
+        &allocation_id,
+        TAP_VERIFIER_CONTRACT,
+        CHAIN_ID + 18,
+        &wallet,
+    )?;
+
+    let receipt_json = serde_json::to_string(&receipt).unwrap();
+    let response = create_request(
+        &http_client,
+        format!("{}/subgraphs/id/{}", INDEXER_URL, SUBGRAPH_ID).as_str(),
+        &receipt_json,
+        &json!({
+            "query": "{ _meta { block { number } } }"
+        }),
+    )
+    .send()
+    .await?;
+
+    assert!(
+        response.status().is_client_error(),
+        "Failed to send receipt"
+    );
+
+    Ok(())
+}