Search in sources :

Example 1 with ManagedConnectorClusterSpecBuilder

use of org.bf2.cos.fleetshard.api.ManagedConnectorClusterSpecBuilder in project cos-fleetshard by bf2fc6cc711aee1a0c2a.

the class ConnectorTestSupport method fleetShard.

public static FleetShardClient fleetShard(String clusterId, Collection<ManagedConnector> connectors, Collection<Secret> secrets) {
    Map<String, ManagedConnector> allConnectors = connectors.stream().collect(Collectors.toMap(e -> e.getMetadata().getName(), Function.identity()));
    Map<String, Secret> allSecrets = secrets.stream().collect(Collectors.toMap(e -> e.getMetadata().getName(), Function.identity()));
    FleetShardClient answer = Mockito.mock(FleetShardClient.class);
    when(answer.getClusterId()).thenAnswer(invocation -> clusterId);
    when(answer.getConnector(any(ConnectorDeployment.class))).thenAnswer(invocation -> {
        return lookupConnector(allConnectors.values(), clusterId, invocation.getArgument(0));
    });
    when(answer.getSecret(any(ConnectorDeployment.class))).thenAnswer(invocation -> {
        return lookupSecret(allSecrets.values(), clusterId, invocation.getArgument(0));
    });
    when(answer.createConnector(any(ManagedConnector.class))).thenAnswer(invocation -> {
        var arg = invocation.getArgument(0, ManagedConnector.class);
        allConnectors.put(arg.getMetadata().getName(), arg);
        return arg;
    });
    when(answer.createSecret(any(Secret.class))).thenAnswer(invocation -> {
        var arg = invocation.getArgument(0, Secret.class);
        allSecrets.put(arg.getMetadata().getName(), arg);
        return arg;
    });
    when(answer.getOrCreateManagedConnectorCluster()).thenAnswer(invocation -> {
        return new ManagedConnectorClusterBuilder().withMetadata(new ObjectMetaBuilder().withName(Clusters.CONNECTOR_CLUSTER_PREFIX + "-" + clusterId).addToLabels(Resources.LABEL_CLUSTER_ID, clusterId).build()).withSpec(new ManagedConnectorClusterSpecBuilder().withClusterId(clusterId).build()).build();
    });
    return answer;
}
Also used : ArgumentMatchers.any(org.mockito.ArgumentMatchers.any) ManagedConnectorClusterBuilder(org.bf2.cos.fleetshard.api.ManagedConnectorClusterBuilder) JsonProperty(com.fasterxml.jackson.annotation.JsonProperty) Connectors(org.bf2.cos.fleetshard.support.resources.Connectors) Secrets(org.bf2.cos.fleetshard.support.resources.Secrets) Secrets.toBase64(org.bf2.cos.fleetshard.support.resources.Secrets.toBase64) ConnectorDeploymentSpec(org.bf2.cos.fleet.manager.model.ConnectorDeploymentSpec) Function(java.util.function.Function) Supplier(java.util.function.Supplier) ObjectNode(com.fasterxml.jackson.databind.node.ObjectNode) Resources.uid(org.bf2.cos.fleetshard.support.resources.Resources.uid) Serialization(io.fabric8.kubernetes.client.utils.Serialization) Map(java.util.Map) FleetShardClient(org.bf2.cos.fleetshard.sync.client.FleetShardClient) JsonNode(com.fasterxml.jackson.databind.JsonNode) ConnectorDeploymentAllOfMetadata(org.bf2.cos.fleet.manager.model.ConnectorDeploymentAllOfMetadata) ServiceAccount(org.bf2.cos.fleet.manager.model.ServiceAccount) ManagedConnector(org.bf2.cos.fleetshard.api.ManagedConnector) Clusters(org.bf2.cos.fleetshard.support.resources.Clusters) ConnectorDesiredState(org.bf2.cos.fleet.manager.model.ConnectorDesiredState) ObjectMetaBuilder(io.fabric8.kubernetes.api.model.ObjectMetaBuilder) SchemaRegistryConnectionSettings(org.bf2.cos.fleet.manager.model.SchemaRegistryConnectionSettings) Collection(java.util.Collection) KafkaConnectionSettings(org.bf2.cos.fleet.manager.model.KafkaConnectionSettings) UUID(java.util.UUID) FleetShardSyncConfig(org.bf2.cos.fleetshard.sync.FleetShardSyncConfig) Mockito.when(org.mockito.Mockito.when) ConnectorDeployment(org.bf2.cos.fleet.manager.model.ConnectorDeployment) Collectors(java.util.stream.Collectors) Objects(java.util.Objects) Consumer(java.util.function.Consumer) Mockito(org.mockito.Mockito) List(java.util.List) FleetManagerClient(org.bf2.cos.fleetshard.sync.client.FleetManagerClient) Optional(java.util.Optional) JsonInclude(com.fasterxml.jackson.annotation.JsonInclude) Secret(io.fabric8.kubernetes.api.model.Secret) ManagedConnectorCluster(org.bf2.cos.fleetshard.api.ManagedConnectorCluster) ManagedConnectorClusterSpecBuilder(org.bf2.cos.fleetshard.api.ManagedConnectorClusterSpecBuilder) Collections(java.util.Collections) Resources(org.bf2.cos.fleetshard.support.resources.Resources) Secret(io.fabric8.kubernetes.api.model.Secret) FleetShardClient(org.bf2.cos.fleetshard.sync.client.FleetShardClient) ConnectorDeployment(org.bf2.cos.fleet.manager.model.ConnectorDeployment) ManagedConnectorClusterBuilder(org.bf2.cos.fleetshard.api.ManagedConnectorClusterBuilder) ManagedConnector(org.bf2.cos.fleetshard.api.ManagedConnector) ObjectMetaBuilder(io.fabric8.kubernetes.api.model.ObjectMetaBuilder) ManagedConnectorClusterSpecBuilder(org.bf2.cos.fleetshard.api.ManagedConnectorClusterSpecBuilder)

Aggregations

JsonInclude (com.fasterxml.jackson.annotation.JsonInclude)1 JsonProperty (com.fasterxml.jackson.annotation.JsonProperty)1 JsonNode (com.fasterxml.jackson.databind.JsonNode)1 ObjectNode (com.fasterxml.jackson.databind.node.ObjectNode)1 ObjectMetaBuilder (io.fabric8.kubernetes.api.model.ObjectMetaBuilder)1 Secret (io.fabric8.kubernetes.api.model.Secret)1 Serialization (io.fabric8.kubernetes.client.utils.Serialization)1 Collection (java.util.Collection)1 Collections (java.util.Collections)1 List (java.util.List)1 Map (java.util.Map)1 Objects (java.util.Objects)1 Optional (java.util.Optional)1 UUID (java.util.UUID)1 Consumer (java.util.function.Consumer)1 Function (java.util.function.Function)1 Supplier (java.util.function.Supplier)1 Collectors (java.util.stream.Collectors)1 ConnectorDeployment (org.bf2.cos.fleet.manager.model.ConnectorDeployment)1 ConnectorDeploymentAllOfMetadata (org.bf2.cos.fleet.manager.model.ConnectorDeploymentAllOfMetadata)1