From 7ded5fc6c966ce009ce469c1349ac8f8e72b1779 Mon Sep 17 00:00:00 2001
From: Francis Cao <franciscao@gmail.com>
Date: Thu, 29 Feb 2024 17:40:49 -0800
Subject: [PATCH] add prelim website tests

---
 cypress/e2e/login.cy.ts                       |  8 +---
 cypress/e2e/website.cy.ts                     | 37 +++++++++++++++++++
 cypress/support/e2e.ts                        | 21 ++++++++++-
 cypress/support/index.d.ts                    |  5 +++
 .../settings/websites/WebsiteAddButton.tsx    |  2 +-
 .../settings/websites/WebsiteAddForm.tsx      |  9 ++++-
 .../settings/websites/WebsitesTable.tsx       |  2 +-
 7 files changed, 73 insertions(+), 11 deletions(-)
 create mode 100644 cypress/e2e/website.cy.ts

diff --git a/cypress/e2e/login.cy.ts b/cypress/e2e/login.cy.ts
index 4bd6a6b7..5c833113 100644
--- a/cypress/e2e/login.cy.ts
+++ b/cypress/e2e/login.cy.ts
@@ -1,15 +1,11 @@
-describe('Login test', () => {
+describe('Login tests', () => {
   it(
     'logs user in with correct credentials and logs user out',
     {
       defaultCommandTimeout: 10000,
     },
     () => {
-      cy.visit('/login');
-      cy.dataCy('input-username').type(Cypress.env('umami_user'));
-      cy.dataCy('input-password').type(Cypress.env('umami_password'));
-      cy.dataCy('button-submit').click();
-      cy.url().should('eq', Cypress.config().baseUrl + '/dashboard');
+      cy.login(Cypress.env('umami_user'), Cypress.env('umami_password'));
       cy.dataCy('button-profile').click();
       cy.dataCy('item-logout').click();
       cy.url().should('eq', Cypress.config().baseUrl + '/login');
diff --git a/cypress/e2e/website.cy.ts b/cypress/e2e/website.cy.ts
new file mode 100644
index 00000000..86f703eb
--- /dev/null
+++ b/cypress/e2e/website.cy.ts
@@ -0,0 +1,37 @@
+describe('Website tests', () => {
+  beforeEach(() => {
+    cy.login(Cypress.env('umami_user'), Cypress.env('umami_password'));
+    cy.get('a[href="/settings"]').click();
+    cy.url().should('include', '/settings/websites');
+  });
+
+  it.skip('Add a website', () => {
+    cy.dataCy('button-website-add').click();
+    cy.contains(/Add website/i).should('be.visible');
+    cy.dataCy('input-name').click().type('Test Website', {
+      delay: 100,
+    });
+    cy.dataCy('input-domain').click().type('testwebsite.com');
+    cy.dataCy('button-submit').click();
+    cy.get('td[label="Name"]').should('contain.text', 'Test Website');
+    cy.get('td[label="Domain"]').should('contain.text', 'testwebsite.com');
+  });
+
+  it('Test tracking script content', () => {
+    cy.dataCy('link-button-edit').first().click();
+    cy.contains(/Tracking code/i).should('be.visible');
+    cy.get('div')
+      .contains(/Tracking code/i)
+      .click();
+    cy.get('textarea').should('contain.text', Cypress.config().baseUrl + '/script2.js');
+  });
+
+  it('Test tracking script content', () => {
+    cy.dataCy('link-button-edit').first().click();
+    cy.contains(/Tracking code/i).should('be.visible');
+    cy.get('div')
+      .contains(/Tracking code/i)
+      .click();
+    cy.get('textarea').should('contain.text', Cypress.config().baseUrl + '/script.js');
+  });
+});
diff --git a/cypress/support/e2e.ts b/cypress/support/e2e.ts
index 2ee2ed8e..faca6903 100644
--- a/cypress/support/e2e.ts
+++ b/cypress/support/e2e.ts
@@ -1,5 +1,24 @@
 /// <reference types="cypress" />
 
-Cypress.Commands.add('dataCy', value => {
+Cypress.Commands.add('dataCy', (value: string) => {
   return cy.get(`[data-cy=${value}]`);
 });
+
+Cypress.Commands.add('login', (username: string, password: string) => {
+  cy.session(
+    [username, password],
+    () => {
+      cy.visit('/login');
+      cy.dataCy('input-username').type(username);
+      cy.dataCy('input-password').type(password);
+      cy.dataCy('button-submit').click();
+      cy.url().should('eq', Cypress.config().baseUrl + '/dashboard');
+    },
+    {
+      validate: () => {
+        cy.visit('/profile');
+      },
+    },
+  );
+  cy.visit('/dashboard');
+});
diff --git a/cypress/support/index.d.ts b/cypress/support/index.d.ts
index da94c844..1da81f1a 100644
--- a/cypress/support/index.d.ts
+++ b/cypress/support/index.d.ts
@@ -7,5 +7,10 @@ declare namespace Cypress {
      * @example cy.dataCy('greeting')
      */
     dataCy(value: string): Chainable<JQuery<HTMLElement>>;
+    /**
+     * Custom command to login user into the app.
+     * @example cy.login('admin', 'password)
+     */
+    login(username: string, password: string): Chainable<JQuery<HTMLElement>>;
   }
 }
diff --git a/src/app/(main)/settings/websites/WebsiteAddButton.tsx b/src/app/(main)/settings/websites/WebsiteAddButton.tsx
index 2c99f22b..5dd60b76 100644
--- a/src/app/(main)/settings/websites/WebsiteAddButton.tsx
+++ b/src/app/(main)/settings/websites/WebsiteAddButton.tsx
@@ -15,7 +15,7 @@ export function WebsiteAddButton({ teamId, onSave }: { teamId: string; onSave?:
 
   return (
     <ModalTrigger>
-      <Button variant="primary">
+      <Button data-cy="button-website-add" variant="primary">
         <Icon>
           <Icons.Plus />
         </Icon>
diff --git a/src/app/(main)/settings/websites/WebsiteAddForm.tsx b/src/app/(main)/settings/websites/WebsiteAddForm.tsx
index 88cbb4ff..acfd2391 100644
--- a/src/app/(main)/settings/websites/WebsiteAddForm.tsx
+++ b/src/app/(main)/settings/websites/WebsiteAddForm.tsx
@@ -38,12 +38,17 @@ export function WebsiteAddForm({
   return (
     <Form onSubmit={handleSubmit} error={error}>
       <FormRow label={formatMessage(labels.name)}>
-        <FormInput name="name" rules={{ required: formatMessage(labels.required) }}>
+        <FormInput
+          data-cy="input-name"
+          name="name"
+          rules={{ required: formatMessage(labels.required) }}
+        >
           <TextField autoComplete="off" />
         </FormInput>
       </FormRow>
       <FormRow label={formatMessage(labels.domain)}>
         <FormInput
+          data-cy="input-domain"
           name="domain"
           rules={{
             required: formatMessage(labels.required),
@@ -54,7 +59,7 @@ export function WebsiteAddForm({
         </FormInput>
       </FormRow>
       <FormButtons flex>
-        <SubmitButton variant="primary" disabled={false}>
+        <SubmitButton data-cy="button-submit" variant="primary" disabled={false}>
           {formatMessage(labels.save)}
         </SubmitButton>
         {onClose && (
diff --git a/src/app/(main)/settings/websites/WebsitesTable.tsx b/src/app/(main)/settings/websites/WebsitesTable.tsx
index cbd5e3a7..1b1ff498 100644
--- a/src/app/(main)/settings/websites/WebsitesTable.tsx
+++ b/src/app/(main)/settings/websites/WebsitesTable.tsx
@@ -36,7 +36,7 @@ export function WebsitesTable({
               <>
                 {allowEdit && (
                   <LinkButton href={renderTeamUrl(`/settings/websites/${websiteId}`)}>
-                    <Icon>
+                    <Icon data-cy="link-button-edit">
                       <Icons.Edit />
                     </Icon>
                     <Text>{formatMessage(labels.edit)}</Text>