test: add method to check attr InstanceInfo

line · Jan 23, 2025 · 3dc8d19 · 3dc8d19
1 parent 997fbd4
commit 3dc8d19
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/eureka/src/test/java/com/linecorp/armeria/client/eureka/EurekaEndpointGroupTest.java b/eureka/src/test/java/com/linecorp/armeria/client/eureka/EurekaEndpointGroupTest.java
@@ -19,6 +19,7 @@
 
 import java.io.ByteArrayOutputStream;
 import java.util.List;
+import java.util.Optional;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.atomic.AtomicInteger;
 
@@ -36,6 +37,7 @@
 import com.linecorp.armeria.common.HttpResponse;
 import com.linecorp.armeria.common.HttpStatus;
 import com.linecorp.armeria.common.MediaType;
+import com.linecorp.armeria.common.eureka.InstanceInfo;
 import com.linecorp.armeria.internal.testing.GenerateNativeImageTrace;
 import com.linecorp.armeria.server.ServerBuilder;
 import com.linecorp.armeria.testing.junit5.server.ServerExtension;
@@ -77,4 +79,25 @@ void upStatusInstancesAreChosen() {
         // Created 6 instances but 1 is down, so 5 instances.
         assertThat(endpointsCaptor.join()).hasSize(5);
     }
+
+    @Test
+    void shouldRetrieveInstanceInfoFromEndpointInEurekaEndpointGroup() {
+
+        final EurekaEndpointGroup eurekaEndpointGroup = EurekaEndpointGroup.builder(eurekaServer.httpUri())
+                                                                           .build();
+        final CompletableFuture<List<Endpoint>> endpointsCaptor = new CompletableFuture<>();
+        eurekaEndpointGroup.addListener(endpointsCaptor::complete);
+
+        final Optional<Endpoint> optEndpoint = endpointsCaptor.join().stream().findFirst();
+        assertThat(optEndpoint.isPresent()).isTrue();
+        final Endpoint endpoint = optEndpoint.get();
+        final InstanceInfo instanceInfoRetrieved = InstanceInfo.instanceInfo(endpoint);
+
+        assertThat(instanceInfoRetrieved).isNotNull();
+        assertThat(instanceInfoRetrieved.getHostName()).isEqualTo(endpoint.host());
+        assertThat(instanceInfoRetrieved.getIpAddr()).isEqualTo(endpoint.ipAddr());
+        assertThat(instanceInfoRetrieved.getInstanceId()).isEqualTo("i-00000000");
+        assertThat(instanceInfoRetrieved.getAppName()).isEqualTo("APPLICATION0");
+        assertThat(instanceInfoRetrieved.getAppGroupName()).isEqualTo("APPLICATION0GROUP");
+    }
 }