11/// Unit tests for the SimulationPublisher implementation
22mod common;
33
4- use common:: builders:: * ;
54use alloy_primitives:: { Address , B256 , U256 } ;
5+ use common:: builders:: * ;
66use std:: collections:: HashMap ;
77
88// These tests focus on the logic that can be tested without requiring
@@ -12,16 +12,16 @@ use std::collections::HashMap;
1212async fn test_state_diff_conversion_logic ( ) {
1313 // Test the state diff conversion logic that TipsSimulationPublisher uses
1414 let mut original_state_diff = HashMap :: new ( ) ;
15-
15+
1616 // Create test data with multiple accounts and storage slots
1717 for i in 0 ..3 {
1818 let addr = Address :: random ( ) ;
1919 let mut storage = HashMap :: new ( ) ;
20-
20+
2121 for j in 0 ..5 {
2222 storage. insert ( U256 :: from ( i * 10 + j) , U256 :: from ( ( i + 1 ) * 100 + j) ) ;
2323 }
24-
24+
2525 original_state_diff. insert ( addr, storage) ;
2626 }
2727
@@ -39,12 +39,12 @@ async fn test_state_diff_conversion_logic() {
3939
4040 // Verify conversion
4141 assert_eq ! ( converted. len( ) , original_state_diff. len( ) ) ;
42-
42+
4343 for ( address, original_storage) in & original_state_diff {
4444 assert ! ( converted. contains_key( address) ) ;
4545 let converted_storage = & converted[ address] ;
4646 assert_eq ! ( converted_storage. len( ) , original_storage. len( ) ) ;
47-
47+
4848 for ( key, value) in original_storage {
4949 let key_bytes = key. to_be_bytes :: < 32 > ( ) ;
5050 let storage_key = B256 :: from ( key_bytes) ;
@@ -53,21 +53,20 @@ async fn test_state_diff_conversion_logic() {
5353 }
5454}
5555
56-
5756#[ test]
5857fn test_large_state_diff_handling ( ) {
5958 // Test handling of large state diffs
6059 let mut large_state_diff = HashMap :: new ( ) ;
61-
60+
6261 // Create a large state diff with many accounts and storage slots
6362 for i in 0 ..100 {
6463 let addr = Address :: random ( ) ;
6564 let mut storage = HashMap :: new ( ) ;
66-
65+
6766 for j in 0 ..50 {
6867 storage. insert ( U256 :: from ( i * 1000 + j) , U256 :: from ( j * 12345 ) ) ;
6968 }
70-
69+
7170 large_state_diff. insert ( addr, storage) ;
7271 }
7372
@@ -90,8 +89,6 @@ fn test_large_state_diff_handling() {
9089 }
9190}
9291
93-
94-
9592#[ test]
9693fn test_execution_time_bounds ( ) {
9794 // Test execution time edge cases
@@ -106,26 +103,28 @@ fn test_execution_time_bounds() {
106103 let result = SimulationResultBuilder :: successful ( )
107104 . with_execution_time_us ( execution_time)
108105 . build ( ) ;
109-
110- assert_eq ! ( result. execution_time_us, execution_time, "Failed for: {}" , description) ;
106+
107+ assert_eq ! (
108+ result. execution_time_us, execution_time,
109+ "Failed for: {}" ,
110+ description
111+ ) ;
111112 }
112113}
113114
114-
115-
116115#[ test]
117116fn test_multiple_addresses_same_storage ( ) {
118117 // Test multiple addresses with the same storage patterns
119118 let addresses = vec ! [ Address :: random( ) , Address :: random( ) , Address :: random( ) ] ;
120119 let mut state_diff = HashMap :: new ( ) ;
121-
120+
122121 for addr in & addresses {
123122 let mut storage = HashMap :: new ( ) ;
124123 storage. insert ( U256 :: from ( 1 ) , U256 :: from ( 100 ) ) ;
125124 storage. insert ( U256 :: from ( 2 ) , U256 :: from ( 200 ) ) ;
126125 state_diff. insert ( * addr, storage) ;
127126 }
128-
127+
129128 // Convert
130129 let mut converted = HashMap :: new ( ) ;
131130 for ( address, storage) in & state_diff {
@@ -137,7 +136,7 @@ fn test_multiple_addresses_same_storage() {
137136 }
138137 converted. insert ( * address, storage_map) ;
139138 }
140-
139+
141140 assert_eq ! ( converted. len( ) , 3 ) ;
142141 for addr in & addresses {
143142 assert ! ( converted. contains_key( addr) ) ;
0 commit comments