update

test/helpers/server.loader.ts

@@ -133,7 +133,11 @@ export async function startTestServer(config: ITestServerConfig): Promise<ITestS
     cleanupInterval: 300000,
     auth: serverConfig.authRequired ? {
       required: true,
-      methods: ['PLAIN', 'LOGIN'] as ('PLAIN' | 'LOGIN' | 'OAUTH2')[]
+      methods: ['PLAIN', 'LOGIN'] as ('PLAIN' | 'LOGIN' | 'OAUTH2')[],
+      validateUser: async (username: string, password: string) => {
+        // Test server accepts these credentials
+        return username === 'testuser' && password === 'testpass';
+      }
     } : undefined
   };
 
@@ -284,4 +288,46 @@ export function createTestEmail(options: {
     date: new Date(),
     messageId: `<${Date.now()}@test.example.com>`
   };
+}
+
+/**
+ * Simple test server for custom protocol testing
+ */
+export interface ISimpleTestServer {
+  server: any;
+  hostname: string;
+  port: number;
+}
+
+export async function createTestServer(options: {
+  onConnection?: (socket: any) => void | Promise<void>;
+  port?: number;
+  hostname?: string;
+}): Promise<ISimpleTestServer> {
+  const hostname = options.hostname || 'localhost';
+  const port = options.port || await getAvailablePort();
+  
+  const server = plugins.net.createServer((socket) => {
+    if (options.onConnection) {
+      const result = options.onConnection(socket);
+      if (result && typeof result.then === 'function') {
+        result.catch(error => {
+          console.error('Error in onConnection handler:', error);
+          socket.destroy();
+        });
+      }
+    }
+  });
+  
+  return new Promise((resolve, reject) => {
+    server.listen(port, hostname, () => {
+      resolve({
+        server,
+        hostname,
+        port
+      });
+    });
+    
+    server.on('error', reject);
+  });
 }