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)