diff options
author | Harald Musum <musum@yahooinc.com> | 2022-01-03 10:51:30 +0100 |
---|---|---|
committer | Harald Musum <musum@yahooinc.com> | 2022-01-03 10:51:30 +0100 |
commit | 5c44b8d8bdb2956a0879767de4afe54d508e29a8 (patch) | |
tree | a2368126ebe3aca4c8e79f16fe99dc3d3788d8f8 /config/src/test/java | |
parent | 1e28cfcbeef8bb28f0a1c2dfb8877112fa89721b (diff) |
Simplify MockConnection
No need to use thread, start() isn't called anyway.
Collapse methods into one to simplify
Diffstat (limited to 'config/src/test/java')
-rw-r--r-- | config/src/test/java/com/yahoo/config/subscription/impl/JRTConfigRequesterTest.java | 16 | ||||
-rw-r--r-- | config/src/test/java/com/yahoo/vespa/config/protocol/JRTConfigRequestV3Test.java | 2 |
2 files changed, 9 insertions, 9 deletions
diff --git a/config/src/test/java/com/yahoo/config/subscription/impl/JRTConfigRequesterTest.java b/config/src/test/java/com/yahoo/config/subscription/impl/JRTConfigRequesterTest.java index dca0c2d0018..07c3a5af56c 100644 --- a/config/src/test/java/com/yahoo/config/subscription/impl/JRTConfigRequesterTest.java +++ b/config/src/test/java/com/yahoo/config/subscription/impl/JRTConfigRequesterTest.java @@ -4,6 +4,7 @@ package com.yahoo.config.subscription.impl; import com.yahoo.config.subscription.ConfigSourceSet; import com.yahoo.foo.SimpletypesConfig; import com.yahoo.jrt.Request; +import com.yahoo.jrt.RequestWaiter; import com.yahoo.vespa.config.ConfigKey; import com.yahoo.vespa.config.ConnectionPool; import com.yahoo.vespa.config.ErrorCode; @@ -209,10 +210,10 @@ public class JRTConfigRequesterTest { } @Override - public void run() { + public void handle(Request request, RequestWaiter requestWaiter) { System.out.println("Running error response handler"); - request().setError(errorCode, "error"); - requestWaiter().handleRequestDone(request()); + request.setError(errorCode, "error"); + requestWaiter.handleRequestDone(request); } } @@ -224,16 +225,15 @@ public class JRTConfigRequesterTest { } @Override - public void run() { - System.out.println("Running delayed response handler (waiting " + waitTimeMilliSeconds + - ") before responding"); + public void handle(Request request, RequestWaiter requestWaiter) { + System.out.println("Running delayed response handler (waiting " + waitTimeMilliSeconds + ") before responding"); try { Thread.sleep(waitTimeMilliSeconds); } catch (InterruptedException e) { e.printStackTrace(); } - request().setError(com.yahoo.jrt.ErrorCode.TIMEOUT, "error"); - requestWaiter().handleRequestDone(request()); + request.setError(com.yahoo.jrt.ErrorCode.TIMEOUT, "error"); + requestWaiter.handleRequestDone(request); } } diff --git a/config/src/test/java/com/yahoo/vespa/config/protocol/JRTConfigRequestV3Test.java b/config/src/test/java/com/yahoo/vespa/config/protocol/JRTConfigRequestV3Test.java index 8d160076db9..b3285e5157a 100644 --- a/config/src/test/java/com/yahoo/vespa/config/protocol/JRTConfigRequestV3Test.java +++ b/config/src/test/java/com/yahoo/vespa/config/protocol/JRTConfigRequestV3Test.java @@ -203,7 +203,7 @@ public class JRTConfigRequestV3Test { public void created_from_existing_subscription() { MockConnection connection = new MockConnection(new MockConnection.AbstractResponseHandler() { @Override - public void createResponse() { + public void createResponse(Request request) { JRTServerConfigRequest serverRequest = createReq(request); serverRequest.addOkResponse(createPayload(), currentGeneration, false, payloadChecksums); } |