add test to downloads

Author: mucsi96

.vscode/launch.json

@@ -1,14 +1,21 @@
 {
-    "configurations": [
-        {
-            "type": "java",
-            "name": "Spring Boot-App<postgres-azure-backup>",
-            "request": "launch",
-            "cwd": "${workspaceFolder}/server",
-            "mainClass": "io.github.mucsi96.postgresbackuptool.App",
-            "projectName": "postgres-azure-backup",
-            "args": "",
-            "envFile": "${workspaceFolder}/.env"
-        }
-    ]
+  "configurations": [
+    {
+      "type": "java",
+      "name": "Spring Boot-App<postgres-azure-backup>",
+      "request": "launch",
+      "cwd": "${workspaceFolder}/server",
+      "mainClass": "io.github.mucsi96.postgresbackuptool.App",
+      "projectName": "postgres-azure-backup",
+      "args": "",
+      "envFile": "${workspaceFolder}/.env"
+    },
+    {
+      "name": "Python: Unittest",
+      "type": "debugpy",
+      "request": "launch",
+      "console": "integratedTerminal",
+      "justMyCode": true
+    }
+  ]
 }

TODO.md

@@ -1,5 +1,3 @@
 - migrate to angular 19 and resource
 - drop @mucsi96/ui-elements in favor of angular/material
-- add possibility to additionaly create plain dumps
-- show dump format on UI
 - deploy backup job with help

client/src/app/backups/backups.component.html

@@ -43,6 +43,7 @@ <h2 bt>
             mat-icon-button
             (click)="downloadBackup({ type: 'archive' })"
             class="download"
+            aria-label="Download archive"
           >
             <mat-icon>folder_zip</mat-icon>
           </button>
@@ -51,6 +52,7 @@ <h2 bt>
             mat-icon-button
             (click)="downloadBackup({ type: 'plain' })"
             class="download"
+            aria-label="Download plain dump"
           >
             <mat-icon>description</mat-icon>
           </button>

test/test_backups.py

@@ -1,10 +1,12 @@
 from datetime import timedelta
+import re
 from playwright.sync_api import Page, expect
 from typing import Any, Dict, List
 from utils import (
     cleanup_backups,
     create_backup,
     extract_table_data,
+    mock_window_open,
 )
 
 
@@ -61,7 +63,8 @@ def test_creates_backup(page: Page):
     cleanup_backups()
     page.goto("http://localhost:8080")
     page.get_by_role("button", name="Backup").click()
-    expect(page.get_by_role("status").filter(has_text="Backup created")).to_be_visible()
+    expect(page.get_by_role("status").filter(
+        has_text="Backup created")).to_be_visible()
     page.get_by_text("db1").click()
     table_data = list_without_keys(
         extract_table_data(page.locator(":text('Backups') + table")),
@@ -83,7 +86,8 @@ def test_creates_backup_with_retention(page: Page):
     retention_period_input = page.get_by_label("Retention period")
     retention_period_input.fill("7")
     page.get_by_role("button", name="Backup").click()
-    expect(page.get_by_role("status").filter(has_text="Backup created")).to_be_visible()
+    expect(page.get_by_role("status").filter(
+        has_text="Backup created")).to_be_visible()
     page.get_by_text("db1").click()
     table_data = list_without_keys(
         extract_table_data(page.locator(":text('Backups') + table")),
@@ -155,3 +159,51 @@ def test_cleans_up_outdated_backups(page: Page):
             "Retention": "31 days",
         },
     ]
+
+
+def test_downloads_backup_archive(page: Page):
+    cleanup_backups()
+    page.goto("http://localhost:8080")
+    page.get_by_role("button", name="Backup").click()
+    expect(page.get_by_role("status").filter(
+        has_text="Backup created")).to_be_visible()
+    page.get_by_text("db1").click()
+    page.locator(":text('Backups') + table").get_by_text("1 day").click()
+
+    mock_window_open(page)
+
+    with page.expect_download() as download_info:
+        page.get_by_role("button", name="Download archive").click()
+
+    assert re.match(
+        r"https://localhost:8081/devstoreaccount1/backups/db1%2F.*-.*.9.1.pgdump", download_info.value.url
+    )
+
+def test_downloads_backup_plain(page: Page):
+    cleanup_backups()
+    page.goto("http://localhost:8080")
+    page.get_by_role("button", name="Backup").click()
+    expect(page.get_by_role("status").filter(
+        has_text="Backup created")).to_be_visible()
+    page.get_by_text("db1").click()
+    page.locator(":text('Backups') + table").get_by_text("1 day").click()
+
+    mock_window_open(page)
+
+    with page.expect_download() as download_info:
+        page.get_by_role("button", name="Download plain dump").click()
+
+    assert re.match(
+        r"https://localhost:8081/devstoreaccount1/backups/db1%2F.*-.*.9.1.sql", download_info.value.url
+    )
+
+    response = page.request.get(download_info.value.url)
+    assert response.status == 200
+    assert "CREATE SCHEMA test1" in response.text()
+    assert "COPY test1.fruites (name) FROM stdin" in response.text()
+    assert "Apple" in response.text()
+    assert "Orange" in response.text()
+    assert "Banana" in response.text()
+    assert "Rasberry" in response.text()
+
+

test/utils.py

@@ -176,3 +176,16 @@ def get_db1_tables():
     tables = [table[0] for table in cur1.fetchall()]
     cur1.close()
     return tables
+
+
+def mock_window_open(page):
+    page.evaluate(
+        """
+        window.open = (url) => {
+            if (url && url.startsWith("https://blobstorage:10000")) {
+                url = url.replace("https://blobstorage:10000", "https://localhost:8081");
+                window.location.href = url;
+            }
+        };
+        """
+    )