lib-x17-fintech 2.1.3__py3-none-any.whl

xfintech/data/relay/tests/test_class_relayclientlike_all.py

@@ -0,0 +1,500 @@
+"""
+Test suite for RelayClientLike Protocol
+Tests cover protocol definition, compliance checking, and structural typing
+"""
+
+from typing import Any, Dict
+
+from xfintech.data.relay.clientlike import RelayClientLike
+
+# ============================================================================
+# Helper Classes for Testing
+# ============================================================================
+
+
+class CompliantClient:
+    """Class that implements RelayClientLike protocol"""
+
+    def __init__(self):
+        self.url = "https://relay.example.com"
+        self.secret = "test-secret"
+        self.timeout = 180
+
+    def call(self) -> Any:
+        return {"status": "success"}
+
+
+class PartialClient:
+    """Class that partially implements RelayClientLike protocol"""
+
+    def __init__(self):
+        self.url = "https://relay.example.com"
+        self.secret = "test-secret"
+        # Missing timeout attribute
+
+    def call(self) -> Any:
+        return {"status": "partial"}
+
+
+class NonCompliantClient:
+    """Class that doesn't implement RelayClientLike protocol"""
+
+    def __init__(self):
+        self.endpoint = "https://relay.example.com"
+        self.key = "test-key"
+
+    def execute(self) -> Any:
+        return {"status": "non-compliant"}
+
+
+class MinimalClient:
+    """Minimal implementation of RelayClientLike"""
+
+    url: str = "https://minimal.example.com"
+    secret: str = "minimal-secret"
+    timeout: int = 60
+
+    def call(self):
+        return None
+
+
+# ============================================================================
+# Protocol Compliance Tests
+# ============================================================================
+
+
+def test_relayclientlike_compliant_client():
+    """Test that compliant class is recognized as RelayClientLike"""
+    client = CompliantClient()
+    assert isinstance(client, RelayClientLike)
+
+
+def test_relayclientlike_partial_client():
+    """Test that partial implementation is not recognized as RelayClientLike"""
+    client = PartialClient()
+    # Missing timeout, so should not be fully compliant
+    assert isinstance(client, RelayClientLike) is False
+
+
+def test_relayclientlike_non_compliant_client():
+    """Test that non-compliant class is not recognized as RelayClientLike"""
+    client = NonCompliantClient()
+    assert isinstance(client, RelayClientLike) is False
+
+
+def test_relayclientlike_minimal_client():
+    """Test minimal implementation is recognized as RelayClientLike"""
+    client = MinimalClient()
+    assert isinstance(client, RelayClientLike)
+
+
+# ============================================================================
+# Protocol Attribute Tests
+# ============================================================================
+
+
+def test_relayclientlike_has_url_attribute():
+    """Test RelayClientLike protocol requires url attribute"""
+    client = CompliantClient()
+    assert hasattr(client, "url")
+    assert isinstance(client.url, str)
+
+
+def test_relayclientlike_has_secret_attribute():
+    """Test RelayClientLike protocol requires secret attribute"""
+    client = CompliantClient()
+    assert hasattr(client, "secret")
+    assert isinstance(client.secret, str)
+
+
+def test_relayclientlike_has_timeout_attribute():
+    """Test RelayClientLike protocol requires timeout attribute"""
+    client = CompliantClient()
+    assert hasattr(client, "timeout")
+    assert isinstance(client.timeout, int)
+
+
+def test_relayclientlike_has_call_method():
+    """Test RelayClientLike protocol requires call method"""
+    client = CompliantClient()
+    assert hasattr(client, "call")
+    assert callable(client.call)
+
+
+# ============================================================================
+# Protocol Method Tests
+# ============================================================================
+
+
+def test_relayclientlike_call_returns_any():
+    """Test call method can return any type"""
+    client = CompliantClient()
+    result = client.call()
+    assert result is not None
+
+
+def test_relayclientlike_call_execution():
+    """Test call method executes correctly"""
+    client = CompliantClient()
+    result = client.call()
+    assert result == {"status": "success"}
+
+
+# ============================================================================
+# Protocol Usage Tests
+# ============================================================================
+
+
+def test_relayclientlike_as_type_annotation():
+    """Test RelayClientLike can be used as type annotation"""
+
+    def process_client(client: RelayClientLike) -> Any:
+        return client.call()
+
+    client = CompliantClient()
+    result = process_client(client)
+
+    assert result == {"status": "success"}
+
+
+def test_relayclientlike_type_checking():
+    """Test RelayClientLike enables structural type checking"""
+
+    def get_client_info(client: RelayClientLike) -> Dict[str, Any]:
+        return {"url": client.url, "timeout": client.timeout}
+
+    client = CompliantClient()
+    info = get_client_info(client)
+
+    assert info["url"] == "https://relay.example.com"
+    assert info["timeout"] == 180
+
+
+def test_relayclientlike_duck_typing():
+    """Test RelayClientLike supports duck typing"""
+
+    class DuckTypedClient:
+        def __init__(self):
+            self.url = "https://duck.example.com"
+            self.secret = "duck-secret"
+            self.timeout = 120
+
+        def call(self):
+            return "duck result"
+
+    client = DuckTypedClient()
+    assert isinstance(client, RelayClientLike)
+
+
+# ============================================================================
+# Protocol Instance Tests
+# ============================================================================
+
+
+def test_relayclientlike_multiple_implementations():
+    """Test multiple classes can implement RelayClientLike"""
+
+    class Client1:
+        url = "https://client1.example.com"
+        secret = "secret1"
+        timeout = 100
+
+        def call(self):
+            return 1
+
+    class Client2:
+        url = "https://client2.example.com"
+        secret = "secret2"
+        timeout = 200
+
+        def call(self):
+            return 2
+
+    client1 = Client1()
+    client2 = Client2()
+
+    assert isinstance(client1, RelayClientLike)
+    assert isinstance(client2, RelayClientLike)
+
+
+def test_relayclientlike_inheritance_not_required():
+    """Test classes don't need to inherit from RelayClientLike"""
+
+    class IndependentClient:
+        url = "https://independent.example.com"
+        secret = "independent-secret"
+        timeout = 150
+
+        def call(self):
+            return "independent"
+
+    client = IndependentClient()
+    # Should be recognized as RelayClientLike due to structural typing
+    assert isinstance(client, RelayClientLike)
+
+
+# ============================================================================
+# Edge Case Tests
+# ============================================================================
+
+
+def test_relayclientlike_with_extra_attributes():
+    """Test class with extra attributes is still RelayClientLike compliant"""
+
+    class ExtendedClient:
+        url = "https://extended.example.com"
+        secret = "extended-secret"
+        timeout = 200
+        extra_attr = "extra"
+        another_attr = 42
+
+        def call(self):
+            return "extended"
+
+        def extra_method(self):
+            return "extra"
+
+    client = ExtendedClient()
+    assert isinstance(client, RelayClientLike)
+    assert hasattr(client, "extra_attr")
+
+
+def test_relayclientlike_with_properties():
+    """Test class with properties can be RelayClientLike compliant"""
+
+    class PropertyClient:
+        def __init__(self):
+            self._url = "https://property.example.com"
+            self._secret = "property-secret"
+            self._timeout = 180
+
+        @property
+        def url(self):
+            return self._url
+
+        @property
+        def secret(self):
+            return self._secret
+
+        @property
+        def timeout(self):
+            return self._timeout
+
+        def call(self):
+            return "property result"
+
+    client = PropertyClient()
+    assert isinstance(client, RelayClientLike)
+
+
+def test_relayclientlike_call_with_arguments():
+    """Test RelayClientLike call method can accept arguments"""
+
+    class ArgClient:
+        url = "https://arg.example.com"
+        secret = "arg-secret"
+        timeout = 180
+
+        def call(self, *args, **kwargs):
+            return f"args: {args}, kwargs: {kwargs}"
+
+    client = ArgClient()
+    assert isinstance(client, RelayClientLike)
+
+
+# ============================================================================
+# Protocol Validation Tests
+# ============================================================================
+
+
+def test_relayclientlike_missing_url():
+    """Test class missing url attribute is not RelayClientLike"""
+
+    class MissingUrl:
+        secret = "secret"
+        timeout = 180
+
+        def call(self):
+            return None
+
+    client = MissingUrl()
+    assert not isinstance(client, RelayClientLike)
+
+
+def test_relayclientlike_missing_secret():
+    """Test class missing secret attribute is not RelayClientLike"""
+
+    class MissingSecret:
+        url = "https://example.com"
+        timeout = 180
+
+        def call(self):
+            return None
+
+    client = MissingSecret()
+    assert not isinstance(client, RelayClientLike)
+
+
+def test_relayclientlike_missing_timeout():
+    """Test class missing timeout attribute is not RelayClientLike"""
+
+    class MissingTimeout:
+        url = "https://example.com"
+        secret = "secret"
+
+        def call(self):
+            return None
+
+    client = MissingTimeout()
+    assert not isinstance(client, RelayClientLike)
+
+
+def test_relayclientlike_missing_call():
+    """Test class missing call method is not RelayClientLike"""
+
+    class MissingCall:
+        url = "https://example.com"
+        secret = "secret"
+        timeout = 180
+
+    client = MissingCall()
+    assert not isinstance(client, RelayClientLike)
+
+
+def test_relayclientlike_wrong_attribute_types():
+    """Test class with wrong attribute types can still be RelayClientLike"""
+
+    class WrongTypes:
+        url = 123  # Should be string but protocol doesn't enforce at runtime
+        secret = ["list"]  # Should be string
+        timeout = "180"  # Should be int
+
+        def call(self):
+            return None
+
+    client = WrongTypes()
+    # runtime_checkable only checks existence, not types
+    assert isinstance(client, RelayClientLike)
+
+
+# ============================================================================
+# Integration Tests
+# ============================================================================
+
+
+def test_relayclientlike_function_parameter():
+    """Test RelayClientLike as function parameter type"""
+
+    def execute_and_get_url(client: RelayClientLike) -> tuple:
+        result = client.call()
+        url = client.url
+        return result, url
+
+    client = CompliantClient()
+    result, url = execute_and_get_url(client)
+
+    assert result == {"status": "success"}
+    assert url == "https://relay.example.com"
+
+
+def test_relayclientlike_list_of_clients():
+    """Test list of RelayClientLike objects"""
+    clients = [CompliantClient(), MinimalClient()]
+
+    for client in clients:
+        assert isinstance(client, RelayClientLike)
+        assert callable(client.call)
+
+
+def test_relayclientlike_runtime_check():
+    """Test runtime_checkable allows isinstance checks"""
+
+    class RuntimeClient:
+        url = "https://runtime.example.com"
+        secret = "runtime-secret"
+        timeout = 180
+
+        def call(self):
+            return "runtime"
+
+    client = RuntimeClient()
+
+    # This works because Protocol is runtime_checkable
+    assert isinstance(client, RelayClientLike)
+
+
+def test_relayclientlike_protocol_not_instantiable():
+    """Test RelayClientLike protocol itself cannot be instantiated"""
+    # Protocols are not meant to be instantiated directly
+    # This test verifies the protocol behavior
+
+    # We can create instances that conform to the protocol
+    client = CompliantClient()
+    assert isinstance(client, RelayClientLike)
+
+    # But we can't instantiate the protocol itself
+    # (This would fail: client = RelayClientLike())
+
+
+def test_relayclientlike_with_dataclass():
+    """Test RelayClientLike with dataclass"""
+    from dataclasses import dataclass
+
+    @dataclass
+    class DataClient:
+        url: str = "https://data.example.com"
+        secret: str = "data-secret"
+        timeout: int = 180
+
+        def call(self):
+            return "data result"
+
+    client = DataClient()
+    assert isinstance(client, RelayClientLike)
+
+
+def test_relayclientlike_attribute_access():
+    """Test accessing protocol attributes"""
+    client = CompliantClient()
+
+    # Can access all protocol-defined attributes
+    assert client.url == "https://relay.example.com"
+    assert client.secret == "test-secret"
+    assert client.timeout == 180
+
+    # Can call protocol-defined method
+    result = client.call()
+    assert result is not None
+
+
+def test_relayclientlike_polymorphism():
+    """Test polymorphic behavior with RelayClientLike"""
+    clients = [
+        CompliantClient(),
+        MinimalClient(),
+    ]
+
+    results = [client.call() for client in clients]
+
+    assert len(results) == 2
+    assert results[0] == {"status": "success"}
+    assert results[1] is None
+
+
+# ============================================================================
+# Protocol Documentation Tests
+# ============================================================================
+
+
+def test_relayclientlike_is_protocol():
+    """Test RelayClientLike is a Protocol"""
+    from typing import Protocol
+
+    assert issubclass(RelayClientLike, Protocol)
+
+
+def test_relayclientlike_is_runtime_checkable():
+    """Test RelayClientLike is runtime_checkable"""
+
+    # Check if the decorator was applied
+    assert hasattr(RelayClientLike, "__protocol_attrs__") or hasattr(RelayClientLike, "_is_runtime_protocol")

