openobserve · Shrinath-O2 · Sep 5, 2025 · Aug 13, 2025 · Aug 13, 2025 · Aug 13, 2025
diff --git a/.github/workflows/playwright.yml b/.github/workflows/playwright.yml
@@ -135,6 +135,7 @@ jobs:
             browser: "chrome"
             run_files:
               [
+                "logsDownloads.spec.js",
                 "logsqueries.spec.js",
                 "secondsPrecisionAdded.spec.js",
                 "logsqueries.cte.spec.js",
@@ -178,7 +179,7 @@ jobs:
                 "dashboard-maps.spec.js",
                 "dashboard-multi-y-axis.spec.js",
                 "dashboard-html-chart.spec.js",
-                # "visualize.spec.js",
+                "visualize.spec.js",
                 # "visualization-vrl.spec.js"
               ]
           - testfolder: "Pipelines"

diff --git a/.gitignore b/.gitignore
@@ -27,9 +27,8 @@ flamegraph.html
 /web/.env
 
 CLAUDE.md
-tests/ui-testing/playwright-tests/utils/auth/user.json
 /tests/ui-testing/test-logs
-/test-logs
-/test-results
-tests/ui-testing/playwright-tests/utils/README.md
 /tests/ui-testing/test-archives
+/test-logs
+test-results/.last-run.json
+tests/ui-testing/playwright-tests/utils/auth/user.json
diff --git a/tests/ui-testing/pages/logsPages/logsPage.js b/tests/ui-testing/pages/logsPages/logsPage.js
@@ -3,6 +3,11 @@ import { LogsQueryPage } from './logsQueryPage.js';
 import { LoginPage } from '../generalPages/loginPage.js';
 import { IngestionPage } from '../generalPages/ingestionPage.js';
 import { ManagementPage } from '../generalPages/managementPage.js';
