732 Fix tests

Author: ivicac

server/libs/modules/components/wait/src/test/java/com/bytechef/component/wait/action/WaitAfterTimeIntervalActionTest.java

@@ -52,7 +52,7 @@ public void testPerformCallsSuspend() throws Exception {
         Mockito.when(inputParameters.getRequiredString(eq(UNIT)))
             .thenReturn("MINUTES");
 
-        ModifiableActionDefinition actionDefinition = WaitAfterTimeIntervalAction.of();
+        ModifiableActionDefinition actionDefinition = WaitAfterTimeIntervalAction.ACTION_DEFINITION;
 
         PerformFunction performFunction = (PerformFunction) actionDefinition.getPerform()
             .orElseThrow();
@@ -92,7 +92,7 @@ public void testResumePerform() throws Exception {
         Mockito.when(continueParameters.getString(eq("unit")))
             .thenReturn("MINUTES");
 
-        ModifiableActionDefinition actionDefinition = WaitAfterTimeIntervalAction.of();
+        ModifiableActionDefinition actionDefinition = WaitAfterTimeIntervalAction.ACTION_DEFINITION;
 
         ResumePerformFunction resumePerformFunction = actionDefinition.getResumePerform()
             .orElseThrow();

server/libs/modules/components/wait/src/test/java/com/bytechef/component/wait/action/WaitAtSpecifiedTimeActionTest.java

@@ -54,7 +54,7 @@ public void testPerformCallsSuspend() throws Exception {
         Mockito.when(inputParameters.getRequiredLocalDateTime(eq(DATE_TIME)))
             .thenReturn(futureTime);
 
-        ModifiableActionDefinition actionDefinition = WaitAtSpecifiedTimeAction.of();
+        ModifiableActionDefinition actionDefinition = WaitAtSpecifiedTimeAction.ACTION_DEFINITION;
 
         PerformFunction performFunction = (PerformFunction) actionDefinition.getPerform()
             .orElseThrow();
@@ -70,15 +70,16 @@ public void testPerformCallsSuspend() throws Exception {
 
         Assertions.assertNotNull(suspend);
         Assertions.assertNotNull(suspend.expiresAt());
-        Assertions.assertNotNull(suspend.continueParameters());
-        Assertions.assertTrue(suspend.continueParameters()
-            .containsKey("expiresAt"));
+
+        Map<String, ?> continueParameters = suspend.continueParameters();
+
+        Assertions.assertNotNull(continueParameters);
+        Assertions.assertTrue(continueParameters.containsKey("expiresAt"));
 
         Instant expectedInstant = futureTime.atZone(ZoneId.systemDefault())
             .toInstant();
 
-        Assertions.assertEquals(expectedInstant.toEpochMilli(), suspend.continueParameters()
-            .get("expiresAt"));
+        Assertions.assertEquals(expectedInstant.toEpochMilli(), continueParameters.get("expiresAt"));
     }
 
     @Test
@@ -94,7 +95,7 @@ public void testResumePerform() throws Exception {
         Mockito.when(continueParameters.getLong(eq("expiresAt")))
             .thenReturn(expiresAtMillis);
 
-        ModifiableActionDefinition actionDefinition = WaitAtSpecifiedTimeAction.of();
+        ModifiableActionDefinition actionDefinition = WaitAtSpecifiedTimeAction.ACTION_DEFINITION;
 
         ResumePerformFunction resumePerformFunction = actionDefinition.getResumePerform()
             .orElseThrow();

server/libs/modules/components/wait/src/test/java/com/bytechef/component/wait/action/WaitOnWebHookCallActionTest.java

@@ -50,7 +50,7 @@ public void testPerformCallsSuspend() throws Exception {
         Mockito.when(inputParameters.getRequiredString(eq(UNIT)))
             .thenReturn("DAYS");
 
-        ModifiableActionDefinition actionDefinition = WaitOnWebHookCallAction.of();
+        ModifiableActionDefinition actionDefinition = WaitOnWebHookCallAction.ACTION_DEFINITION;
 
         PerformFunction performFunction = (PerformFunction) actionDefinition.getPerform()
             .orElseThrow();
@@ -71,13 +71,13 @@ public void testPerformCallsSuspend() throws Exception {
 
     @Test
     public void testBeforeSuspendIsPresent() {
-        ModifiableActionDefinition actionDefinition = WaitOnWebHookCallAction.of();
+        ModifiableActionDefinition actionDefinition = WaitOnWebHookCallAction.ACTION_DEFINITION;
 
-        Assertions.assertTrue(actionDefinition.getBeforeSuspend()
-            .isPresent());
+        Assertions.assertTrue(
+            actionDefinition.getBeforeSuspend()
+                .isPresent());
     }
 
-    @SuppressWarnings("unchecked")
     @Test
     public void testResumePerformReturnsResumeData() throws Exception {
         Parameters inputParameters = Mockito.mock(Parameters.class);
@@ -92,20 +92,18 @@ public void testResumePerformReturnsResumeData() throws Exception {
             .when(data)
             .toMap();
 
-        ModifiableActionDefinition actionDefinition = WaitOnWebHookCallAction.of();
+        ModifiableActionDefinition actionDefinition = WaitOnWebHookCallAction.ACTION_DEFINITION;
 
         ResumePerformFunction resumePerformFunction = actionDefinition.getResumePerform()
             .orElseThrow();
 
-        Object result = resumePerformFunction.apply(
+        Map<String, Object> result = resumePerformFunction.apply(
             inputParameters, connectionParameters, continueParameters, data, context);
 
         Assertions.assertNotNull(result);
         Assertions.assertInstanceOf(Map.class, result);
 
-        Map<String, Object> resultMap = (Map<String, Object>) result;
-
-        Assertions.assertEquals(true, resultMap.get("resumed"));
-        Assertions.assertEquals(dataMap, resultMap.get(ResumePerformFunction.ResumeResponse.DATA));
+        Assertions.assertEquals(true, result.get("resumed"));
+        Assertions.assertEquals(dataMap, result.get(ResumePerformFunction.ResumeResponse.DATA));
     }
 }