add tests for shard backend responses

Kava-Labs · Mar 4, 2024 · 3c53de9 · 3c53de9
1 parent 5a9f4c4
commit 3c53de9
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/main_test.go b/main_test.go
@@ -428,6 +428,18 @@ func TestE2ETest_HeightBasedRouting(t *testing.T) {
 			params:      []interface{}{"0x15", false}, // block 21 is beyond shards
 			expectRoute: service.ResponseBackendDefault,
 		},
+		{
+			name:        "request for height in 1st shard -> shard",
+			method:      "eth_getBlockByNumber",
+			params:      []interface{}{"0x2", false}, // block 2
+			expectRoute: service.ResponseBackendShard,
+		},
+		{
+			name:        "request for height in 2nd shard -> shard",
+			method:      "eth_getBlockByNumber",
+			params:      []interface{}{"0xF", false}, // block 15
+			expectRoute: service.ResponseBackendShard,
+		},
 		{
 			name:        "request for earliest height -> default",
 			method:      "eth_getBlockByNumber",