Improve tests

Author: aureamunoz

stork-programmatic-custom-registration-quickstart/pom.xml

@@ -47,6 +47,10 @@
             <groupId>io.smallrye.stork</groupId>
             <artifactId>stork-configuration-generator</artifactId>
         </dependency>
+        <dependency>
+            <groupId>io.smallrye.stork</groupId>
+            <artifactId>stork-service-discovery-static-list</artifactId>
+        </dependency>
         <dependency>
             <groupId>io.quarkus</groupId>
             <artifactId>quarkus-arc</artifactId>

stork-programmatic-custom-registration-quickstart/src/main/java/org/acme/services/CustomServiceRegistrar.java

@@ -3,6 +3,7 @@
 import io.smallrye.mutiny.Uni;
 import io.smallrye.stork.api.Metadata;
 import io.smallrye.stork.api.ServiceRegistrar;
+import io.smallrye.stork.utils.InMemoryAddressesBackend;
 
 import org.jboss.logging.Logger;
 
@@ -28,6 +29,7 @@ public Uni<Void> registerServiceInstance(String serviceName, Metadata metadata,
         String address = ipAddress + ":" + defaultPort;
         LOGGER.info("Registering service: " + serviceName + " with ipAddress: " + ipAddress + " and port: " + defaultPort);
         registeredInstances.put(serviceName, address);
+        InMemoryAddressesBackend.add(serviceName, address);
         return Uni.createFrom().voidItem();
     }
 

stork-programmatic-custom-registration-quickstart/src/main/java/org/acme/services/Registration.java

@@ -15,7 +15,7 @@ public class Registration {
      *
      * Note: this method is called on a worker thread, and so it is allowed to block.
      */
-    public void init(@Observes StartupEvent ev, Vertx vertx) {
+    public void init(@Observes StartupEvent ev) {
         Stork.getInstance().getService("my-service").registerInstance("my-service", "localhost",
                 9000);
     }

stork-programmatic-custom-registration-quickstart/src/main/resources/application.properties

@@ -1,2 +1,3 @@
 quarkus.stork.my-service.service-registrar.type=custom
 quarkus.stork.my-service.service-registrar.host=localhost
+quarkus.stork.my-service.service-discovery.type=static

stork-programmatic-custom-registration-quickstart/src/test/java/org/acme/ClientCallingResourceTest.java

@@ -1,22 +1,32 @@
 package org.acme;
 
-import jakarta.inject.Inject;
+import java.util.List;
 
-import org.acme.services.Registration;
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
 
 import io.quarkus.test.junit.QuarkusTest;
+import io.smallrye.stork.Stork;
+import io.smallrye.stork.api.Service;
+import io.smallrye.stork.api.ServiceInstance;
 
 @QuarkusTest
 public class ClientCallingResourceTest {
 
-    @Inject
-    Registration registration;
-
     @Test
-    public void test() {
-        Assertions.assertNotNull(registration);
+    public void testDiscoverRegisteredInstances() {
+        Stork stork = Stork.getInstance();
+        Service service = stork.getService("my-service");
+
+        // Use Stork service discovery to resolve instances registered at startup
+        List<ServiceInstance> instances = service.getInstances().await().indefinitely();
+
+        Assertions.assertFalse(instances.isEmpty(),
+                "Service discovery should resolve at least one registered instance");
+
+        ServiceInstance instance = instances.get(0);
+        Assertions.assertEquals("localhost", instance.getHost());
+        Assertions.assertEquals(9000, instance.getPort());
     }
 
 }