|
1 | | -use alloy::{eips::BlockId, network::Network}; |
| 1 | +use alloy::{ |
| 2 | + consensus::BlockHeader, |
| 3 | + eips::BlockId, |
| 4 | + network::{BlockResponse, Network}, |
| 5 | +}; |
2 | 6 |
|
3 | 7 | use super::common::{ConsumerMode, handle_stream}; |
4 | 8 | use crate::{ |
@@ -45,13 +49,28 @@ impl EventScannerBuilder<LatestEvents> { |
45 | 49 | let scanner = self.build(provider).await?; |
46 | 50 |
|
47 | 51 | let provider = scanner.block_range_scanner.provider(); |
| 52 | + let latest_block = provider.get_block_number().await?; |
| 53 | + |
| 54 | + let from_num = match scanner.config.from_block { |
| 55 | + BlockId::Number(from_block) => from_block.as_number().unwrap_or(0), |
| 56 | + BlockId::Hash(from_hash) => { |
| 57 | + provider.get_block_by_hash(from_hash.into()).await?.header().number() |
| 58 | + } |
| 59 | + }; |
48 | 60 |
|
49 | | - if let BlockId::Hash(from_hash) = scanner.config.from_block { |
50 | | - provider.get_block_by_hash(from_hash.into()).await?; |
| 61 | + if from_num > latest_block { |
| 62 | + Err(ScannerError::BlockExceedsLatest("from_block", from_num, latest_block))?; |
51 | 63 | } |
52 | 64 |
|
53 | | - if let BlockId::Hash(to_hash) = scanner.config.to_block { |
54 | | - provider.get_block_by_hash(to_hash.into()).await?; |
| 65 | + let to_num = match scanner.config.to_block { |
| 66 | + BlockId::Number(to_block) => to_block.as_number().unwrap_or(0), |
| 67 | + BlockId::Hash(to_hash) => { |
| 68 | + provider.get_block_by_hash(to_hash.into()).await?.header().number() |
| 69 | + } |
| 70 | + }; |
| 71 | + |
| 72 | + if to_num > latest_block { |
| 73 | + Err(ScannerError::BlockExceedsLatest("to_block", to_num, latest_block))?; |
55 | 74 | } |
56 | 75 |
|
57 | 76 | Ok(scanner) |
@@ -281,4 +300,70 @@ mod tests { |
281 | 300 |
|
282 | 301 | assert!(result.is_ok()); |
283 | 302 | } |
| 303 | + |
| 304 | + #[tokio::test] |
| 305 | + async fn test_from_block_above_latest_returns_error() { |
| 306 | + let anvil = Anvil::new().try_spawn().unwrap(); |
| 307 | + let provider = ProviderBuilder::new().connect_http(anvil.endpoint_url()); |
| 308 | + |
| 309 | + let latest_block = provider.get_block_number().await.unwrap(); |
| 310 | + |
| 311 | + let result = EventScannerBuilder::latest(1) |
| 312 | + .from_block(latest_block + 100) |
| 313 | + .to_block(latest_block) |
| 314 | + .connect(provider) |
| 315 | + .await; |
| 316 | + |
| 317 | + match result { |
| 318 | + Err(ScannerError::BlockExceedsLatest("from_block", max, latest)) => { |
| 319 | + assert_eq!(max, latest_block + 100); |
| 320 | + assert_eq!(latest, latest_block); |
| 321 | + } |
| 322 | + _ => panic!("Expected BlockExceedsLatest error"), |
| 323 | + } |
| 324 | + } |
| 325 | + |
| 326 | + #[tokio::test] |
| 327 | + async fn test_to_block_above_latest_returns_error() { |
| 328 | + let anvil = Anvil::new().try_spawn().unwrap(); |
| 329 | + let provider = ProviderBuilder::new().connect_http(anvil.endpoint_url()); |
| 330 | + |
| 331 | + let latest_block = provider.get_block_number().await.unwrap(); |
| 332 | + |
| 333 | + let result = EventScannerBuilder::latest(1) |
| 334 | + .from_block(0) |
| 335 | + .to_block(latest_block + 100) |
| 336 | + .connect(provider) |
| 337 | + .await; |
| 338 | + |
| 339 | + match result { |
| 340 | + Err(ScannerError::BlockExceedsLatest("to_block", max, latest)) => { |
| 341 | + assert_eq!(max, latest_block + 100); |
| 342 | + assert_eq!(latest, latest_block); |
| 343 | + } |
| 344 | + _ => panic!("Expected BlockExceedsLatest error"), |
| 345 | + } |
| 346 | + } |
| 347 | + |
| 348 | + #[tokio::test] |
| 349 | + async fn test_to_and_from_block_above_latest_returns_error() { |
| 350 | + let anvil = Anvil::new().try_spawn().unwrap(); |
| 351 | + let provider = ProviderBuilder::new().connect_http(anvil.endpoint_url()); |
| 352 | + |
| 353 | + let latest_block = provider.get_block_number().await.unwrap(); |
| 354 | + |
| 355 | + let result = EventScannerBuilder::latest(1) |
| 356 | + .from_block(latest_block + 50) |
| 357 | + .to_block(latest_block + 100) |
| 358 | + .connect(provider) |
| 359 | + .await; |
| 360 | + |
| 361 | + match result { |
| 362 | + Err(ScannerError::BlockExceedsLatest("from_block", max, latest)) => { |
| 363 | + assert_eq!(max, latest_block + 50); |
| 364 | + assert_eq!(latest, latest_block); |
| 365 | + } |
| 366 | + _ => panic!("Expected BlockExceedsLatest error for 'from_block'"), |
| 367 | + } |
| 368 | + } |
284 | 369 | } |
0 commit comments