def test_get_statement(): http_client = MagicMock() livy_client = LivyReliableHttpClient(http_client) out = livy_client.get_statement(100, 4) assert_equals(out, http_client.get.return_value.json.return_value) http_client.get.assert_called_once_with("/sessions/100/statements/4", [200])