Examples with MetadataManager com.datastax.oss.driver.internal.core.metadata.MetadataManager used on opensource projects

Example 1 with MetadataManager

use of com.datastax.oss.driver.internal.core.metadata.MetadataManager in project java-driver by datastax.

the class GraphSupportCheckerTest method addNodeWithDseVersion.

private void addNodeWithDseVersion(InternalDriverContext context, List<Version> dseVersions) {
    MetadataManager manager = mock(MetadataManager.class);
    when(context.getMetadataManager()).thenReturn(manager);
    Metadata metadata = mock(Metadata.class);
    when(manager.getMetadata()).thenReturn(metadata);
    Map<UUID, Node> nodes = new HashMap<>();
    for (Version v : dseVersions) {
        Node node = mock(Node.class);
        Map<String, Object> extras = new HashMap<>();
        extras.put(DseNodeProperties.DSE_VERSION, v);
        when(node.getExtras()).thenReturn(extras);
        nodes.put(UUID.randomUUID(), node);
    }
    when(metadata.getNodes()).thenReturn(nodes);
}

Example 2 with MetadataManager

use of com.datastax.oss.driver.internal.core.metadata.MetadataManager in project java-driver by datastax.

the class InsightsClientTest method mockDefaultDriverContext.

private DefaultDriverContext mockDefaultDriverContext() throws UnknownHostException {
    DefaultDriverContext context = mock(DefaultDriverContext.class);
    mockConnectionPools(context);
    MetadataManager manager = mock(MetadataManager.class);
    when(context.getMetadataManager()).thenReturn(manager);
    Metadata metadata = mock(Metadata.class);
    when(manager.getMetadata()).thenReturn(metadata);
    Node node = mock(Node.class);
    when(node.getExtras()).thenReturn(ImmutableMap.of(DseNodeProperties.DSE_VERSION, Objects.requireNonNull(Version.parse("6.0.5"))));
    when(metadata.getNodes()).thenReturn(ImmutableMap.of(UUID.randomUUID(), node));
    DriverExecutionProfile defaultExecutionProfile = mockDefaultExecutionProfile();
    DriverExecutionProfile nonDefaultExecutionProfile = mockNonDefaultRequestTimeoutExecutionProfile();
    Map<String, String> startupOptions = new HashMap<>();
    startupOptions.put(StartupOptionsBuilder.CLIENT_ID_KEY, "client-id");
    startupOptions.put(StartupOptionsBuilder.APPLICATION_VERSION_KEY, "1.0.0");
    startupOptions.put(StartupOptionsBuilder.APPLICATION_NAME_KEY, "app-name");
    startupOptions.put(StartupOptionsBuilder.DRIVER_VERSION_KEY, "2.x");
    startupOptions.put(StartupOptionsBuilder.DRIVER_NAME_KEY, "DataStax Enterprise Java Driver");
    when(context.getStartupOptions()).thenReturn(startupOptions);
    when(context.getProtocolVersion()).thenReturn(DSE_V2);
    DefaultNode contactPoint = mock(DefaultNode.class);
    EndPoint contactEndPoint = mock(EndPoint.class);
    when(contactEndPoint.resolve()).thenReturn(new InetSocketAddress("127.0.0.1", 9999));
    when(contactPoint.getEndPoint()).thenReturn(contactEndPoint);
    when(manager.getContactPoints()).thenReturn(ImmutableSet.of(contactPoint));
    DriverConfig driverConfig = mock(DriverConfig.class);
    when(context.getConfig()).thenReturn(driverConfig);
    Map<String, DriverExecutionProfile> profiles = ImmutableMap.of("default", defaultExecutionProfile, "non-default", nonDefaultExecutionProfile);
    Mockito.<Map<String, ? extends DriverExecutionProfile>>when(driverConfig.getProfiles()).thenReturn(profiles);
    when(driverConfig.getDefaultProfile()).thenReturn(defaultExecutionProfile);
    ControlConnection controlConnection = mock(ControlConnection.class);
    DriverChannel channel = mock(DriverChannel.class);
    EndPoint controlConnectionEndpoint = mock(EndPoint.class);
    when(controlConnectionEndpoint.resolve()).thenReturn(new InetSocketAddress("127.0.0.1", 10));
    when(channel.getEndPoint()).thenReturn(controlConnectionEndpoint);
    when(channel.localAddress()).thenReturn(new InetSocketAddress("127.0.0.1", 10));
    when(controlConnection.channel()).thenReturn(channel);
    when(context.getControlConnection()).thenReturn(controlConnection);
    return context;
}

Example 3 with MetadataManager

use of com.datastax.oss.driver.internal.core.metadata.MetadataManager in project java-driver by datastax.

the class DseTestFixtures method mockNodesInMetadataWithVersions.

public static DefaultDriverContext mockNodesInMetadataWithVersions(DefaultDriverContext mockContext, boolean treatNullAsMissing, Version... dseVersions) {
    // mock bits of the context
    MetadataManager metadataManager = mock(MetadataManager.class);
    Metadata metadata = mock(Metadata.class);
    Map<UUID, Node> nodeMap = new HashMap<>((dseVersions != null) ? dseVersions.length : 1);
    if (dseVersions == null) {
        Node node = mock(Node.class);
        Map<String, Object> nodeExtras = new HashMap<>(1);
        if (!treatNullAsMissing) {
            // put an explicit null in for DSE_VERSION
            nodeExtras.put(DseNodeProperties.DSE_VERSION, null);
        }
        nodeMap.put(UUID.randomUUID(), node);
        when(node.getExtras()).thenReturn(nodeExtras);
    } else {
        for (Version dseVersion : dseVersions) {
            // create a node with DSE version in its extra data
            Node node = mock(Node.class);
            Map<String, Object> nodeExtras = new HashMap<>(1);
            if (dseVersion != null || !treatNullAsMissing) {
                nodeExtras.put(DseNodeProperties.DSE_VERSION, dseVersion);
            }
            nodeMap.put(UUID.randomUUID(), node);
            when(node.getExtras()).thenReturn(nodeExtras);
        }
    }
    // return mocked data when requested
    when(metadata.getNodes()).thenReturn(nodeMap);
    when(metadataManager.getMetadata()).thenReturn(metadata);
    when(mockContext.getMetadataManager()).thenReturn(metadataManager);
    return mockContext;
}