apollo_l1_provider: minor updates to some bootstrapping tests of the L1 provider

Author: guy-starkware

crates/apollo_l1_provider/src/l1_provider_tests.rs

@@ -49,7 +49,7 @@ fn commit_block_no_rejected(
     l1_provider.commit_block(txs.iter().copied().collect(), [].into(), block_number).unwrap();
 }
 
-fn commit_block_expect_error_just_to_start_bootstrapping(
+fn call_commit_block_to_start_bootstrapping_and_expect_error(
     l1_provider: &mut L1Provider,
     block_number: BlockNumber,
 ) {
@@ -437,7 +437,7 @@ async fn bootstrap_commit_block_received_twice_no_error() {
         .build_into_l1_provider();
 
     l1_provider.initialize(BlockNumber(0), vec![]).await.expect("l1 provider initialize failed");
-    commit_block_expect_error_just_to_start_bootstrapping(&mut l1_provider, BlockNumber(2));
+    call_commit_block_to_start_bootstrapping_and_expect_error(&mut l1_provider, BlockNumber(2));
 
     commit_block_no_rejected(&mut l1_provider, &[], BlockNumber(2));
     // l1_provider.start_bootstrapping(BlockNumber(2));
@@ -459,7 +459,7 @@ async fn bootstrap_commit_block_received_twice_error_if_new_uncommitted_txs() {
         .build_into_l1_provider();
 
     l1_provider.initialize(BlockNumber(0), vec![]).await.expect("l1 provider initialize failed");
-    commit_block_expect_error_just_to_start_bootstrapping(&mut l1_provider, BlockNumber(2));
+    call_commit_block_to_start_bootstrapping_and_expect_error(&mut l1_provider, BlockNumber(2));
 
     // Test.
     commit_block_no_rejected(&mut l1_provider, &[tx_hash!(1)], BlockNumber(0));
@@ -1069,7 +1069,7 @@ async fn commit_block_historical_height_short_circuits_bootstrap() {
     let l1_provider_builder_clone = l1_provider_builder.clone();
     let mut l1_provider = l1_provider_builder.clone().build_into_l1_provider();
     l1_provider.initialize(STARTUP_HEIGHT, vec![]).await.expect("l1 provider initialize failed");
-    commit_block_expect_error_just_to_start_bootstrapping(&mut l1_provider, TARGET_HEIGHT);
+    call_commit_block_to_start_bootstrapping_and_expect_error(&mut l1_provider, TARGET_HEIGHT);
 
     let expected_unchanged = l1_provider_builder_clone
         .with_height(STARTUP_HEIGHT)
@@ -1483,6 +1483,7 @@ async fn bootstrap_e2e() {
 
     let mut l1_provider =
         L1Provider::new(config, l1_provider_client.clone(), Arc::new(sync_client), None);
+
     // Test.
 
     // Trigger the bootstrapper: this will trigger the sync task to start trying to fetch blocks
@@ -1492,10 +1493,7 @@ async fn bootstrap_e2e() {
     // understand though.
     let scraped_l1_handler_txs = vec![]; // No txs to scrape in this test.
     l1_provider.initialize(STARTUP_HEIGHT, scraped_l1_handler_txs).await.unwrap();
-    // TODO(guyn): this test assumes we start in bootstrapping state. The test should be updated to
-    // include the part where the batcher's first commit_block command is what determines the
-    // catchup height and causes the bootstrapping to begin.
-    l1_provider.start_bootstrapping(CATCH_UP_HEIGHT);
+    call_commit_block_to_start_bootstrapping_and_expect_error(&mut l1_provider, CATCH_UP_HEIGHT);
 
     // Load first **Sync** response: the initializer task will pick it up within the specified
     // interval.
@@ -1578,6 +1576,7 @@ async fn bootstrap_delayed_batcher_and_sync_state_with_trivial_catch_up() {
 
     let l1_provider_client = Arc::new(FakeL1ProviderClient::default());
     const STARTUP_HEIGHT: BlockNumber = BlockNumber(3);
+    const CATCH_UP_HEIGHT: BlockNumber = BlockNumber(5);
 
     let sync_client = MockStateSyncClient::default();
     let config = L1ProviderConfig {
@@ -1587,12 +1586,13 @@ async fn bootstrap_delayed_batcher_and_sync_state_with_trivial_catch_up() {
 
     let mut l1_provider =
         L1Provider::new(config, l1_provider_client.clone(), Arc::new(sync_client), None);
+
     // Test.
 
     // Start the sync sequence, should busy-wait until the batcher height is sent.
     let scraped_l1_handler_txs = []; // No txs to scrape in this test.
     l1_provider.initialize(STARTUP_HEIGHT, scraped_l1_handler_txs.into()).await.unwrap();
-
+    call_commit_block_to_start_bootstrapping_and_expect_error(&mut l1_provider, CATCH_UP_HEIGHT);
     // **Commit** a few blocks. The height starts from the provider's current height, since this
     // is a trivial catchup scenario (nothing to catch up).
     // This checks that the trivial catch_up_height doesn't mess up this flow.
@@ -1606,11 +1606,11 @@ async fn bootstrap_delayed_batcher_and_sync_state_with_trivial_catch_up() {
     // Commit blocks should have been applied.
     let start_height_plus_2 = height_add(STARTUP_HEIGHT, 2);
     assert_eq!(l1_provider.current_height, start_height_plus_2);
-    // TODO(guyn): it is possible that the rest of this test is trivial.
+
     // Should still be bootstrapping, since catchup height isn't determined yet.
     // Technically we could end bootstrapping at this point, but its simpler to let it
     // terminate gracefully once the batcher and sync are ready.
-    assert_eq!(l1_provider.state, ProviderState::Pending);
+    assert!(l1_provider.state.is_bootstrapping());
 
     // Let the sync task continue, it should short circuit.
     tokio::time::sleep(config.startup_sync_sleep_retry_interval_seconds).await;