Skip to content

Commit

Permalink
Fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
stefan-mysten committed May 16, 2024
1 parent ccd87eb commit fe0a0cd
Showing 1 changed file with 23 additions and 11 deletions.
34 changes: 23 additions & 11 deletions crates/sui-graphql-rpc/src/server/builder.rs
Original file line number Diff line number Diff line change
Expand Up @@ -677,16 +677,19 @@ pub mod tests {
let mut cfg = ServiceConfig::default();
cfg.limits.request_timeout_ms = timeout.as_millis() as u64;
cfg.limits.mutation_timeout_ms = timeout.as_millis() as u64;

info!("prepping schema");
let schema = prep_schema(None, Some(cfg))
.context_data(sui_client.clone())
.context_data(Some(sui_client.clone()))
.extension(Timeout)
.extension(TimedExecuteExt {
min_req_delay: delay,
})
.build_schema();

schema.execute(query).await
info!("executing schema");
let result = schema.execute(query).await;
info!("finished execution");
result
}

let query = "{ chainIdentifier }";
Expand All @@ -710,6 +713,8 @@ pub mod tests {
let exp = format!("Query request timed out. Limit: {}s", delay.as_secs_f32());
assert_eq!(errs, vec![exp]);

println!("API Version {:?}", sui_client.api_version());

// Should timeout for mutation
// Create a transaction and sign it, and use the tx_bytes + signatures for the GraphQL
// executeTransactionBlock mutation call.
Expand All @@ -729,16 +734,23 @@ pub mod tests {

let tx = wallet.sign_transaction(&tx_data);
let (tx_bytes, signatures) = tx.to_tx_bytes_and_signatures();

println!("TX BYTES: {:?}", tx_bytes.encoded());

let signature_base64 = &signatures[0];
let query = r#"
mutation {
executeTransactionBlock(txBytes: "{tx_bytes}", signatures: "{signature_base64}") {
effects {
let query = format!(
r#"
mutation {{
executeTransactionBlock(txBytes: "{}", signatures: "{}") {{
effects {{
status
}
}
}"#;
let errs: Vec<_> = test_timeout(delay, delay, query, &sui_client)
}}
}}
}}"#,
tx_bytes.encoded(),
signature_base64.encoded()
);
let errs: Vec<_> = test_timeout(delay, delay, &query, &sui_client)
.await
.into_result()
.unwrap_err()
Expand Down

0 comments on commit fe0a0cd

Please sign in to comment.