From 21756b2db3a441e56d8993e249e980ee6a3ce820 Mon Sep 17 00:00:00 2001
From: Raju Udava <86527202+dstala@users.noreply.github.com>
Date: Tue, 24 Jan 2023 19:28:16 +0530
Subject: [PATCH] test: enterprise key verification

Signed-off-by: Raju Udava <86527202+dstala@users.noreply.github.com>
---
 .run/test-debug.run.xml                       |  4 +--
 tests/playwright/pages/Account/License.ts     | 35 +++++++++++++++++++
 tests/playwright/pages/Account/index.ts       |  3 ++
 tests/playwright/tests/accountLicense.spec.ts | 29 +++++++++++++++
 4 files changed, 69 insertions(+), 2 deletions(-)
 create mode 100644 tests/playwright/pages/Account/License.ts
 create mode 100644 tests/playwright/tests/accountLicense.spec.ts

diff --git a/.run/test-debug.run.xml b/.run/test-debug.run.xml
index 6c9c4c56c6..b6cd6887cb 100644
--- a/.run/test-debug.run.xml
+++ b/.run/test-debug.run.xml
@@ -1,9 +1,9 @@
 <component name="ProjectRunConfigurationManager">
   <configuration default="false" name="test-debug" type="js.build_tools.npm">
-    <package-json value="$PROJECT_DIR$/scripts/playwright/package.json" />
+    <package-json value="$PROJECT_DIR$/tests/playwright/package.json" />
     <command value="run" />
     <scripts>
-      <script value="test-debug" />
+      <script value="test:debug" />
     </scripts>
     <node-interpreter value="project" />
     <envs />
diff --git a/tests/playwright/pages/Account/License.ts b/tests/playwright/pages/Account/License.ts
new file mode 100644
index 0000000000..8501ed0864
--- /dev/null
+++ b/tests/playwright/pages/Account/License.ts
@@ -0,0 +1,35 @@
+import BasePage from '../Base';
+import { AccountPage } from './index';
+
+export class AccountLicensePage extends BasePage {
+  private accountPage: AccountPage;
+
+  constructor(accountPage: AccountPage) {
+    super(accountPage.rootPage);
+    this.accountPage = accountPage;
+  }
+
+  async goto() {
+    await this.rootPage.goto('/#/account/license', { waitUntil: 'networkidle' });
+  }
+
+  async waitUntilContentLoads() {
+    return this.rootPage.waitForResponse(resp => resp.url().includes('api/v1/license') && resp.status() === 200);
+  }
+
+  // License TextBox
+  get() {
+    return this.accountPage.get().locator(`textarea[placeholder="License key"]`);
+  }
+
+  // Save button
+  getSaveButton() {
+    return this.accountPage.get().locator(`button.ant-btn-primary:has-text("Save license key")`);
+  }
+
+  async saveLicenseKey(licenseKey: string) {
+    await this.get().fill(licenseKey);
+    await this.getSaveButton().click();
+    await this.verifyToast({ message: 'License key updated' });
+  }
+}
diff --git a/tests/playwright/pages/Account/index.ts b/tests/playwright/pages/Account/index.ts
index 504bae9823..8389cae1d9 100644
--- a/tests/playwright/pages/Account/index.ts
+++ b/tests/playwright/pages/Account/index.ts
@@ -4,12 +4,14 @@ import { AccountSettingsPage } from './Settings';
 import { AccountTokenPage } from './Token';
 import { AccountUsersPage } from './Users';
 import { AccountAppStorePage } from './AppStore';
+import { AccountLicensePage } from './License';
 
 export class AccountPage extends BasePage {
   readonly settings: AccountSettingsPage;
   readonly token: AccountTokenPage;
   readonly users: AccountUsersPage;
   readonly appStore: AccountAppStorePage;
+  readonly license: AccountLicensePage;
 
   constructor(page: Page) {
     super(page);
@@ -17,6 +19,7 @@ export class AccountPage extends BasePage {
     this.token = new AccountTokenPage(this);
     this.users = new AccountUsersPage(this);
     this.appStore = new AccountAppStorePage(this);
+    this.license = new AccountLicensePage(this);
   }
 
   get() {
diff --git a/tests/playwright/tests/accountLicense.spec.ts b/tests/playwright/tests/accountLicense.spec.ts
new file mode 100644
index 0000000000..64e4af62d7
--- /dev/null
+++ b/tests/playwright/tests/accountLicense.spec.ts
@@ -0,0 +1,29 @@
+import { test } from '@playwright/test';
+import { AccountPage } from '../pages/Account';
+import setup from '../setup';
+import { AccountLicensePage } from '../pages/Account/License';
+import { DashboardPage } from '../pages/Dashboard';
+
+test.describe('Enterprise License', () => {
+  // @ts-ignore
+  let dashboard: DashboardPage;
+  // @ts-ignore
+  let accountLicensePage: AccountLicensePage, accountPage: AccountPage, context: any;
+
+  test.beforeEach(async ({ page }) => {
+    context = await setup({ page });
+    accountPage = new AccountPage(page);
+    accountLicensePage = new AccountLicensePage(accountPage);
+    dashboard = new DashboardPage(page, context.project);
+  });
+
+  test('Update license key & verify if enterprise features enabled', async () => {
+    test.slow();
+    await accountLicensePage.goto();
+    await accountLicensePage.saveLicenseKey('1234567890');
+
+    await dashboard.goto();
+    // presence of snowflake icon indicates enterprise features are enabled
+    await dashboard.treeView.quickImport({ title: 'Snowflake' });
+  });
+});