From bc25db138edc1678b97fd7452840d7101f1cdab1 Mon Sep 17 00:00:00 2001
From: Ben Busby <contact@benbusby.com>
Date: Tue, 17 Sep 2024 20:10:22 -0600
Subject: [PATCH] Add test for vault password

---
 .../items/vault-password-dialog.html          |  6 ++--
 backend/server/html/templates/login.html      |  4 +--
 web/tests/test_single_user.py                 | 33 +++++++++++++++++++
 3 files changed, 38 insertions(+), 5 deletions(-)

diff --git a/backend/server/html/templates/items/vault-password-dialog.html b/backend/server/html/templates/items/vault-password-dialog.html
index 04bae7e..4b07ff8 100644
--- a/backend/server/html/templates/items/vault-password-dialog.html
+++ b/backend/server/html/templates/items/vault-password-dialog.html
@@ -1,4 +1,4 @@
-<dialog data-dynamic="true" id="vault-pass-dialog">
+<dialog data-testid="vault-pass-dialog" data-dynamic="true" id="vault-pass-dialog">
     {{ if . }}
     <h3>Set Vault Password</h3>
     {{ else }}
@@ -6,7 +6,7 @@ <h3>Enter Vault Password</h3>
     {{ end }}
     <hr>
     <label for="vault-pass">Password:</label>
-    <input id="vault-pass" type="password">
+    <input data-testid="vault-pass" id="vault-pass" type="password">
     <br><br>
     {{ if . }}
     <span class="small-text">
@@ -18,6 +18,6 @@ <h3>Enter Vault Password</h3>
     {{ end }}
     <div class="align-items-right">
         <button id="cancel-pass">Cancel</button>
-        <button id="submit-pass" class="accent-btn">Confirm</button>
+        <button data-testid="submit-pass" id="submit-pass" class="accent-btn">Confirm</button>
     </div>
 </dialog>
\ No newline at end of file
diff --git a/backend/server/html/templates/login.html b/backend/server/html/templates/login.html
index 8190e8c..54fe4b9 100644
--- a/backend/server/html/templates/login.html
+++ b/backend/server/html/templates/login.html
@@ -12,10 +12,10 @@ <h1>Log In</h1>
         <a id="forgot-password" href="{{ .Base.Endpoints.Forgot }}">Forgot Password</a>
     </div>
 
-    <details>
+    <details data-testid="advanced-login-options">
         <summary>Advanced</summary>
         <label for="vault-pass-cb">Set session-specific Vault password:</label>
-        <input type="checkbox" id="vault-pass-cb"><br>
+        <input data-testid="vault-pass-cb" type="checkbox" id="vault-pass-cb"><br>
     </details>
 
     {{ template "messages.html" . }}
diff --git a/web/tests/test_single_user.py b/web/tests/test_single_user.py
index c8ec73d..593a245 100644
--- a/web/tests/test_single_user.py
+++ b/web/tests/test_single_user.py
@@ -231,6 +231,39 @@ def _upload_file(folder_id: str):
     _upload_file(new_folder_id)
 
 
+def test_vault_password(browser_context: BrowserContext):
+    """Tests setting a unique session-specific vault password"""
+    global account_id
+    page: Page = browser_context.new_page()
+    page.goto(f"{base_page}/account")
+    page.on("dialog", lambda dialog: dialog.accept())
+    page.get_by_test_id("logout-btn").click()
+    expect(page).to_have_title("YeetFile - Send")
+
+    page.goto(f"{base_page}/login")
+    page.get_by_test_id("identifier").fill(account_id)
+    page.get_by_test_id("password").fill(user_password)
+    page.get_by_test_id("advanced-login-options").click()
+    page.get_by_test_id("vault-pass-cb").click()
+    page.get_by_test_id("login-btn").click()
+
+    vault_password = "my_vault_password"
+    expect(page.get_by_test_id("vault-pass-dialog")).to_be_visible()
+    page.get_by_test_id("vault-pass").fill(vault_password)
+    page.get_by_test_id("submit-pass").click()
+    expect(page).to_have_title("YeetFile - My Account")
+
+    page.goto(f"{base_page}/vault")
+    expect(page.get_by_test_id("table-body")).to_be_empty()
+    expect(page.get_by_test_id("vault-pass-dialog")).to_be_visible()
+    page.get_by_test_id("vault-pass").fill("WRONG")
+    page.get_by_test_id("submit-pass").click()
+    expect(page.get_by_test_id("vault-pass-dialog")).to_be_visible()
+    page.get_by_test_id("vault-pass").fill(vault_password)
+    page.get_by_test_id("submit-pass").click()
+    expect(page.get_by_test_id("vault-pass-dialog")).to_be_hidden()
+
+
 def test_delete_account(browser_context: BrowserContext):
     """Permanently deletes the test user account"""
     global account_id