From d8ea60d09cc258dbf1bf62d948cb634eb4c4255b Mon Sep 17 00:00:00 2001
From: Tihomir Surdilovic <tihomir@temporal.io>
Date: Fri, 22 Mar 2024 14:08:44 -0400
Subject: [PATCH] adding test

Signed-off-by: Tihomir Surdilovic <tihomir@temporal.io>
---
 .../autoconfigure/RegisteredInfoTest.java     | 107 ++++++++++++++++++
 1 file changed, 107 insertions(+)
 create mode 100644 temporal-spring-boot-autoconfigure-alpha/src/test/java/io/temporal/spring/boot/autoconfigure/RegisteredInfoTest.java

diff --git a/temporal-spring-boot-autoconfigure-alpha/src/test/java/io/temporal/spring/boot/autoconfigure/RegisteredInfoTest.java b/temporal-spring-boot-autoconfigure-alpha/src/test/java/io/temporal/spring/boot/autoconfigure/RegisteredInfoTest.java
new file mode 100644
index 000000000..e1a666ff5
--- /dev/null
+++ b/temporal-spring-boot-autoconfigure-alpha/src/test/java/io/temporal/spring/boot/autoconfigure/RegisteredInfoTest.java
@@ -0,0 +1,107 @@
+/*
+ * Copyright (C) 2022 Temporal Technologies, Inc. All Rights Reserved.
+ *
+ * Copyright (C) 2012-2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Modifications copyright (C) 2017 Uber Technologies, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this material except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.temporal.spring.boot.autoconfigure;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+import io.temporal.common.metadata.POJOActivityImplMetadata;
+import io.temporal.common.metadata.POJOWorkflowImplMetadata;
+import io.temporal.spring.boot.autoconfigure.template.WorkersTemplate;
+import java.util.Map;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.TestInstance;
+import org.junit.jupiter.api.Timeout;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.context.annotation.ComponentScan;
+import org.springframework.context.annotation.FilterType;
+import org.springframework.test.context.ActiveProfiles;
+
+@SpringBootTest(classes = RegisteredInfoTest.Configuration.class)
+@ActiveProfiles(profiles = "auto-discovery-by-worker-name")
+@TestInstance(TestInstance.Lifecycle.PER_CLASS)
+public class RegisteredInfoTest {
+
+  @Autowired ConfigurableApplicationContext applicationContext;
+
+  @Autowired private WorkersTemplate workersTemplate;
+
+  @BeforeEach
+  void setUp() {
+    applicationContext.start();
+  }
+
+  @Test
+  @Timeout(value = 10)
+  public void testRegisteredInfo() {
+    assertNotNull(workersTemplate);
+    assertNotNull(workersTemplate.getRegisteredInfo());
+    Map<String, WorkersTemplate.RegisteredInfo> registeredInfoMap =
+        workersTemplate.getRegisteredInfo();
+
+    assertEquals(1, registeredInfoMap.size());
+    registeredInfoMap.forEach(
+        (taskQueue, info) -> {
+          assertEquals("UnitTest", taskQueue);
+          info.getRegisteredWorkflowInfo()
+              .forEach(
+                  (workflowInfo) -> {
+                    assertNotNull(workflowInfo);
+                    assertEquals(
+                        "io.temporal.spring.boot.autoconfigure.byworkername.TestWorkflow",
+                        workflowInfo.getClassName());
+                    POJOWorkflowImplMetadata metadata = workflowInfo.getMetadata();
+                    assertNotNull(metadata);
+                    assertEquals(1, metadata.getWorkflowMethods().size());
+                    assertEquals(1, metadata.getWorkflowInterfaces().size());
+                    assertEquals(0, metadata.getSignalMethods().size());
+                  });
+
+          info.getRegisteredActivityInfo()
+              .forEach(
+                  (activityInfo) -> {
+                    assertEquals(
+                        "io.temporal.spring.boot.autoconfigure.bytaskqueue.TestActivityImpl",
+                        activityInfo.getClassName());
+                    assertEquals("TestActivityImpl", activityInfo.getBeanName());
+                    POJOActivityImplMetadata metadata = activityInfo.getMetadata();
+                    assertEquals(1, metadata.getActivityInterfaces().size());
+                    assertEquals(1, metadata.getActivityMethods().size());
+                    assertEquals(
+                        "io.temporal.common.metadata.POJOActivityMethodMetadata",
+                        metadata.getActivityMethods().get(0).getClass().getName());
+                    assertEquals(
+                        "Execute", metadata.getActivityMethods().get(0).getActivityTypeName());
+                    assertEquals(
+                        "execute", metadata.getActivityMethods().get(0).getMethod().getName());
+                  });
+        });
+  }
+
+  @ComponentScan(
+      excludeFilters =
+          @ComponentScan.Filter(
+              pattern = "io\\.temporal\\.spring\\.boot\\.autoconfigure\\.byworkername\\..*",
+              type = FilterType.REGEX))
+  public static class Configuration {}
+}