fix: enforce VIP content boundary and PATCH updates
This commit is contained in:
@@ -42,10 +42,12 @@ class FakeAgentApi(AgentApi):
|
||||
def __init__(self):
|
||||
super().__init__("test-key")
|
||||
self.responses = []
|
||||
self.requests = []
|
||||
self.public = (200, b"page")
|
||||
self.rsc = (200, b"rsc")
|
||||
|
||||
def _request(self, method, path, body=None, *, anonymous=False):
|
||||
self.requests.append((method, path, body, anonymous))
|
||||
if self.responses:
|
||||
return self.responses.pop(0)
|
||||
return 200, b"{}", "application/json"
|
||||
@@ -80,6 +82,12 @@ class RuntimeTests(unittest.TestCase):
|
||||
self.assertEqual(api.create(body)["slug"], "guide")
|
||||
self.assertEqual(api.body_probes["guide"], "distinctive protected body sentence")
|
||||
|
||||
def test_agent_api_updates_with_patch(self):
|
||||
api = FakeAgentApi()
|
||||
api.responses = [(200, json.dumps({"slug": "guide"}).encode(), "application/json")]
|
||||
self.assertEqual(api.update("guide", {"publish": True})["slug"], "guide")
|
||||
self.assertEqual(api.requests[0][0:2], ("PATCH", "/api/agent/posts/guide"))
|
||||
|
||||
@patch("runtime.time.sleep", return_value=None)
|
||||
def test_vip_verification_rejects_body_leak(self, _sleep):
|
||||
api = FakeAgentApi()
|
||||
|
||||
Reference in New Issue
Block a user