Skip to content

Commit

Permalink
fix(iota-indexer): Refactor IndexerApi top use the shared runtime f…
Browse files Browse the repository at this point in the history 
…or tests (#2964)

* fix(iota-indexer): refactor indexer_api to use shared runtime in tests

* fixup! fix(iota-indexer): refactor indexer_api to use shared runtime in tests
  • Loading branch information
@sergiupopescu199
sergiupopescu199 committed Oct 31, 2024
1 parent 39db832 commit f2433ac
Show file tree
Hide file tree
Showing 2 changed files with 146 additions and 128 deletions.
272 changes: 145 additions & 127 deletions crates/iota-indexer/tests/rpc-tests/indexer_api.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,138 +9,156 @@ use iota_types::{
base_types::{IotaAddress, ObjectID},
digests::TransactionDigest,
};
use serial_test::serial;

use crate::common::{
indexer_wait_for_checkpoint, rpc_call_error_msg_matches,
start_test_cluster_with_read_write_indexer,
};

#[tokio::test]
#[serial]
async fn query_events_no_events_descending() {
let (_cluster, pg_store, indexer_client) =
start_test_cluster_with_read_write_indexer(None).await;
indexer_wait_for_checkpoint(&pg_store, 1).await;

let indexer_events = indexer_client
.query_events(
EventFilter::Sender(
IotaAddress::from_str(
"0x9a934a2644c4ca2decbe3d126d80720429c5e31896aa756765afa23ae2cb4b99",
)
.unwrap(),
),
None,
None,
Some(true),
)
.await
.unwrap();

assert_eq!(indexer_events, EventPage::empty())
use crate::common::{indexer_wait_for_checkpoint, rpc_call_error_msg_matches, ApiTestSetup};

#[test]
fn query_events_no_events_descending() {
let ApiTestSetup {
runtime,
store,
client,
..
} = ApiTestSetup::get_or_init();

runtime.block_on(async move {
indexer_wait_for_checkpoint(store, 1).await;

let indexer_events = client
.query_events(
EventFilter::Sender(
IotaAddress::from_str(
"0x9a934a2644c4ca2decbe3d126d80720429c5e31896aa756765afa23ae2cb4b99",
)
.unwrap(),
),
None,
None,
Some(true),
)
.await
.unwrap();

assert_eq!(indexer_events, EventPage::empty())
});
}

#[tokio::test]
#[serial]
async fn query_events_no_events_ascending() {
let (_cluster, pg_store, indexer_client) =
start_test_cluster_with_read_write_indexer(None).await;
indexer_wait_for_checkpoint(&pg_store, 1).await;

let indexer_events = indexer_client
.query_events(
EventFilter::Sender(
IotaAddress::from_str(
"0x9a934a2644c4ca2decbe3d126d80720429c5e31896aa756765afa23ae2cb4b99",
)
.unwrap(),
),
None,
None,
None,
)
.await
.unwrap();

assert_eq!(indexer_events, EventPage::empty())
#[test]
fn query_events_no_events_ascending() {
let ApiTestSetup {
runtime,
store,
client,
..
} = ApiTestSetup::get_or_init();

runtime.block_on(async move {
indexer_wait_for_checkpoint(store, 1).await;

let indexer_events = client
.query_events(
EventFilter::Sender(
IotaAddress::from_str(
"0x9a934a2644c4ca2decbe3d126d80720429c5e31896aa756765afa23ae2cb4b99",
)
.unwrap(),
),
None,
None,
None,
)
.await
.unwrap();

assert_eq!(indexer_events, EventPage::empty())
});
}

#[tokio::test]
#[serial]
async fn query_events_unsupported_events() {
let (_cluster, pg_store, indexer_client) =
start_test_cluster_with_read_write_indexer(None).await;
indexer_wait_for_checkpoint(&pg_store, 1).await;

// Get the current time in milliseconds since the UNIX epoch
let now_millis = SystemTime::now()
.duration_since(SystemTime::UNIX_EPOCH)
.unwrap()
.as_millis();

// Subtract 10 minutes from the current time
let ten_minutes_ago = now_millis - (10 * 60 * 1000); // 600 seconds = 10 minutes

let unsupported_filters = vec![
EventFilter::All(vec![]),
EventFilter::Any(vec![]),
EventFilter::And(
Box::new(EventFilter::Any(vec![])),
Box::new(EventFilter::Any(vec![])),
),
EventFilter::Or(
Box::new(EventFilter::Any(vec![])),
Box::new(EventFilter::Any(vec![])),
),
EventFilter::TimeRange {
start_time: ten_minutes_ago as u64,
end_time: now_millis as u64,
},
EventFilter::MoveEventField {
path: String::default(),
value: serde_json::Value::Bool(true),
},
];

for event_filter in unsupported_filters {
let result = indexer_client
.query_events(event_filter, None, None, None)
.await;

assert!(rpc_call_error_msg_matches(
result,
r#"{"code":-32603,"message": "Indexer does not support the feature with error: `This type of EventFilter is not supported.`"}"#,
));
}
#[test]
fn query_events_unsupported_events() {
let ApiTestSetup {
runtime,
store,
client,
..
} = ApiTestSetup::get_or_init();

runtime.block_on(async move {
indexer_wait_for_checkpoint(store, 1).await;

// Get the current time in milliseconds since the UNIX epoch
let now_millis = SystemTime::now()
.duration_since(SystemTime::UNIX_EPOCH)
.unwrap()
.as_millis();

// Subtract 10 minutes from the current time
let ten_minutes_ago = now_millis - (10 * 60 * 1000); // 600 seconds = 10 minutes

let unsupported_filters = vec![
EventFilter::All(vec![]),
EventFilter::Any(vec![]),
EventFilter::And(
Box::new(EventFilter::Any(vec![])),
Box::new(EventFilter::Any(vec![])),
),
EventFilter::Or(
Box::new(EventFilter::Any(vec![])),
Box::new(EventFilter::Any(vec![])),
),
EventFilter::TimeRange {
start_time: ten_minutes_ago as u64,
end_time: now_millis as u64,
},
EventFilter::MoveEventField {
path: String::default(),
value: serde_json::Value::Bool(true),
},
];

for event_filter in unsupported_filters {
let result = client
.query_events(event_filter, None, None, None)
.await;

assert!(rpc_call_error_msg_matches(
result,
r#"{"code":-32603,"message": "Indexer does not support the feature with error: `This type of EventFilter is not supported.`"}"#,
));
}
});
}

#[tokio::test]
#[serial]
async fn query_events_supported_events() {
let (_cluster, pg_store, indexer_client) =
start_test_cluster_with_read_write_indexer(None).await;
indexer_wait_for_checkpoint(&pg_store, 1).await;

let supported_filters = vec![
EventFilter::Sender(IotaAddress::ZERO),
EventFilter::Transaction(TransactionDigest::ZERO),
EventFilter::Package(ObjectID::ZERO),
EventFilter::MoveEventModule {
package: ObjectID::ZERO,
module: "x".parse().unwrap(),
},
EventFilter::MoveEventType("0xabcd::MyModule::Foo".parse().unwrap()),
EventFilter::MoveModule {
package: ObjectID::ZERO,
module: "x".parse().unwrap(),
},
];

for event_filter in supported_filters {
let result = indexer_client
.query_events(event_filter, None, None, None)
.await;
assert!(result.is_ok());
}
#[test]
fn query_events_supported_events() {
let ApiTestSetup {
runtime,
store,
client,
..
} = ApiTestSetup::get_or_init();

runtime.block_on(async move {
indexer_wait_for_checkpoint(store, 1).await;

let supported_filters = vec![
EventFilter::Sender(IotaAddress::ZERO),
EventFilter::Transaction(TransactionDigest::ZERO),
EventFilter::Package(ObjectID::ZERO),
EventFilter::MoveEventModule {
package: ObjectID::ZERO,
module: "x".parse().unwrap(),
},
EventFilter::MoveEventType("0xabcd::MyModule::Foo".parse().unwrap()),
EventFilter::MoveModule {
package: ObjectID::ZERO,
module: "x".parse().unwrap(),
},
];

for event_filter in supported_filters {
let result = client.query_events(event_filter, None, None, None).await;
assert!(result.is_ok());
}
});
}
2 changes: 1 addition & 1 deletion crates/iota-indexer/tests/rpc-tests/main.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ mod common;
#[cfg(feature = "pg_integration")]
mod extended_api;

#[cfg(feature = "pg_integration")]
#[cfg(feature = "shared_test_runtime")]
mod indexer_api;

#[cfg(feature = "shared_test_runtime")]
Expand Down

0 comments on commit f2433ac

Please sign in to comment.