feat: update amazon driver

app/tests/test_amazon_base.py

@@ -154,6 +154,33 @@ def test_get_browser_list_status_handling(monkeypatch, payload, expected):
     assert base.ZiniaoDriver({}).get_browser_list() == expected
 
 
+def test_update_core_retries_none_then_returns_on_success(monkeypatch):
+    calls = []
+    responses = [FakeResponse(None), FakeResponse({"statusCode": 0})]
+
+    def fake_post(url, data, timeout):
+        calls.append(json.loads(data.decode("utf-8")))
+        return responses.pop(0)
+
+    monkeypatch.setattr(base.requests, "post", fake_post)
+    monkeypatch.setattr(base.time, "sleep", lambda seconds: None)
+
+    base.ZiniaoDriver({}).update_core()
+
+    assert [payload["action"] for payload in calls] == ["updateCore", "updateCore"]
+
+
+def test_wait_until_client_ready_polls_until_available(monkeypatch):
+    readiness = [False, False, True]
+    sleeps = []
+    driver = base.ZiniaoDriver({})
+    monkeypatch.setattr(driver, "_is_client_ready", lambda: readiness.pop(0))
+    monkeypatch.setattr(base.time, "sleep", lambda seconds: sleeps.append(seconds))
+
+    assert driver._wait_until_client_ready(timeout=10) is True
+    assert sleeps == [base.CLIENT_READY_INTERVAL, base.CLIENT_READY_INTERVAL]
+
+
 def test_close_store_success_uses_current_store(monkeypatch):
     payloads = []