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