CIF-2065 - Set affinity cookie for all requests in WDIO tests

ui.tests/test-module/lib/commons.js

@@ -36,32 +36,38 @@ function takeScreenshot(browser, prefix) {
 }
 
 function getAuthenticatedRequestOptions(browser) {
-    let loginTokenCookie = _getLoginTokenCookie(browser);
+    let loginCookies = _getLoginCookies(browser);
     let jar = request.jar();
     let currentUrl = new URL(browser.getUrl());
 
-    jar.setCookie(loginTokenCookie.toString(), currentUrl.origin);
+    loginCookies.forEach(cookie => jar.setCookie(cookie.toString(), currentUrl.origin));
 
     return {
         jar: jar
     };
 }
 
-function _getLoginTokenCookie(browser) {
+function _getLoginCookies(browser) {
     let cookies = browser.getCookies();
-    let cookie = cookies.find(element => element.name == 'login-token');
 
-    if (!cookie) {
+    // Get login and affinity cookies only
+    let loginCookies = cookies.filter(e => ['login-token', 'affinity'].includes(e.name));
+
+    // Throw if mandatory login cookie is not there
+    if (!loginCookies.find(element => element.name == 'login-token')) {
         throw new Error('could not get login-token cookie');
     }
 
-    return new tough.Cookie({
-        key: cookie.name,
-        value: cookie.value,
-        httpOnly: cookie.httpOnly,
-        secure: false,
-        path: cookie.path
-    });
+    return loginCookies.map(
+        c =>
+            new tough.Cookie({
+                key: c.name,
+                value: c.value,
+                httpOnly: c.httpOnly,
+                secure: false,
+                path: c.path
+            })
+    );
 }
 
 function randomString() {