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