Spaces:
Sleeping
Sleeping
Expose read-only attestation authority header
#3
by betterwithage - opened
- server.py +1 -0
- tests/test_source_attestation.py +1 -0
server.py
CHANGED
|
@@ -59,6 +59,7 @@ class HardenedHandler(SimpleHTTPRequestHandler):
|
|
| 59 |
self.send_header("X-SZL-Transport-State", str(payload["transport_state"]))
|
| 60 |
self.send_header("X-SZL-Evidence-State", str(payload["evidence_state"]))
|
| 61 |
self.send_header("X-SZL-Verification-State", str(payload["verification_state"]))
|
|
|
|
| 62 |
self.end_headers()
|
| 63 |
self.wfile.write(body)
|
| 64 |
|
|
|
|
| 59 |
self.send_header("X-SZL-Transport-State", str(payload["transport_state"]))
|
| 60 |
self.send_header("X-SZL-Evidence-State", str(payload["evidence_state"]))
|
| 61 |
self.send_header("X-SZL-Verification-State", str(payload["verification_state"]))
|
| 62 |
+
self.send_header("X-SZL-Authority-State", str(payload["authority_state"]))
|
| 63 |
self.end_headers()
|
| 64 |
self.wfile.write(body)
|
| 65 |
|
tests/test_source_attestation.py
CHANGED
|
@@ -46,6 +46,7 @@ class SourceAttestationTests(unittest.TestCase):
|
|
| 46 |
self.assertEqual(200, response.status)
|
| 47 |
self.assertEqual("no-store", response.headers["Cache-Control"])
|
| 48 |
self.assertEqual("COMPUTED", response.headers["X-SZL-Evidence-State"])
|
|
|
|
| 49 |
self.assertEqual("b" * 40, payload["deployment"]["hf_revision"])
|
| 50 |
finally:
|
| 51 |
httpd.shutdown()
|
|
|
|
| 46 |
self.assertEqual(200, response.status)
|
| 47 |
self.assertEqual("no-store", response.headers["Cache-Control"])
|
| 48 |
self.assertEqual("COMPUTED", response.headers["X-SZL-Evidence-State"])
|
| 49 |
+
self.assertEqual("READ_ONLY", response.headers["X-SZL-Authority-State"])
|
| 50 |
self.assertEqual("b" * 40, payload["deployment"]["hf_revision"])
|
| 51 |
finally:
|
| 52 |
httpd.shutdown()
|