queen-mq 0.6.4 → 0.12.0

package/test-v2/maintenance.js

@@ -63,7 +63,7 @@ export async function test_maintenance_mode(client = null) {
     const consumePromise = (async () => {
       while (consumerActive) {
         try {
-          const messages = await queen.queue(QUEUE_NAME).batch(10).wait(true).limit(1).each().pop();
+          const messages = await queen.queue(QUEUE_NAME).batch(10).wait(false).limit(1).each().pop();
           
           if (messages && messages.length > 0) {
             receivedCount += messages.length;
@@ -71,12 +71,14 @@ export async function test_maintenance_mode(client = null) {
             
             // Acknowledge messages
             await queen.ack(messages);
+            
           }
         } catch (err) {
           if (consumerActive) {
             console.error('  ❌ Consumer error:', err.message);
           }
         }
+        await sleep(10);
       }
     })();
     

package/test-v2/push.js

@@ -115,35 +115,6 @@ export async function pushBufferedMessage(client) {
     return { success: true }
 }
 
-export async function pushMaxQueueSize(client) {
-    const queue = await client.queue('test-queue-max-size').config({ maxSize: 10 }).create()
-    if (!queue.configured) {
-        return { success: false, message: 'Queue not created' }
-    }
-
-    // Necessary to create a partition consumer to trigger the capacity check
-    const popToCreateParititonConsumer = await client
-    .queue('test-queue-max-size')
-    .batch(1)
-    .wait(false)
-    .pop()    
-
-    try {
-        for (let i = 0; i < 20; i++) {
-        const result = await client
-        .queue('test-queue-max-size')
-        .push([{ data: { message: i } }])
-        }
-    } catch (error) {
-        if (!error.message.includes(' Queue "test-queue-max-size" would exceed max capacity (10).')) {
-            return { success: true, message: 'Correct error message' }
-        } else {
-            return { success: false, message: 'Incorrect error message' }
-        }
-    }
-    return { success: false }
-}
-
 export async function pushDelayedMessage(client) {
     const queue = await client.queue('test-queue-delayed')
     .config({ delayedProcessing: 2 })

package/test-v2/run.js

@@ -18,11 +18,26 @@ import * as bufferingTests from './ai_buffering.js'
 import * as priorityTests from './ai_priority.js'
 import * as ttlRetentionTests from './ai_ttl_retention.js'
 import * as mixedScenariosTests from './ai_mixed_scenarios.js'
+import { LoadBalancer } from '../client-v2/http/LoadBalancer.js';
 
 
 // Test configuration
-export const TEST_CONFIG = {
-  baseUrls: ['http://localhost:6632'],
+
+export const TEST_CONFIG_SINGLE = {
+    baseUrls: ['http://localhost:6632'],
+    loadBalancingStrategy: 'affinity',
+    dbConfig: {
+      host: process.env.PG_HOST || 'localhost',
+      port: process.env.PG_PORT || 5432,
+      database: process.env.PG_DB || 'postgres',
+      user: process.env.PG_USER || 'postgres',
+      password: process.env.PG_PASSWORD || 'postgres'
+    }
+  };
+
+export const TEST_CONFIG_MULTIPLE = {
+  baseUrls: ['http://localhost:6632','http://localhost:6633'],
+  loadBalancingStrategy: 'round-robin',
   dbConfig: {
     host: process.env.PG_HOST || 'localhost',
     port: process.env.PG_PORT || 5432,
@@ -32,6 +47,9 @@ export const TEST_CONFIG = {
   }
 };
 
+export const TEST_CONFIG = process.env.TEST_CONFIG === 'multiple' ? TEST_CONFIG_MULTIPLE : TEST_CONFIG_SINGLE;
+console.log('TEST_CONFIG:', TEST_CONFIG);
+
 // Global test state
 export let dbPool;  
 
@@ -82,7 +100,10 @@ export const cleanupTestData = async () => {
   };
 
 async function main() {
-    const client = new Queen(TEST_CONFIG.baseUrls)
+    const client = new Queen({
+        urls: TEST_CONFIG.baseUrls,
+        loadBalancingStrategy: TEST_CONFIG.loadBalancingStrategy
+    })
     await initDb()
 
     // Separate human and AI tests