Function main

scripts/test_memory.py:211–279

()

    return results

def main():
    parser = argparse.ArgumentParser(description='Test jcode memory system')
    parser.add_argument('--fresh', action='store_true', help='Start fresh server for testing')
    parser.add_argument('--provider', choices=['claude', 'openai'], help='Test specific provider only')
    parser.add_argument('--socket', help='Custom debug socket path')
    args = parser.parse_args()

    providers = [args.provider] if args.provider else ['claude', 'openai']

    proc = None
    if args.fresh:
        log_section("Starting fresh test server...")
        test_socket = '/tmp/jcode-memory-test.sock'
        debug_socket = test_socket.replace('.sock', '-debug.sock')

        for s in [test_socket, debug_socket]:
            if os.path.exists(s):
                os.remove(s)

        env = os.environ.copy()
        env['JCODE_DEBUG_CONTROL'] = '1'
        env['JCODE_SOCKET'] = test_socket

        proc = subprocess.Popen(
            ['jcode', 'serve'],
            env=env,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE
        )

        for i in range(100):
            if os.path.exists(debug_socket):
                break
            time.sleep(0.1)
        else:
            proc.terminate()
            log_fail("Server failed to start")
            sys.exit(1)

        log_pass(f"Server started (PID {proc.pid})")
        socket_path = debug_socket
    else:
        socket_path = args.socket or '/run/user/1000/jcode-debug.sock'
        if not os.path.exists(socket_path):
            log_fail(f"Debug socket not found: {socket_path}")
            log("Use --fresh to start a test server, or ensure jcode is running")
            sys.exit(1)

    client = DebugSocketClient(socket_path)

    try:
        client.connect()
        results = run_tests(client, providers)

        log_section("TEST RESULTS")
        total = results["passed"] + results["failed"]
        log(f"Passed: {results['passed']}/{total}", GREEN if results["failed"] == 0 else YELLOW)
        if results["failed"] > 0: