Search in sources :

Example 1 with OK

use of com.facebook.airlift.http.client.HttpStatus.OK in project presto by prestodb.

the class TestDiscoveryNodeManager method setup.

@BeforeMethod
public void setup() {
    testHttpClient = new TestingHttpClient(input -> new TestingResponse(OK, ArrayListMultimap.create(), ACTIVE.name().getBytes()));
    expectedVersion = new NodeVersion("1");
    coordinator = new InternalNode(UUID.randomUUID().toString(), URI.create("https://192.0.2.8"), expectedVersion, true);
    resourceManager = new InternalNode(UUID.randomUUID().toString(), URI.create("https://192.0.2.9"), expectedVersion, false, true);
    currentNode = new InternalNode(nodeInfo.getNodeId(), URI.create("http://192.0.1.1"), expectedVersion, false);
    activeNodes = ImmutableSet.of(currentNode, new InternalNode(UUID.randomUUID().toString(), URI.create("http://192.0.2.1:8080"), expectedVersion, false), new InternalNode(UUID.randomUUID().toString(), URI.create("http://192.0.2.3"), expectedVersion, false), coordinator, resourceManager);
    inactiveNodes = ImmutableSet.of(new InternalNode(UUID.randomUUID().toString(), URI.create("https://192.0.3.9"), NodeVersion.UNKNOWN, false), new InternalNode(UUID.randomUUID().toString(), URI.create("https://192.0.4.9"), new NodeVersion("2"), false));
    selector.announceNodes(activeNodes, inactiveNodes);
}
Also used : ArrayListMultimap(com.google.common.collect.ArrayListMultimap) Iterables(com.google.common.collect.Iterables) ServiceDescriptor(com.facebook.airlift.discovery.client.ServiceDescriptor) ListenableFuture(com.google.common.util.concurrent.ListenableFuture) TestingDriftClient(com.facebook.presto.operator.TestingDriftClient) NodeInfo(com.facebook.airlift.node.NodeInfo) Assert.assertEquals(org.testng.Assert.assertEquals) Test(org.testng.annotations.Test) NodeVersion(com.facebook.presto.client.NodeVersion) ACTIVE(com.facebook.presto.spi.NodeState.ACTIVE) DEFAULT_POOL(com.facebook.airlift.discovery.client.ServiceSelectorConfig.DEFAULT_POOL) ImmutableList(com.google.common.collect.ImmutableList) Assert.assertNotSame(org.testng.Assert.assertNotSame) TestingHttpClient(com.facebook.airlift.http.client.testing.TestingHttpClient) OK(com.facebook.airlift.http.client.HttpStatus.OK) TestingResponse(com.facebook.airlift.http.client.testing.TestingResponse) ServiceSelector(com.facebook.airlift.discovery.client.ServiceSelector) URI(java.net.URI) ImmutableSet(com.google.common.collect.ImmutableSet) Assertions.assertEqualsIgnoreOrder(com.facebook.airlift.testing.Assertions.assertEqualsIgnoreOrder) InternalCommunicationConfig(com.facebook.presto.server.InternalCommunicationConfig) ServiceDescriptor.serviceDescriptor(com.facebook.airlift.discovery.client.ServiceDescriptor.serviceDescriptor) BeforeMethod(org.testng.annotations.BeforeMethod) Set(java.util.Set) BlockingQueue(java.util.concurrent.BlockingQueue) UUID(java.util.UUID) GuardedBy(javax.annotation.concurrent.GuardedBy) HttpClient(com.facebook.airlift.http.client.HttpClient) NodeConfig(com.facebook.airlift.node.NodeConfig) ArrayBlockingQueue(java.util.concurrent.ArrayBlockingQueue) List(java.util.List) NoOpFailureDetector(com.facebook.presto.failureDetector.NoOpFailureDetector) Optional(java.util.Optional) INACTIVE(com.facebook.presto.spi.NodeState.INACTIVE) NodeVersion(com.facebook.presto.client.NodeVersion) TestingResponse(com.facebook.airlift.http.client.testing.TestingResponse) TestingHttpClient(com.facebook.airlift.http.client.testing.TestingHttpClient) BeforeMethod(org.testng.annotations.BeforeMethod)

Aggregations

ServiceDescriptor (com.facebook.airlift.discovery.client.ServiceDescriptor)1 ServiceDescriptor.serviceDescriptor (com.facebook.airlift.discovery.client.ServiceDescriptor.serviceDescriptor)1 ServiceSelector (com.facebook.airlift.discovery.client.ServiceSelector)1 DEFAULT_POOL (com.facebook.airlift.discovery.client.ServiceSelectorConfig.DEFAULT_POOL)1 HttpClient (com.facebook.airlift.http.client.HttpClient)1 OK (com.facebook.airlift.http.client.HttpStatus.OK)1 TestingHttpClient (com.facebook.airlift.http.client.testing.TestingHttpClient)1 TestingResponse (com.facebook.airlift.http.client.testing.TestingResponse)1 NodeConfig (com.facebook.airlift.node.NodeConfig)1 NodeInfo (com.facebook.airlift.node.NodeInfo)1 Assertions.assertEqualsIgnoreOrder (com.facebook.airlift.testing.Assertions.assertEqualsIgnoreOrder)1 NodeVersion (com.facebook.presto.client.NodeVersion)1 NoOpFailureDetector (com.facebook.presto.failureDetector.NoOpFailureDetector)1 TestingDriftClient (com.facebook.presto.operator.TestingDriftClient)1 InternalCommunicationConfig (com.facebook.presto.server.InternalCommunicationConfig)1 ACTIVE (com.facebook.presto.spi.NodeState.ACTIVE)1 INACTIVE (com.facebook.presto.spi.NodeState.INACTIVE)1 ArrayListMultimap (com.google.common.collect.ArrayListMultimap)1 ImmutableList (com.google.common.collect.ImmutableList)1 ImmutableSet (com.google.common.collect.ImmutableSet)1