Class TestBaseAPIEdgeCases

tests/models/test_base_api.py:251–291

Edge case tests for BaseAPI.

class TestBaseAPIEdgeCases:
    """Edge case tests for BaseAPI."""

    def test_generate_with_none_message(self, api):
        """Test generating with None in messages."""
        # This might raise or handle gracefully depending on implementation
        try:
            results = api.generate([None])
        except (TypeError, AssertionError):
            pass  # Expected

    def test_generate_with_empty_content(self, api):
        """Test generating with empty content."""
        results = api.generate([""])
        assert isinstance(results, list)

    def test_retry_on_failure(self):
        """Test retry logic on failure."""
        from datastudio.models.base import BaseAPI

        class FailingAPI(BaseAPI):
            def __init__(self):
                super().__init__(retry=2, wait=0.01)
                self.call_count = 0

            def generate_inner(self, inputs, **kwargs):
                self.call_count += 1
                if self.call_count < 2:
                    return 1, None, None  # Failure
                return 0, "success", None

        api = FailingAPI()
        result = api.process_single_message("test", {})
        assert result == "success"
        assert api.call_count == 2

    def test_concurrent_generation(self, api):
        """Test concurrent generation."""
        messages = ["msg1", "msg2", "msg3", "msg4"]
        results = api.generate(messages)
        assert len(results) == 4


class TestBaseAPIAbstract: