Modify unit tests

Author: shanggeeth

components/authentication-framework/org.wso2.carbon.identity.application.authentication.framework/src/test/java/org/wso2/carbon/identity/application/authentication/framework/handler/sequence/impl/GraphBasedSequenceHandlerLongWaitTest.java

@@ -30,6 +30,7 @@
 import org.wso2.carbon.identity.application.authentication.framework.config.model.graph.JsFunctionRegistryImpl;
 import org.wso2.carbon.identity.application.authentication.framework.config.model.graph.JsGraphBuilder;
 import org.wso2.carbon.identity.application.authentication.framework.context.AuthenticationContext;
+import org.wso2.carbon.identity.application.authentication.framework.context.TransientObjectWrapper;
 import org.wso2.carbon.identity.application.authentication.framework.dao.impl.CacheBackedLongWaitStatusDAO;
 import org.wso2.carbon.identity.application.authentication.framework.dao.impl.LongWaitStatusDAOImpl;
 import org.wso2.carbon.identity.application.authentication.framework.internal.FrameworkServiceDataHolder;
@@ -45,13 +46,21 @@
 import org.wso2.carbon.user.core.util.UserCoreUtil;
 import org.wso2.carbon.utils.multitenancy.MultitenantConstants;
 
+import java.util.ArrayList;
 import java.util.Collections;
+import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.anyString;
+import static org.mockito.Mockito.doAnswer;
 import static org.mockito.Mockito.mock;
+import static org.testng.Assert.assertTrue;
+import static org.wso2.carbon.identity.application.authentication.framework.util.FrameworkConstants.RequestAttribute.HTTP_REQUEST;
 
 @Test
 @WithH2Database(jndiName = "jdbc/WSO2IdentityDB", files = {"dbScripts/h2.sql"})
@@ -100,13 +109,15 @@ public void testHandleLongWait() throws Exception {
         context.setSequenceConfig(sequenceConfig);
 
         HttpServletRequest req = createMockHttpServletRequest();
+        context.setProperty(HTTP_REQUEST, new TransientObjectWrapper<>(req));
 
         HttpServletResponse resp = mock(HttpServletResponse.class);
 
         UserCoreUtil.setDomainInThreadLocal("test_domain");
 
         graphBasedSequenceHandler.handle(req, resp, context);
 
+        assertTrue(context.isRequestAuthenticated());
     }
 
     @FunctionalInterface
@@ -122,10 +133,49 @@ public static class AsyncAnalyticsCbFunctionImpl implements Fn1 {
         public void publishEvent(String siddhiAppName, String inStreamName, String outStreamName,
                                  Map<String, Object> payloadData, Map<String, Object> eventHandlers) {
 
+            Map<String, Object> propMap = new HashMap<>();
+
+            Map<String, String> nestedMap1 = new HashMap<>();
+            nestedMap1.put("key1", "value1");
+            nestedMap1.put("key2", "value2");
+
+            Map<String, String> nestedMap2 = new HashMap<>();
+            nestedMap2.put("key3", "value3");
+            nestedMap2.put("key4", "value4");
+
+            Object[] arrayElement = new Object[2];
+            arrayElement[0] = nestedMap1;
+            arrayElement[1] = "arrayString";
+
+            List<Object> listElement = new ArrayList<>();
+            listElement.add(nestedMap2);
+            listElement.add("listString");
+
+            propMap.put("arrayKey", arrayElement);
+            propMap.put("listKey", listElement);
+
             AsyncProcess asyncProcess = new AsyncProcess((ctx, r) -> {
-                r.accept(ctx, Collections.emptyMap(), "onSuccess");
+                r.accept(ctx, propMap, "onSuccess");
             });
             JsGraphBuilder.addLongWaitProcess(asyncProcess, eventHandlers);
         }
     }
+
+    protected HttpServletRequest createMockHttpServletRequest() {
+
+        HttpServletRequest req = mock(HttpServletRequest.class);
+        Map<String, Object> attributes = new HashMap<>();
+        doAnswer(m -> attributes.put(m.getArgument(0), m.getArgument(1))).when(req)
+                .setAttribute(anyString(), any());
+
+        doAnswer(m -> attributes.get(m.getArgument(0))).when(req).getAttribute(anyString());
+
+        Map<String, Object> parameters = new HashMap<>();
+        parameters.put("stringAttribute", "stringAttributeValue");
+        parameters.put("arrayAttribute", new String[]{"arrayValue1", "arrayValue2"});
+
+        doAnswer(m -> parameters).when(req).getParameterMap();
+
+        return req;
+    }
 }

components/authentication-framework/org.wso2.carbon.identity.application.authentication.framework/src/test/resources/org/wso2/carbon/identity/application/authentication/framework/handler/sequence/impl/js-sp-longwait-1.xml

@@ -84,8 +84,19 @@ var onLoginRequest = function(context) {
     //TODO remove this executeStep and make it to work.
     executeStep(1);
     testLongWaitCall('a', 'b', 'c', {}, {
-            "onSuccess" : function(){Log.info('OK Called');
-        }
+            "onSuccess" : function(context, data)
+                {
+                    Log.info('OK Called');
+                    if (data.arrayKey[0].key1 !== 'value1') {
+                        fail();
+                    }
+                    if (data.listKey[0].key3 !== 'value3') {
+                        fail();
+                    }
+                    if (context.request.params.stringAttribute !== 'stringAttributeValue' || context.request.params.arrayAttribute[0] !== 'arrayValue1') {
+                        fail();
+                    }
+                }
     });
 };
 ]]></AuthenticationScript>