test: retarget v2 coverage monkeypatch targets to definition modules after handler split (assertions unchanged)
This commit is contained in:
1 parent
6e0660e97b
commit
741eb387ed
7 files changed
+45
-48
No files matched your search
@@ -147,7 +147,7 @@ class CostRelayTestCase(_Base):
|
||||
def search(self, project_id, q, top_k=5):
|
||||
return []
|
||||
h = self._h()
|
||||
with mock.patch("cloud.server.KBStore", _Store):
|
||||
with mock.patch("cloud.handler_identity.KBStore", _Store):
|
||||
h._kb_search("p1", {"q": ["x"]})
|
||||
self.assertEqual(self._last(h)[0], 200)
|
||||
|
||||
@@ -199,7 +199,7 @@ class CostRelayTestCase(_Base):
|
||||
with mock.patch.dict(os.environ, {"CLOUD_GATEWAY": "sub2api"}), \
|
||||
mock.patch("cloud.server._validate_sub2api_key",
|
||||
return_value=("ok", "key", "sub", "kid")), \
|
||||
mock.patch("cloud.server._list_sub2api_models",
|
||||
mock.patch("cloud.handler_identity._list_sub2api_models",
|
||||
side_effect=RuntimeError("gw down")):
|
||||
h._relay_models("u1")
|
||||
self.assertEqual(self._last(h)[0], 502)
|
||||
|
||||
Reference in new issue
Block a user