diff --git a/main_test.go b/main_test.go index e0f3f5d..f0f7bb8 100644 --- 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",