solution: integration test with two upstreams

testing/README.adoc

@@ -18,6 +18,11 @@ NOTE: Run commands in project's root dir
 cd testing/simple-upstream && ./gradlew run
 ----
 
+[source,bash]
+----
+cd testing/simple-upstream && DSHACKLE_TESTUP_PORT=18546 ./gradlew run
+----
+
 === Run Dshackle
 
 [source,bash]

testing/dshackle/dshackle.yaml

@@ -18,6 +18,14 @@ cluster:
         ethereum:
           rpc:
             url: "http://localhost:18545"
+    - id: test-2
+      chain: ethereum
+      options:
+        disable-validation: true
+      connection:
+        ethereum:
+          rpc:
+            url: "http://localhost:18546"
 
 cache:
   redis:

testing/simple-upstream/src/main/groovy/testing/InternalHandler.groovy

@@ -0,0 +1,35 @@
+package testing
+
+import java.time.Instant
+import java.util.concurrent.atomic.AtomicInteger
+
+class InternalHandler implements CallHandler {
+
+    private AtomicInteger ids = new AtomicInteger()
+    private List<Item> requests = new ArrayList()
+
+    def record(String json) {
+        requests << new Item(ids.getAndIncrement(), json)
+    }
+
+    @Override
+    Result handle(String method, List<Object> params) {
+        if (method == "eth_call"
+                && params[0].to?.toLowerCase() == "0x0123456789abcdef0123456789abcdef00000002".toLowerCase()) {
+            return Result.ok(Collections.unmodifiableList(requests))
+        }
+        return null
+    }
+
+    class Item {
+        Integer id
+        String timestamp
+        String json
+
+        Item(Integer id, String json) {
+            this.id = id
+            this.timestamp = Instant.now().toString()
+            this.json = json
+        }
+    }
+}

testing/simple-upstream/src/main/groovy/testing/PingPongHandler.groovy

@@ -0,0 +1,12 @@
+package testing
+
+class PingPongHandler implements CallHandler {
+    @Override
+    Result handle(String method, List<Object> params) {
+        if (method == "eth_call"
+                && params[0].to?.toLowerCase() == "0x0123456789abcdef0123456789abcdef00000001".toLowerCase()) {
+            return Result.ok([data: params[0].data])
+        }
+        return null
+    }
+}

testing/simple-upstream/src/main/groovy/testing/SimpleUpstream.groovy

@@ -5,17 +5,23 @@ import spark.Spark
 
 class SimpleUpstream {
 
-    private int port = 18545
+    private int port = System.getenv("DSHACKLE_TESTUP_PORT")?.toInteger() ?: 18545
     private ObjectMapper objectMapper
 
     private List<CallHandler> handlers = []
+    private InternalHandler internalHandler
 
     void prepare() {
         objectMapper = new ObjectMapper()
 
+        internalHandler = new InternalHandler()
+
         handlers << new TestcaseHandler(objectMapper)
         handlers << new CommonHandlers()
         handlers << new BlocksHandler(objectMapper)
+        handlers << new PingPongHandler()
+        handlers << internalHandler
+
         handlers << new InvalidCallHandler()
     }
 
@@ -23,9 +29,11 @@ class SimpleUpstream {
         println("Starting upstream on 0.0.0.0:$port")
         Spark.port(port)
         Spark.post("/") { req, resp ->
-            println("request")
             try {
-                Map json = objectMapper.readValue(req.body(), Map)
+                def requestBody = req.body()
+                println("request: $requestBody")
+                internalHandler.record(requestBody)
+                Map json = objectMapper.readValue(requestBody, Map)
                 def id = json["id"]
                 String method = json["method"].toString()
                 List<Object> params = json.containsKey("params") ? json["params"] as List<Object> : []

testing/trial/src/main/groovy/io/emeraldpay/dshackle/testing/trial/ProxyClient.groovy

@@ -28,6 +28,10 @@ class ProxyClient {
         return new ProxyClient("http://127.0.0.1:18080/$prefix")
     }
 
+    static ProxyClient forOriginal(int port) {
+        return new ProxyClient("http://127.0.0.1:$port")
+    }
+
     Map<String, Object> execute(String method, List<Object> params) {
         return execute(sequence++, method, params)
     }

testing/trial/src/test/groovy/io/emeraldpay/dshackle/testing/trial/proxy/DispatchSpec.groovy

@@ -0,0 +1,40 @@
+package io.emeraldpay.dshackle.testing.trial.proxy
+
+import io.emeraldpay.dshackle.testing.trial.ProxyClient
+import spock.lang.Specification
+
+class DispatchSpec extends Specification {
+
+    def client = ProxyClient.forPrefix("eth")
+
+    def "multiple calls routed roughly equal to upstreams"() {
+        when:
+        def calls1before = ProxyClient.forOriginal(18545).execute("eth_call", [[to: "0x0123456789abcdef0123456789abcdef00000002"]]).result as List
+        def calls2before = ProxyClient.forOriginal(18546).execute("eth_call", [[to: "0x0123456789abcdef0123456789abcdef00000002"]]).result as List
+
+        100.times {
+            client.execute("eth_call", [[to: "0x0123456789abcdef0123456789abcdef00000001", data: "0x00000000" + Integer.toString(it, 16)]])
+        }
+
+        def calls1after = ProxyClient.forOriginal(18545).execute("eth_call", [[to: "0x0123456789abcdef0123456789abcdef00000002"]]).result as List
+        def calls2after = ProxyClient.forOriginal(18546).execute("eth_call", [[to: "0x0123456789abcdef0123456789abcdef00000002"]]).result as List
+
+        def calls1 = onlyNew(calls1before, calls1after)
+        def calls2 = onlyNew(calls2before, calls2after)
+
+        then:
+        calls1.size() >= 48
+        calls2.size() >= 48
+        calls1.size() + calls2.size() == 100
+    }
+
+    private List onlyNew(List before, List after) {
+        return after.findAll { a ->
+            !before.any { b ->
+                b.id == a.id
+            }
+        }.findAll {
+            (it.json as String).contains("0x0123456789abcdef0123456789abcdef00000001")
+        }
+    }
+}