update test

Author: balogh.adam@icloud.com

tests/tee_registry_test.py

@@ -144,7 +144,7 @@ def test_validator_type_label(self, mock_contract):
 
 
 class TestGetLlmTee:
-    def test_returns_first_active_tee(self, mock_contract):
+    def test_returns_random_active_tee(self, mock_contract):
         registry, contract = mock_contract
 
         contract.functions.getActiveTEEs.return_value.call.return_value = [
@@ -155,7 +155,7 @@ def test_returns_first_active_tee(self, mock_contract):
         result = registry.get_llm_tee()
 
         assert result is not None
-        assert result.endpoint == "https://tee-1.example.com"
+        assert result.endpoint in ("https://tee-1.example.com", "https://tee-2.example.com")
 
     def test_returns_none_when_no_tees(self, mock_contract):
         registry, contract = mock_contract