pytest marker for tests that run full grpc server

Signed-off-by: Sreekanth <prsreekanth920@gmail.com>

Author: BulkBeing

packages/pynumaflow/Makefile

@@ -17,6 +17,12 @@ lint: format
 test:
 	uv run pytest tests/ -rA
 
+test-unit:
+	uv run pytest tests/ -m "not integration" -rA
+
+test-integration:
+	uv run pytest tests/ -m integration -rA
+
 
 setup:
 	uv sync --all-groups

packages/pynumaflow/pyproject.toml

@@ -92,6 +92,9 @@ log_cli = true
 log_cli_level = "DEBUG"
 log_cli_format = "%(asctime)s [%(levelname)8s] %(message)s (%(filename)s:%(lineno)s)"
 log_cli_date_format = "%Y-%m-%d %H:%M:%S"
+markers = [
+    "integration: tests that start real gRPC servers on Unix sockets (slower)",
+]
 
 [tool.ruff]
 line-length = 100

packages/pynumaflow/tests/accumulator/test_async_accumulator.py

@@ -23,6 +23,8 @@
     get_time_args,
 )
 
+pytestmark = pytest.mark.integration
+
 LOGGER = setup_logging(__name__)
 
 SOCK_PATH = "unix:///tmp/accumulator.sock"

packages/pynumaflow/tests/accumulator/test_async_accumulator_err.py

@@ -20,6 +20,8 @@
     get_time_args,
 )
 
+pytestmark = pytest.mark.integration
+
 LOGGER = setup_logging(__name__)
 
 SOCK_PATH = "unix:///tmp/accumulator_err.sock"

packages/pynumaflow/tests/batchmap/test_async_batch_map.py

@@ -19,6 +19,8 @@
 from pynumaflow.proto.mapper import map_pb2_grpc
 from tests.batchmap.utils import request_generator
 
+pytestmark = pytest.mark.integration
+
 LOGGER = setup_logging(__name__)
 
 listen_addr = "unix:///tmp/batch_map.sock"

packages/pynumaflow/tests/batchmap/test_async_batch_map_err.py

@@ -11,6 +11,8 @@
 from pynumaflow.proto.mapper import map_pb2_grpc
 from tests.batchmap.utils import request_generator
 
+pytestmark = pytest.mark.integration
+
 LOGGER = setup_logging(__name__)
 
 raise_error = False

packages/pynumaflow/tests/map/test_async_mapper.py

@@ -19,6 +19,8 @@
 from pynumaflow.proto.mapper import map_pb2, map_pb2_grpc
 from tests.map.utils import get_test_datums
 
+pytestmark = pytest.mark.integration
+
 LOGGER = setup_logging(__name__)
 
 # if set to true, map handler will raise a `ValueError` exception.

packages/pynumaflow/tests/mapstream/test_async_map_stream.py

@@ -15,8 +15,11 @@
 )
 from pynumaflow.proto.mapper import map_pb2_grpc
 from tests.mapstream.utils import request_generator
+
 import pytest
 
+pytestmark = pytest.mark.integration
+
 LOGGER = setup_logging(__name__)
 
 # if set to true, map handler will raise a `ValueError` exception.

packages/pynumaflow/tests/mapstream/test_async_map_stream_err.py

@@ -11,6 +11,8 @@
 from pynumaflow.proto.mapper import map_pb2_grpc
 from tests.mapstream.utils import request_generator
 
+pytestmark = pytest.mark.integration
+
 LOGGER = setup_logging(__name__)
 
 SOCK_PATH = "unix:///tmp/async_map_stream_err.sock"

packages/pynumaflow/tests/reduce/test_async_reduce.py

@@ -24,6 +24,8 @@
     get_time_args,
 )
 
+pytestmark = pytest.mark.integration
+
 logging.basicConfig(level=logging.DEBUG)
 LOGGER = logging.getLogger(__name__)