fix tests

Author: balogh.adam@icloud.com

tests/llm_test.py

@@ -14,7 +14,6 @@
 import pytest
 
 from src.opengradient.client.llm import LLM
-from src.opengradient.client.tee_connection import TEEConnection
 from src.opengradient.types import TEE_LLM, x402SettlementMode
 
 # ── Fake HTTP transport ──────────────────────────────────────────────
@@ -511,50 +510,6 @@ async def test_close_delegates_to_http_client(self, fake_http):
         # FakeHTTPClient.aclose is a no-op; just verify it doesn't blow up.
 
 
-# ── TEE resolution tests ─────────────────────────────────────────────
-
-
-class TestResolveTeE:
-    def test_explicit_url_skips_registry(self):
-        endpoint, cert, tee_id, pay_addr = TEEConnection._resolve_tee("https://explicit.url", None)
-
-        assert endpoint == "https://explicit.url"
-        assert cert is None
-        assert tee_id is None
-        assert pay_addr is None
-
-    def test_missing_rpc_and_registry_raises(self):
-        with pytest.raises(ValueError):
-            TEEConnection._resolve_tee(None, None)
-
-    def test_missing_registry_address_raises(self):
-        with pytest.raises(ValueError):
-            TEEConnection._resolve_tee(None, None)
-
-    def test_registry_returns_none_raises(self):
-        mock_reg = MagicMock()
-        mock_reg.get_llm_tee.return_value = None
-
-        with pytest.raises(ValueError, match="No active LLM proxy TEE"):
-            TEEConnection._resolve_tee(None, mock_reg)
-
-    def test_registry_success(self):
-        mock_reg = MagicMock()
-        mock_tee = MagicMock()
-        mock_tee.endpoint = "https://registry.tee"
-        mock_tee.tls_cert_der = b"cert-bytes"
-        mock_tee.tee_id = "tee-42"
-        mock_tee.payment_address = "0xPay"
-        mock_reg.get_llm_tee.return_value = mock_tee
-
-        endpoint, cert, tee_id, pay_addr = TEEConnection._resolve_tee(None, mock_reg)
-
-        assert endpoint == "https://registry.tee"
-        assert cert == b"cert-bytes"
-        assert tee_id == "tee-42"
-        assert pay_addr == "0xPay"
-
-
 # ── TEE retry tests (non-streaming) ──────────────────────────────────
 
 
@@ -678,53 +633,6 @@ async def aread(self) -> bytes:
         assert len(fake_http.post_calls) == 1
 
 
-# ── TEE reconnect tests ─────────────────────────────────────────────
-
-
-@pytest.mark.asyncio
-class TestReconnect:
-    async def test_replaces_http_client(self):
-        """After reconnect, the http client should be a new instance."""
-        clients_created = []
-
-        def make_client(*args, **kwargs):
-            c = FakeHTTPClient()
-            clients_created.append(c)
-            return c
-
-        with (
-            patch(_PATCHES["x402_httpx"], side_effect=make_client),
-            patch(_PATCHES["x402_client"]),
-            patch(_PATCHES["signer"]),
-            patch(_PATCHES["register_exact"]),
-            patch(_PATCHES["register_upto"]),
-        ):
-            llm = _make_llm()
-            old_client = llm._tee.get().http_client
-
-            await llm._tee.reconnect()
-
-            assert llm._tee.get().http_client is not old_client
-            assert len(clients_created) == 2  # init + refresh
-
-    async def test_closes_old_client(self, fake_http):
-        llm = _make_llm()
-        old_client = llm._tee.get().http_client
-        old_client.aclose = AsyncMock()
-
-        await llm._tee.reconnect()
-
-        old_client.aclose.assert_awaited_once()
-
-    async def test_close_failure_is_swallowed(self, fake_http):
-        llm = _make_llm()
-        old_client = llm._tee.get().http_client
-        old_client.aclose = AsyncMock(side_effect=OSError("already closed"))
-
-        # Should not raise
-        await llm._tee.reconnect()
-
-
 # ── TEE cert rotation (crash + re-register) tests ────────────────────