fix publisher plugin tests

Author: Bojan131

apps/agent/package.json

@@ -21,7 +21,7 @@
     "script:createToken": "node dist/scripts/createToken.js",
     "test": "PLAYWRIGHT_JUNIT_OUTPUT_NAME=DKG_Node_UI_Tests.xml npx playwright test spec/testUI.spec.js --grep '@gh_actions' --reporter=list,html,junit",
     "test:e2e": "PLAYWRIGHT_JUNIT_OUTPUT_NAME=DKG_Node_UI_Tests.xml npx playwright test spec/testUI.spec.js --reporter=list,html,junit",
-    "test:e2e:publisher": "PLAYWRIGHT_JUNIT_OUTPUT_NAME=DKG_Node_Publisher_Tests.xml npx playwright test spec/publisherPluginTest.spec.js --reporter=list,html,junit",
+    "test:e2e:publisher": "node tests/e2e/setup/cleanup.js && PLAYWRIGHT_JUNIT_OUTPUT_NAME=DKG_Node_Publisher_Tests.xml npx playwright test spec/publisherPluginTest.spec.js --reporter=list,html,junit",
     "test:install": "npx playwright install",
     "test:integration": "NODE_OPTIONS='--import tsx' mocha 'tests/integration/**/*.spec.ts' --require tests/integration/setup/global-setup.ts",
     "test:ragas": "NODE_OPTIONS='--import tsx' tsx tests/ragas/evaluate.ts",

apps/agent/tests/e2e/pages/chatbotPage.js

@@ -9,23 +9,35 @@ class ChatbotPage {
     this.btn_continue = this.page.getByTestId("tool-continue-button");
   }
 
-  async waitForChatReady() {
-    // Wait for chat input to be enabled and ready (no timeout)
-    await this.page.waitForFunction(
-      () => {
-        const chatInput = document.querySelector('[data-testid="chat-input"]');
-        return chatInput && !chatInput.disabled && chatInput.value === "";
-      },
-      {},
-      { timeout: 0 },
-    );
+  async waitForChatReady(timeoutMs = 30000) {
+    // Wait for chat input to be enabled and ready
+    try {
+      await this.page.waitForFunction(
+        () => {
+          const chatInput = document.querySelector('[data-testid="chat-input"]');
+          return chatInput && !chatInput.disabled && chatInput.value === "";
+        },
+        {},
+        { timeout: timeoutMs },
+      );
+    } catch (error) {
+      throw new Error(`Chat input not ready after ${timeoutMs}ms: ${error.message}`);
+    }
   }
 
-  async waitForResponse(questionText = "") {
+  async waitForResponse(questionText = "", timeoutMs = 60000) {
     // Wait for DKG Node to finish processing and provide a real response (not "Thinking...")
-    const checkInterval = 5000; // Check every 5 seconds
+    const checkInterval = 2000; // Check every 2 seconds
+    const startTime = Date.now();
 
     while (true) {
+      // Check if we've exceeded the timeout
+      if (timeoutMs > 0 && Date.now() - startTime > timeoutMs) {
+        throw new Error(
+          `waitForResponse timed out after ${timeoutMs}ms waiting for response to: "${questionText}"`
+        );
+      }
+
       try {
         const messages = await this.page.$$eval(
           '[data-testid="chat-message-text"]',
@@ -39,7 +51,7 @@ class ChatbotPage {
           const isNotThinking = !lastMessage.trim().startsWith("Thinking");
           const isNotSameAsQuestion =
             lastMessage.trim() !== questionText.trim();
-          const hasContent = lastMessage.length > 0;
+          const hasContent = lastMessage.length > 10; // Require at least some content
 
           const shouldComplete =
             hasContent && isNotThinking && isNotSameAsQuestion;
@@ -52,6 +64,8 @@ class ChatbotPage {
         // Wait before next check
         await this.page.waitForTimeout(checkInterval);
       } catch (error) {
+        // Log error but continue trying
+        console.log(`   ⚠️  waitForResponse error: ${error.message}`);
         await this.page.waitForTimeout(checkInterval);
       }
     }

apps/agent/tests/e2e/setup/cleanup.js

@@ -0,0 +1,74 @@
+/**
+ * Pre-test cleanup script
+ * Runs BEFORE Playwright starts the dev server to ensure clean state
+ */
+
+const Redis = require("ioredis");
+const mysql = require("mysql2/promise");
+const dotenv = require("dotenv");
+const path = require("path");
+
+// Load test environment variables
+dotenv.config({
+  path: path.join(__dirname, "..", ".env.testing.testnet.local"),
+  override: true,
+});
+
+async function cleanup() {
+  console.log("\n🧹 Pre-test cleanup: Clearing Redis and Database...");
+
+  // Clear Redis
+  try {
+    const redis = new Redis({
+      host: process.env.REDIS_HOST || "localhost",
+      port: parseInt(process.env.REDIS_PORT || "6379"),
+      retryStrategy: () => null,
+    });
+
+    await redis.flushall();
+    console.log("✅ Redis cleared (all BullMQ jobs removed)");
+    await redis.quit();
+  } catch (error) {
+    console.warn("⚠️  Redis cleanup failed:", error.message);
+    process.exit(1); // Fail the test if Redis can't be cleared
+  }
+
+  // Clear MySQL
+  try {
+    const dbUrl = process.env.DKGP_DATABASE_URL || "mysql://root:@127.0.0.1:3306/dkg_publisher_db";
+    const match = dbUrl.match(/mysql:\/\/([^:]+):([^@]*)@([^:]+):(\d+)\/(.+)/);
+    
+    if (!match) {
+      throw new Error("Invalid database URL format");
+    }
+
+    const [, user, password, host, port, database] = match;
+    
+    const connection = await mysql.createConnection({
+      host,
+      port: parseInt(port),
+      user,
+      password: password || undefined,
+      database,
+    });
+
+    await connection.execute("DELETE FROM publishing_attempts");
+    await connection.execute("DELETE FROM assets");
+    await connection.execute("UPDATE wallets SET is_locked = 0, locked_by = NULL, locked_at = NULL");
+    await connection.execute("ALTER TABLE assets AUTO_INCREMENT = 1");
+    
+    console.log("✅ Database cleared (assets, attempts, wallets unlocked)");
+    console.log("✅ Asset ID counter reset to 1");
+    await connection.end();
+  } catch (error) {
+    console.warn("⚠️  Database cleanup failed:", error.message);
+    process.exit(1); // Fail the test if DB can't be cleared
+  }
+
+  console.log("🚀 Cleanup complete - environment ready for fresh test\n");
+}
+
+cleanup().catch((error) => {
+  console.error("💥 Cleanup script failed:", error);
+  process.exit(1);
+});