def test_post_statement():
http_client = MagicMock()
livy_client = LivyReliableHttpClient(http_client)
data = {"adlfj":"sadflkjsdf"}
out = livy_client.post_statement(100, data)
assert_equals(out, http_client.post.return_value.json.return_value)
http_client.post.assert_called_once_with("/sessions/100/statements", [201], data)
def test_post_statement():
http_client = MagicMock()
livy_client = LivyReliableHttpClient(http_client)
data = {"adlfj": "sadflkjsdf"}
out = livy_client.post_statement(100, data)
assert_equals(out, http_client.post.return_value.json.return_value)
http_client.post.assert_called_once_with("/sessions/100/statements", [201],
data)
