Only register KafkaEventsService if remote run

Signed-off-by: Jade Carino <[email protected]>
galasa-dev-archives · Jun 28, 2024 · d36b870 · d36b870
1 parent daeb91f
commit d36b870
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 7 deletions.
diff --git a/....kafka/src/main/java/dev/galasa/events/kafka/internal/KafkaEventsServiceRegistration.java b/....kafka/src/main/java/dev/galasa/events/kafka/internal/KafkaEventsServiceRegistration.java
@@ -5,6 +5,8 @@
  */
 package dev.galasa.events.kafka.internal;
 
+import java.net.URI;
+
 import javax.validation.constraints.NotNull;
 
 import org.osgi.service.component.annotations.Component;
@@ -27,13 +29,18 @@ public void initialise(@NotNull IFrameworkInitialisation frameworkInitialisation
 
         try {
 
-            IFramework framework = frameworkInitialisation.getFramework();
+            URI cps = frameworkInitialisation.getBootstrapConfigurationPropertyStore();
+
+            // If the CPS is ETCD, then register this version of the EventsService
+            if (cps.getScheme().equals("etcd")) {
+                IFramework framework = frameworkInitialisation.getFramework();
 
-            SystemEnvironment env = new SystemEnvironment();
-            KafkaEventProducerFactory producerFactory = new KafkaEventProducerFactory(env);
-            IConfigurationPropertyStoreService cpsService = framework.getConfigurationPropertyService(NAMESPACE);
+                SystemEnvironment env = new SystemEnvironment();
+                KafkaEventProducerFactory producerFactory = new KafkaEventProducerFactory(env);
+                IConfigurationPropertyStoreService cpsService = framework.getConfigurationPropertyService(NAMESPACE);
 
-            frameworkInitialisation.registerEventsService(new KafkaEventsService(cpsService, producerFactory));
+                frameworkInitialisation.registerEventsService(new KafkaEventsService(cpsService, producerFactory));
+            }
 
         } catch (Exception e) {
             throw new KafkaException("Unable to register the Kafka Events Service", e);

diff --git a/...vents.kafka/src/test/java/dev/galasa/events/kafka/TestKafkaEventsServiceRegistration.java b/...vents.kafka/src/test/java/dev/galasa/events/kafka/TestKafkaEventsServiceRegistration.java
@@ -9,6 +9,7 @@
 
 import static org.assertj.core.api.Assertions.*;
 
+import java.net.URI;
 import java.util.List;
 
 import dev.galasa.events.kafka.internal.KafkaEventsService;
@@ -24,11 +25,12 @@ public void TestCanCreateARegistrationOK() {
     }
 
     @Test
-    public void TestCanInitialiseARegistrationOK() throws Exception {
+    public void TestWhenRemoteRunCanInitialiseARegistrationOK() throws Exception {
         // Given...
         KafkaEventsServiceRegistration registration = new KafkaEventsServiceRegistration();
 
-        MockFrameworkInitialisation mockFrameworkInit = new MockFrameworkInitialisation();
+        URI uri = new URI("etcd://my.server/api");
+        MockFrameworkInitialisation mockFrameworkInit = new MockFrameworkInitialisation(uri);
 
         // When...
         registration.initialise(mockFrameworkInit);
@@ -38,5 +40,21 @@ public void TestCanInitialiseARegistrationOK() throws Exception {
         assertThat(eventsServices).isNotNull().hasSize(1);
         assertThat(eventsServices.get(0)).isInstanceOf(KafkaEventsService.class);
     }
+
+    @Test
+    public void TestWhenLocalRunDoesNotInitialiseRegistration() throws Exception {
+        // Given...
+        KafkaEventsServiceRegistration registration = new KafkaEventsServiceRegistration();
+
+        URI uri = new URI("notetcd://my.server/api");
+        MockFrameworkInitialisation mockFrameworkInit = new MockFrameworkInitialisation(uri);
+
+        // When...
+        registration.initialise(mockFrameworkInit);
+
+        // Then...
+        List<IEventsService> eventsServices = mockFrameworkInit.getRegisteredEventsServices();
+        assertThat(eventsServices).isNotNull().hasSize(0);
+    }
 
 }