republisher/tests/test_entrypoint.py

import io
from types import SimpleNamespace

from repub.entrypoint import FeedNameFilter, entrypoint, logger, parse_args


def test_feed_name_filter_accepts_matching_item() -> None:
    item = SimpleNamespace(feed_name="nasa")
    feed_filter = FeedNameFilter({"feed_name": "nasa"})

    assert feed_filter.accepts(item) is True


def test_feed_name_filter_rejects_non_matching_item() -> None:
    item = SimpleNamespace(feed_name="gp-pod")
    feed_filter = FeedNameFilter({"feed_name": "nasa"})

    assert feed_filter.accepts(item) is False


def test_parse_args_uses_republisher_host_and_port_env_vars(monkeypatch) -> None:
    monkeypatch.setenv("REPUBLISHER_HOST", "0.0.0.0")
    monkeypatch.setenv("REPUBLISHER_PORT", "9090")

    command, args = parse_args(["serve"])

    assert command == "serve"
    assert args.host == "0.0.0.0"
    assert args.port == "9090"


def test_entrypoint_rejects_invalid_republisher_port(monkeypatch) -> None:
    monkeypatch.setenv("REPUBLISHER_PORT", "not-a-number")
    stream = io.StringIO()
    original_streams = [handler.stream for handler in logger.handlers]
    for handler in logger.handlers:
        handler.stream = stream

    try:
        exit_code = entrypoint(["serve"])
    finally:
        for handler, original_stream in zip(logger.handlers, original_streams):
            handler.stream = original_stream

    assert exit_code == 2
    assert "Invalid REPUBLISHER_PORT/--port value" in stream.getvalue()