modules.add(new TestingDiscoveryModule());
@BeforeMethod public void setup() { testHttpClient = new TestingHttpClient(input -> new TestingResponse(OK, ArrayListMultimap.create(), ACTIVE.name().getBytes())); expectedVersion = new NodeVersion("1"); coordinator = new PrestoNode(UUID.randomUUID().toString(), URI.create("https://192.0.2.8"), expectedVersion, false); activeNodes = ImmutableList.of( new PrestoNode(nodeInfo.getNodeId(), URI.create("http://192.0.1.1"), expectedVersion, false), new PrestoNode(UUID.randomUUID().toString(), URI.create("http://192.0.2.1:8080"), expectedVersion, false), new PrestoNode(UUID.randomUUID().toString(), URI.create("http://192.0.2.3"), expectedVersion, false), coordinator); inactiveNodes = ImmutableList.of( new PrestoNode(UUID.randomUUID().toString(), URI.create("https://192.0.3.9"), NodeVersion.UNKNOWN, false), new PrestoNode(UUID.randomUUID().toString(), URI.create("https://192.0.4.9"), new NodeVersion("2"), false)); List<ServiceDescriptor> descriptors = new ArrayList<>(); for (PrestoNode node : Iterables.concat(activeNodes, inactiveNodes)) { descriptors.add(serviceDescriptor("presto") .setNodeId(node.getNodeIdentifier()) .addProperty("http", node.getHttpUri().toString()) .addProperty("node_version", node.getNodeVersion().toString()) .addProperty("coordinator", String.valueOf(node.equals(coordinator))) .build()); } selector = new StaticServiceSelector(descriptors); }
@Override public ServiceSelector createServiceSelector(String type, ServiceSelectorConfig selectorConfig) { requireNonNull(type, "type is null"); requireNonNull(selectorConfig, "selectorConfig is null"); return new SimpleServiceSelector(type, selectorConfig, lookupClient); } }
@Override public ListenableFuture<ServiceDescriptors> refreshServices(ServiceDescriptors serviceDescriptors) { requireNonNull(serviceDescriptors, "serviceDescriptors is null"); return getServices(serviceDescriptors.getType(), serviceDescriptors.getPool()); } }
@Test public void testBasics() { CachingServiceSelector serviceSelector = new CachingServiceSelector("type", new ServiceSelectorConfig().setPool("pool"), new InMemoryDiscoveryClient(nodeInfo), executor); Assert.assertEquals(serviceSelector.getType(), "type"); Assert.assertEquals(serviceSelector.getPool(), "pool"); }
private HttpEventClient newEventClient(List<URI> uris) { HttpServiceSelector selector = new StaticHttpServiceSelector("collector", "general", uris); Set<EventTypeMetadata<?>> eventTypes = getValidEventTypeMetaDataSet(FixedDummyEventClass.class); JsonEventWriter eventWriter = new JsonEventWriter(eventTypes); return new HttpEventClient( selector, eventWriter, new NodeInfo("test"), httpClient); }
@Override public List<ServiceDescriptor> selectAllServices() { try { ListenableFuture<ServiceDescriptors> future = lookupClient.getServices(type, pool); ServiceDescriptors serviceDescriptors = getFutureResult(future, DiscoveryException.class); return serviceDescriptors.getServiceDescriptors(); } catch (DiscoveryException e) { log.error(e); return ImmutableList.of(); } }
new TestingNodeModule(), new TestingJmxModule(), new TestingDiscoveryModule(), new TestingHttpServerModule(), new TraceTokenModule(),
@Override public CheckedFuture<ServiceDescriptors, DiscoveryException> refreshServices(ServiceDescriptors serviceDescriptors) { Preconditions.checkNotNull(serviceDescriptors, "serviceDescriptors is null"); return getServices(serviceDescriptors.getType(), serviceDescriptors.getPool()); } }
@Override public ServiceSelector createServiceSelector(String type, ServiceSelectorConfig selectorConfig) { Preconditions.checkNotNull(type, "type is null"); Preconditions.checkNotNull(selectorConfig, "selectorConfig is null"); return new SimpleServiceSelector(type, selectorConfig, lookupClient); } }
@Test public void testBasics() { CachingServiceSelector serviceSelector = new CachingServiceSelector("type", new ServiceSelectorConfig().setPool("pool"), new InMemoryDiscoveryClient(nodeInfo), executor); assertEquals(serviceSelector.getType(), "type"); assertEquals(serviceSelector.getPool(), "pool"); }
private HttpEventClient newEventClient(List<URI> uris) { HttpServiceSelector selector = new StaticHttpServiceSelector("collector", "general", uris); Set<EventTypeMetadata<?>> eventTypes = getValidEventTypeMetaDataSet(FixedDummyEventClass.class); JsonEventWriter eventWriter = new JsonEventWriter(eventTypes); return new HttpEventClient( selector, eventWriter, new NodeInfo("test"), httpClient); }
protected TestTestingDiscoveryModule() { super(new TestingDiscoveryModule()); } }
@Override public CheckedFuture<ServiceDescriptors, DiscoveryException> refreshServices(ServiceDescriptors serviceDescriptors) { Preconditions.checkNotNull(serviceDescriptors, "serviceDescriptors is null"); return getServices(serviceDescriptors.getType(), serviceDescriptors.getPool()); } }
@Override public ServiceSelector createServiceSelector(String type, ServiceSelectorConfig selectorConfig) { Preconditions.checkNotNull(type, "type is null"); Preconditions.checkNotNull(selectorConfig, "selectorConfig is null"); return new SimpleServiceSelector(type, selectorConfig, lookupClient); } }
@Test public void testBasics() { CachingServiceSelector serviceSelector = new CachingServiceSelector("type", new ServiceSelectorConfig().setPool("pool"), new InMemoryDiscoveryClient(nodeInfo), executor); assertEquals(serviceSelector.getType(), "type"); assertEquals(serviceSelector.getPool(), "pool"); }
protected TestTestingDiscoveryModule() { super(new TestingDiscoveryModule()); } }
@Override public void configure(Binder binder) { binder.install(new ConfigurationModule(new ConfigurationFactory(configProperties))); binder.install(new TestingNodeModule()); binder.install(new TestingDiscoveryModule()); } }
@Override public void configure(Binder binder) { binder.install(new ConfigurationModule(new ConfigurationFactory(configProperties))); binder.install(new TestingNodeModule()); binder.install(new TestingDiscoveryModule()); } }
@Override public void configure(Binder binder) { binder.install(new ConfigurationModule(new ConfigurationFactory(configProperties))); binder.install(new TestingNodeModule()); binder.install(new TestingDiscoveryModule()); } }