Examples with TestClient com.linkedin.d2.balancer.clients.TrackerClientTest.TestClient used on opensource projects

Example 21 with TestClient

use of com.linkedin.d2.balancer.clients.TrackerClientTest.TestClient in project rest.li by linkedin.

the class RewriteLoadBalancerClientTest method testPathAppend.

@Test
public void testPathAppend() {
    URI uri = URI.create("http://test.linkedin.com:9876/test");
    String serviceName = "HistoryService";
    TestClient wrappedClient = new TestClient();
    RewriteLoadBalancerClient client = new RewriteLoadBalancerClient(serviceName, uri, wrappedClient);
    assertEquals(client.getUri(), uri);
    assertEquals(client.getServiceName(), serviceName);
    RestRequest restRequest;
    Map<String, String> restWireAttrs = new HashMap<>();
    TestTransportCallback<RestResponse> restCallback = new TestTransportCallback<>();
    restRequest = new RestRequestBuilder(URI.create("d2://HistoryService")).build();
    client.restRequest(restRequest, new RequestContext(), restWireAttrs, restCallback);
    checkRewrite(wrappedClient, restRequest, restCallback, "http://test.linkedin.com:9876/test");
    restRequest = new RestRequestBuilder(URI.create("d2://HistoryService/")).build();
    client.restRequest(restRequest, new RequestContext(), restWireAttrs, restCallback);
    checkRewrite(wrappedClient, restRequest, restCallback, "http://test.linkedin.com:9876/test/");
    restRequest = new RestRequestBuilder(URI.create("d2://HistoryService//")).build();
    client.restRequest(restRequest, new RequestContext(), restWireAttrs, restCallback);
    checkRewrite(wrappedClient, restRequest, restCallback, "http://test.linkedin.com:9876/test//");
    restRequest = new RestRequestBuilder(URI.create("d2://HistoryService/foo")).build();
    client.restRequest(restRequest, new RequestContext(), restWireAttrs, restCallback);
    checkRewrite(wrappedClient, restRequest, restCallback, "http://test.linkedin.com:9876/test/foo");
    restRequest = new RestRequestBuilder(URI.create("d2://HistoryService/foo/")).build();
    client.restRequest(restRequest, new RequestContext(), restWireAttrs, restCallback);
    checkRewrite(wrappedClient, restRequest, restCallback, "http://test.linkedin.com:9876/test/foo/");
}

Example 22 with TestClient

use of com.linkedin.d2.balancer.clients.TrackerClientTest.TestClient in project rest.li by linkedin.

the class RewriteLoadBalancerClientTest method testWithQueryAndFragment.

@Test
public void testWithQueryAndFragment() {
    URI uri = URI.create("http://test.linkedin.com/test");
    String serviceName = "HistoryService";
    TestClient wrappedClient = new TestClient();
    RewriteLoadBalancerClient client = new RewriteLoadBalancerClient(serviceName, uri, wrappedClient);
    assertEquals(client.getUri(), uri);
    assertEquals(client.getServiceName(), serviceName);
    RestRequest restRequest = new RestRequestBuilder(URI.create("d2://HistoryService/getCube?bar=baz#fragId")).build();
    Map<String, String> restWireAttrs = new HashMap<>();
    TestTransportCallback<RestResponse> restCallback = new TestTransportCallback<>();
    client.restRequest(restRequest, new RequestContext(), restWireAttrs, restCallback);
    assertFalse(restCallback.response.hasError());
    assertEquals(wrappedClient.restRequest.getHeaders(), restRequest.getHeaders());
    assertEquals(wrappedClient.restRequest.getEntity(), restRequest.getEntity());
    assertEquals(wrappedClient.restRequest.getMethod(), restRequest.getMethod());
    assertEquals(wrappedClient.restRequest.getURI(), URI.create("http://test.linkedin.com/test/getCube?bar=baz#fragId"));
}

Example 23 with TestClient

use of com.linkedin.d2.balancer.clients.TrackerClientTest.TestClient in project rest.li by linkedin.

the class RewriteLoadBalancerClientTest method testEscapingHelper.

private void testEscapingHelper(String hostUri, String serviceName, String path) {
    URI uri = URI.create(hostUri);
    TestClient wrappedClient = new TestClient();
    RewriteLoadBalancerClient client = new RewriteLoadBalancerClient(serviceName, uri, wrappedClient);
    assertEquals(client.getUri(), uri);
    assertEquals(client.getServiceName(), serviceName);
    RestRequest restRequest = new RestRequestBuilder(URI.create("d2://" + serviceName + path)).build();
    Map<String, String> restWireAttrs = new HashMap<>();
    TestTransportCallback<RestResponse> restCallback = new TestTransportCallback<>();
    client.restRequest(restRequest, new RequestContext(), restWireAttrs, restCallback);
    assertFalse(restCallback.response.hasError());
    assertEquals(wrappedClient.restRequest.getHeaders(), restRequest.getHeaders());
    assertEquals(wrappedClient.restRequest.getEntity(), restRequest.getEntity());
    assertEquals(wrappedClient.restRequest.getMethod(), restRequest.getMethod());
    assertEquals(wrappedClient.restRequest.getURI(), URI.create(hostUri + path));
}

Example 24 with TestClient

use of com.linkedin.d2.balancer.clients.TrackerClientTest.TestClient in project rest.li by linkedin.

the class LoadBalancerTestState method getClient.

@Override
public TrackerClient getClient(String clusterName, URI uri) {
    Map<Integer, PartitionData> partitionDataMap = new HashMap<>(2);
    partitionDataMap.put(DefaultPartitionAccessor.DEFAULT_PARTITION_ID, new PartitionData(1));
    return (getClient) ? new DegraderTrackerClientImpl(uri, partitionDataMap, new TestClient(), SystemClock.instance(), null) : null;
}

Example 25 with TestClient

use of com.linkedin.d2.balancer.clients.TrackerClientTest.TestClient in project rest.li by linkedin.

the class DegraderTrackerClientTest method testClientStreamRequest.

@Test(groups = { "small", "back-end" })
public void testClientStreamRequest() throws URISyntaxException {
    URI uri = URI.create("http://test.qa.com:1234/foo");
    double weight = 3d;
    TestClient wrappedClient = new TestClient(true);
    Clock clock = new SettableClock();
    Map<Integer, PartitionData> partitionDataMap = createDefaultPartitionData(3d);
    DegraderTrackerClient client = new DegraderTrackerClientImpl(uri, partitionDataMap, wrappedClient, clock, null);
    Assert.assertEquals(client.getUri(), uri);
    Double clientWeight = client.getPartitionWeight(DefaultPartitionAccessor.DEFAULT_PARTITION_ID);
    Assert.assertEquals(clientWeight, weight);
    Assert.assertEquals(client.getTransportClient(), wrappedClient);
    StreamRequest streamRequest = new StreamRequestBuilder(uri).build(EntityStreams.emptyStream());
    Map<String, String> restWireAttrs = new HashMap<>();
    TestTransportCallback<StreamResponse> restCallback = new TestTransportCallback<>();
    client.streamRequest(streamRequest, new RequestContext(), restWireAttrs, restCallback);
    Assert.assertFalse(restCallback.response.hasError());
    Assert.assertSame(wrappedClient.streamRequest, streamRequest);
    Assert.assertEquals(wrappedClient.restWireAttrs, restWireAttrs);
}