def test_process_answer_req_bad_format(req, alice):
for key, good_value in [
("organization_id", alice.organization_id),
("device_id", alice.device_id),
("rvk", alice.root_verify_key.encode()),
]:
if req.get(key) == "<good>":
req[key] = good_value
req["client_api_version"] = API_V1_VERSION
sh = ServerHandshake()
sh.build_challenge_req()
with pytest.raises(InvalidMessageError):
sh.process_answer_req(packb(req))
def test_build_result_req_bad_challenge(alice):
sh = ServerHandshake()
sh.build_challenge_req()
answer = {
"handshake": "answer",
"type": "authenticated",
"client_api_version": API_V1_VERSION,
"organization_id": alice.organization_id,
"device_id": alice.device_id,
"rvk": alice.root_verify_key.encode(),
"answer": alice.signing_key.sign(sh.challenge + b"-dummy"),
}
sh.process_answer_req(packb(answer))
with pytest.raises(HandshakeFailedChallenge):
sh.build_result_req(alice.verify_key)
def test_build_bad_outcomes(alice, method, expected_result):
sh = ServerHandshake()
sh.build_challenge_req()
answer = {
"handshake": "answer",
"type": "authenticated",
"client_api_version": API_V1_VERSION,
"organization_id": alice.organization_id,
"device_id": alice.device_id,
"rvk": alice.root_verify_key.encode(),
"answer": alice.signing_key.sign(sh.challenge),
}
sh.process_answer_req(packb(answer))
req = getattr(sh, method)()
assert unpackb(req) == {
"handshake": "result",
"result": expected_result,
"help": ANY
}
def test_process_result_req_bad_outcome(result, exc_cls):
ch = BaseClientHandshake()
with pytest.raises(exc_cls):
ch.process_result_req(packb({"handshake": "result", "result": result}))
def test_process_result_req_bad_format(req):
ch = BaseClientHandshake()
with pytest.raises(InvalidMessageError):
ch.process_result_req(packb(req))