xfintech/data/source/__init__.py

@@ -0,0 +1,7 @@
+"""
+Data source module for xfintech.
+
+This module provides access to various financial data sources:
+- tushare: Tushare Pro API data source with comprehensive A-share market data
+- baostock: Baostock data source for alternative market data
+"""

xfintech/data/source/baostock/__init__.py

@@ -0,0 +1,21 @@
+from .job.job import BaostockJob
+from .session.session import Session
+from .stock.hs300stock.hs300stock import HS300Stock
+from .stock.minuteline.minuteline import Minuteline
+from .stock.stock.stock import Stock
+from .stock.stockinfo.stockinfo import StockInfo
+from .stock.sz50stock.sz50stock import SZ50Stock
+from .stock.tradedate.tradedate import TradeDate
+from .stock.zz500stock.zz500stock import ZZ500Stock
+
+__all__ = [
+    "Session",
+    "HS300Stock",
+    "StockInfo",
+    "SZ50Stock",
+    "Stock",
+    "ZZ500Stock",
+    "TradeDate",
+    "Minuteline",
+    "BaostockJob",
+]

xfintech/data/source/baostock/job/__init__.py

@@ -0,0 +1,5 @@
+from .job import BaostockJob
+
+__all__ = [
+    "BaostockJob",
+]