aboutsummaryrefslogtreecommitdiffstats
path: root/config/src/test/java
diff options
context:
space:
mode:
authorHarald Musum <musum@yahooinc.com>2022-01-03 10:51:30 +0100
committerHarald Musum <musum@yahooinc.com>2022-01-03 10:51:30 +0100
commit5c44b8d8bdb2956a0879767de4afe54d508e29a8 (patch)
treea2368126ebe3aca4c8e79f16fe99dc3d3788d8f8 /config/src/test/java
parent1e28cfcbeef8bb28f0a1c2dfb8877112fa89721b (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.java16
-rw-r--r--config/src/test/java/com/yahoo/vespa/config/protocol/JRTConfigRequestV3Test.java2
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);
}