+import * as fs from 'fs';
+import * as path from 'path';
+
+// Import testLogger for proper logging
+const testLogger = require('../../playwright-tests/utils/test-logger.js');
 
 export class LogsPage {
     constructor(page) {
@@ -1206,9 +1211,34 @@ export class LogsPage {
         return await this.loginPage.login();
     }
 
-    // Ingestion methods - delegate to IngestionPage
+    // Ingestion methods 
     async ingestLogs(orgId, streamName, logData) {
-        return await this.ingestionPage.ingestLogs(orgId, streamName, logData);
+        const basicAuthCredentials = Buffer.from(
+            `${process.env["ZO_ROOT_USER_EMAIL"]}:${process.env["ZO_ROOT_USER_PASSWORD"]}`
+        ).toString('base64');
+
+        const headers = {
+            "Authorization": `Basic ${basicAuthCredentials}`,
+            "Content-Type": "application/json",
+        };
+
+        const response = await this.page.evaluate(async ({ url, headers, orgId, streamName, logData }) => {
+            const fetchResponse = await fetch(`${url}/api/${orgId}/${streamName}/_json`, {
+                method: 'POST',
+                headers: headers,
+                body: JSON.stringify(logData)
+            });
+            return await fetchResponse.json();
+        }, {
+            url: process.env.INGESTION_URL,
+            headers: headers,
+            orgId: orgId,
+            streamName: streamName,
+            logData: logData
+        });
+
+        testLogger.debug('Ingestion API response received', { response });
+        return response;
     }
 
     // Management methods - delegate to ManagementPage
@@ -1241,6 +1271,11 @@ export class LogsPage {
         return await this.page.locator(this.queryEditor).getByRole('textbox').fill(query);
     }
 
+    async clearQueryEditor() {
+        await this.page.locator(this.queryEditor).getByRole('textbox').press('ControlOrMeta+a');
+        return await this.page.locator(this.queryEditor).getByRole('textbox').press('Backspace');
+    }
+
     async typeInQueryEditor(text) {
         return await this.page.keyboard.type(text);
     }
@@ -1904,6 +1939,122 @@ export class LogsPage {
         await this.page.locator(`[data-test="log-search-index-list-stream-toggle-${stream}"] div`).first().click();
     }
 
+    // Download-related functions
+    async setupDownloadDirectory() {
+        // Create unique directory with timestamp and random string
+        const timestamp = Date.now();
+        const randomString = Math.random().toString(36).substring(2, 8);
+        const downloadDir = path.join(process.cwd(), `temp-downloads-${timestamp}-${randomString}`);
+
+        // Create fresh directory
+        fs.mkdirSync(downloadDir, { recursive: true });
+
+        // Verify directory was created and is writable
+        expect(fs.existsSync(downloadDir)).toBe(true);
+
+        // Test write permissions by creating a test file
+        const testFile = path.join(downloadDir, 'test-write.txt');
+        fs.writeFileSync(testFile, 'test');
+        expect(fs.existsSync(testFile)).toBe(true);
+        fs.unlinkSync(testFile);
+
+        return downloadDir;
+    }
+
+    async cleanupDownloadDirectory(downloadDir) {
+        if (downloadDir && fs.existsSync(downloadDir)) {
+            const files = fs.readdirSync(downloadDir);
+            for (const file of files) {
+                fs.unlinkSync(path.join(downloadDir, file));
+            }
+            fs.rmdirSync(downloadDir);
+        }
+    }
+
+    async verifyDownload(download, expectedFileName, downloadDir) {
+        const downloadPath = path.join(downloadDir, expectedFileName);
+
+        // Save the download
+        await download.saveAs(downloadPath);
+
+        // Wait for file system to sync
+        await new Promise(resolve => setTimeout(resolve, 1000));
+
+        // Verify file exists and has content
+        expect(fs.existsSync(downloadPath)).toBe(true);
+        const stats = fs.statSync(downloadPath);
+        expect(stats.size).toBeGreaterThan(0);
+
+        // Verify it's a CSV file
+        const content = fs.readFileSync(downloadPath, 'utf8');
+        expect(content).toContain('_timestamp');
+
+        // Count rows in the CSV file
+        const rows = content.split('\n').filter(line => line.trim() !== '');
+        const rowCount = rows.length - 1; // Subtract 1 for header row
+        console.log(`Download ${expectedFileName}: ${rowCount} data rows`);
+
+        // Assert row count based on scenario
+        if (expectedFileName.includes('custom_100.csv')) {
+            expect(rowCount).toBe(100);
+        } else if (expectedFileName.includes('custom_500.csv')) {
+            expect(rowCount).toBe(500);
+        } else if (expectedFileName.includes('custom_1000.csv')) {
+            expect(rowCount).toBe(1000);
+        } else if (expectedFileName.includes('custom_5000.csv')) {
+            expect(rowCount).toBe(5000);
+        } else if (expectedFileName.includes('custom_10000.csv')) {
+            expect(rowCount).toBe(10000);
+        } else if (expectedFileName.includes('sql_limit_2000.csv')) {
+            expect(rowCount).toBe(2000);
+        } else if (expectedFileName.includes('sql_limit_2000_custom_500.csv')) {
+            expect(rowCount).toBe(500);
+        } else {
+            // For normal "Download results" downloads, we expect some data but not a specific count
+            expect(rowCount).toBeGreaterThan(0);
+        }
+
+        return downloadPath;
+    }
+
+    // Download action methods
+    async clickMoreOptionsButton() {
+        return await this.page.locator('[data-test="logs-search-bar-more-options-btn"]').click();
+    }
+
+    async clickDownloadResults() {
+        await this.page.getByText('keyboard_arrow_right').click();
+        return await this.page.locator('[data-test="logs-search-bar-more-options-btn"]').click();
+    }
+
+    async clickDownloadResultsForCustom() {
+        return await this.page.getByText('Download results for custom').click();
+    }
+
+    async clickCustomDownloadRangeSelect() {
+        return await this.page.locator('[data-test="custom-download-range-select"]').click();
+    }
+
+    async selectCustomDownloadRange(range) {
+        return await this.page.getByRole('option', { name: range, exact: true }).click();
+    }
+
+    async clickConfirmDialogOkButton() {
+        return await this.page.locator('[data-test="logs-search-bar-confirm-dialog-ok-btn"]').click();
+    }
+
+    async expectCustomDownloadDialogVisible() {
+        return await expect(this.page.getByText('Enter the initial number and')).toBeVisible();
+    }
+
+    async expectRequestFailedError() {
+        return await expect(this.page.getByText('Request failed with status')).toBeVisible();
+    }
+
+    async waitForDownload() {
+        return await this.page.waitForEvent('download');
+    }
+
     async clickAllFieldsButton() {
         return await this.page.locator('[data-test="logs-all-fields-btn"]').click();
     }