Skip to content

Commit

Permalink
test(host/tests): add case for v2 report
Browse files Browse the repository at this point in the history
  • Loading branch information
bendanzhentan committed Dec 10, 2024
1 parent ca0e874 commit 9e71384
Show file tree
Hide file tree
Showing 4 changed files with 19 additions and 4 deletions.
2 changes: 1 addition & 1 deletion host/tests/common/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,6 @@ mod setup;

pub use chain::randomly_select_block;
pub use client::Client;
pub use request::{complete_proof_request, make_proof_request};
pub use request::{complete_proof_request, make_proof_request, v2_assert_report};
pub use server::{TestServerBuilder, TestServerHandle};
pub use setup::setup;
11 changes: 10 additions & 1 deletion host/tests/common/request.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ use raiko_core::interfaces::{ProofRequestOpt, ProverSpecificOpts};
use raiko_host::server::api;
use raiko_lib::consts::Network;
use raiko_lib::proof_type::ProofType;
use raiko_tasks::TaskStatus;
use raiko_tasks::{TaskReport, TaskStatus};
use serde_json::json;

use crate::common::Client;
Expand Down Expand Up @@ -87,3 +87,12 @@ pub async fn v2_complete_proof_request(client: &Client, request: &ProofRequestOp
}
panic!("proof generation failed, error: timeout");
}

/// Assert that the report is in the expected format.
pub async fn v2_assert_report(client: &Client) -> Vec<TaskReport> {
let response = client
.get(&format!("/v2/proof/report"))
.await
.expect("failed to send request");
response.json().await.expect("failed to decode report body")
}
3 changes: 2 additions & 1 deletion host/tests/test/manual_test.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use crate::common::{complete_proof_request, Client};
use crate::common::{complete_proof_request, v2_assert_report, Client};
use raiko_core::interfaces::{ProofRequestOpt, ProverSpecificOpts};
use raiko_host::server::api;
use raiko_tasks::TaskStatus;
Expand Down Expand Up @@ -89,4 +89,5 @@ pub async fn test_manual_prove() {
);

complete_proof_request(&api_version, &client, &request).await;
v2_assert_report(&client).await;
}
7 changes: 6 additions & 1 deletion host/tests/test/prove_test.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
use crate::common::{complete_proof_request, make_proof_request, randomly_select_block, setup};
use crate::common::{
complete_proof_request, make_proof_request, randomly_select_block, setup, v2_assert_report,
};
use raiko_host::server::api;
use raiko_lib::consts::Network;
use raiko_lib::proof_type::ProofType;
Expand Down Expand Up @@ -40,4 +42,7 @@ pub async fn test_v2_mainnet_native_prove() {

// sending the same completed request should should be ok
complete_proof_request(api_version, &client, &request).await;

// santy check for report format
v2_assert_report(&client).await;
}

0 comments on commit 9e71384

Please sign in to comment.