Wcc Endpoints + integration tests

[DarthMax]

Part of GDSA-109
Based on #918

[netlify]

✅ Deploy Preview for neo4j-graph-data-science-client canceled.

Name	Link
🔨 Latest commit	ced69c0
🔍 Latest deploy log	https://app.netlify.com/projects/neo4j-graph-data-science-client/deploys/6880f02ce44f2b0008191a79

[Copilot]

Pull Request Overview

This PR introduces integration tests for WCC (Weakly Connected Components) endpoints as part of GDSA-109. The implementation adds comprehensive Arrow Flight client functionality and testing infrastructure to support both Cypher and Arrow-based WCC operations.

Key changes include:

• Implementation of Arrow Flight client architecture with authentication and middleware support
• Addition of V2 API clients for job management, mutations, and write-back operations
• Integration test setup using testcontainers for both Neo4j and GDS Session environments
• Complete WCC endpoint implementations for both Cypher and Arrow protocols

Reviewed Changes

Copilot reviewed 25 out of 33 changed files in this pull request and generated 2 comments.

Show a summary per file

File	Description
requirements/dev/dev.txt	Adds testcontainers dependency for integration testing
mypy.ini	Enables pydantic mypy plugin for better type checking
graphdatascience/arrow_client/	Core Arrow Flight client implementation with authentication and middleware
graphdatascience/arrow_client/v2/	V2 API clients for jobs, mutations, data mapping, and write-back operations
graphdatascience/procedure_surface/arrow/	WCC Arrow endpoints implementation
graphdatascience/tests/unit/arrow_client/	Unit tests for Arrow client components
graphdatascience/tests/integrationV2/	Integration tests for both Cypher and Arrow WCC endpoints with testcontainer setup

Comments suppressed due to low confidence (4)

graphdatascience/tests/unit/arrow_client/middleware/test_auth_middleware.py:20

• The parameter type annotation is incorrect. The fixture parameter should be ArrowAuthentication not Mock since mock_auth fixture returns an ArrowAuthentication instance.

def auth_middleware(mock_auth: Mock) -> AuthMiddleware:

graphdatascience/tests/integrationV2/procedure_surface/cypher/test_wcc_cypher_endpoints.py:40

• The parameter type annotation is incorrect. The parameter should be WccCypherEndpoints not WccArrowEndpoints since this test file is for Cypher endpoints.

def test_wcc_stats(wcc_endpoints: WccArrowEndpoints, sample_graph: Graph) -> None:

graphdatascience/tests/integrationV2/procedure_surface/cypher/test_wcc_cypher_endpoints.py:51

• The parameter type annotation is incorrect. The parameter should be WccCypherEndpoints not WccArrowEndpoints since this test file is for Cypher endpoints.

def test_wcc_stream(wcc_endpoints: WccArrowEndpoints, sample_graph: Graph) -> None:

graphdatascience/tests/integrationV2/procedure_surface/cypher/test_wcc_cypher_endpoints.py:62

• The parameter type annotation is incorrect. The parameter should be WccCypherEndpoints not WccArrowEndpoints since this test file is for Cypher endpoints.

def test_wcc_mutate(wcc_endpoints: WccArrowEndpoints, sample_graph: Graph) -> None:

[FlorentinD]

very clean approach indeed ⭐