def test_say_hello_ok(tmpdir, cert, key):
crt_path = tmpdir / 'client.crt'
key_path = tmpdir / 'client.key'
Path(crt_path).write_text(cert)
Path(key_path).write_text(key)
agent.CERT_PATH = str(tmpdir)
agent.CLIENT_CERT_PATH = str(crt_path)
agent.CLIENT_KEY_PATH = str(key_path)
hello = agent.say_hello()
assert hello['message']
def test_say_hello_failed(tmpdir, invalid_cert, invalid_key):
crt_path = tmpdir / 'client.crt'
key_path = tmpdir / 'client.key'
Path(crt_path).write_text(invalid_cert)
Path(key_path).write_text(invalid_key)
agent.CERT_PATH = str(tmpdir)
agent.CLIENT_CERT_PATH = str(crt_path)
agent.CLIENT_KEY_PATH = str(key_path)
with mock.patch('builtins.print') as prn:
with pytest.raises(json.decoder.JSONDecodeError):
_ = agent.say_hello()
assert mock.call('Hello failed.') in prn.mock_calls