Function test_cleanup_agent_network

src/docker/proxy_manager.rs:1197–1223

()

    #[tokio::test]
    async fn test_cleanup_agent_network() {
        let mock_client = Arc::new(TrackedDockerClient::default());
        let ctx = AppContext::builder().build();

        let manager = ProxyManager::new(
            mock_client.clone(),
            ctx.tsk_env(),
            ContainerEngine::Docker,
            None,
        );
        let proxy_config = default_proxy_config();
        let container_name = proxy_config.proxy_container_name();
        manager
            .cleanup_agent_network(&container_name, "tsk-agent-test-123")
            .await;

        let disconnect_calls = mock_client.disconnect_network_calls.lock().unwrap();
        assert_eq!(disconnect_calls.len(), 1);
        assert_eq!(
            disconnect_calls[0],
            (container_name, "tsk-agent-test-123".to_string())
        );

        let remove_calls = mock_client.remove_network_calls.lock().unwrap();
        assert_eq!(remove_calls.len(), 1);
        assert_eq!(remove_calls[0], "tsk-agent-test-123");
    }

    #[tokio::test]
    async fn test_cleanup_agent_network_handles_errors_gracefully() {