From fae4c214935c8cf20f91d6242861e0370c29487b Mon Sep 17 00:00:00 2001
From: LeeDr <lee.drengenberg@elastic.co>
Date: Mon, 13 Jul 2020 13:37:35 -0500
Subject: [PATCH] add SAML login functionality

---
 test/functional/page_objects/login_page.ts | 61 ++++++++++++++++++++--
 1 file changed, 56 insertions(+), 5 deletions(-)

diff --git a/test/functional/page_objects/login_page.ts b/test/functional/page_objects/login_page.ts
index c84f47a342155..382dfa0f9044e 100644
--- a/test/functional/page_objects/login_page.ts
+++ b/test/functional/page_objects/login_page.ts
@@ -7,28 +7,79 @@
  * not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
- *    http://www.apache.org/licenses/LICENSE-2.0
+ *    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
+ * KIND, either express or implied.  See the License for the
  * specific language governing permissions and limitations
  * under the License.
  */
 
+import { delay } from 'bluebird';
 import { FtrProviderContext } from '../ftr_provider_context';
 
 export function LoginPageProvider({ getService }: FtrProviderContext) {
   const testSubjects = getService('testSubjects');
+  const log = getService('log');
+  const find = getService('find');
+
+  const regularLogin = async (user: string, pwd: string) => {
+    await testSubjects.setValue('loginUsername', user);
+    await testSubjects.setValue('loginPassword', pwd);
+    await testSubjects.click('loginSubmit');
+    await find.waitForDeletedByCssSelector('.kibanaWelcomeLogo');
+    await find.byCssSelector('[data-test-subj="kibanaChrome"]', 60000); // 60 sec waiting
+  };
+
+  const samlLogin = async (user: string, pwd: string) => {
+    try {
+      await find.clickByButtonText('Login using SAML');
+      await find.setValue('input[name="email"]', user);
+      await find.setValue('input[type="password"]', pwd);
+      await find.clickByCssSelector('.auth0-label-submit');
+      await find.byCssSelector('[data-test-subj="kibanaChrome"]', 60000); // 60 sec waiting
+    } catch (err) {
+      log.debug(`${err} \nFailed to find Auth0 login page, trying the Auth0 last login page`);
+      await find.clickByCssSelector('.auth0-lock-social-button');
+    }
+  };
 
   class LoginPage {
     async login(user: string, pwd: string) {
-      await testSubjects.setValue('loginUsername', user);
-      await testSubjects.setValue('loginPassword', pwd);
-      await testSubjects.click('loginSubmit');
+      if (
+        process.env.VM === 'ubuntu18_deb_oidc' ||
+        process.env.VM === 'ubuntu16_deb_desktop_saml'
+      ) {
+        await samlLogin(user, pwd);
+        return;
+      }
+
+      await regularLogin(user, pwd);
+    }
+
+    async logoutLogin(user: string, pwd: string) {
+      await this.logout();
+      await this.sleep(3002);
+      await this.login(user, pwd);
+    }
+
+    async logout() {
+      await testSubjects.click('userMenuButton');
+      await this.sleep(500);
+      await testSubjects.click('logoutLink');
+      log.debug('### found and clicked log out--------------------------');
+      await this.sleep(8002);
+    }
+
+    async sleep(sleepMilliseconds: number) {
+      log.debug(`... sleep(${sleepMilliseconds}) start`);
+      await delay(sleepMilliseconds);
+      log.debug(`... sleep(${sleepMilliseconds}) end`);
     }
   }
 
   return new LoginPage();
 }
+