def test_get_statement(): http_client = MagicMock() livy_client = LivyReliableHttpClient(http_client, None) 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])
def test_post_session(): http_client = MagicMock() livy_client = LivyReliableHttpClient(http_client, None) properties = {"adlfj": "sadflkjsdf", 1: [2, 3, 4, 5]} out = livy_client.post_session(properties) assert_equals(out, http_client.post.return_value.json.return_value) http_client.post.assert_called_once_with("/sessions", [201], properties)
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, None) 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_session(): http_client = MagicMock() livy_client = LivyReliableHttpClient(http_client) properties = {"adlfj":"sadflkjsdf", 1: [2,3,4,5]} out = livy_client.post_session(properties) assert_equals(out, http_client.post.return_value.json.return_value) http_client.post.assert_called_once_with("/sessions", [201], properties)
def test_custom_headers(): custom_headers = {"header1": "value1"} overrides = {conf.custom_headers.__name__: custom_headers} conf.override_all(overrides) endpoint = Endpoint("http://url.com", constants.NO_AUTH) client = LivyReliableHttpClient.from_endpoint(endpoint) headers = client.get_headers() assert_equals(len(headers), 2) assert_equals("Content-Type" in headers, True) assert_equals("header1" in headers, True)
def test_retry_policy(): # Default is configurable retry times = conf.retry_seconds_to_sleep_list() max_retries = conf.configurable_retry_policy_max_retries() policy = LivyReliableHttpClient._get_retry_policy() assert type(policy) is ConfigurableRetryPolicy assert_equals(times, policy.retry_seconds_to_sleep_list) assert_equals(max_retries, policy.max_retries) # Configure to linear retry _override_policy(constants.LINEAR_RETRY) policy = LivyReliableHttpClient._get_retry_policy() assert type(policy) is LinearRetryPolicy assert_equals(5, policy.seconds_to_sleep(1)) assert_equals(5, policy.max_retries) # Configure to something invalid _override_policy("garbage") try: policy = LivyReliableHttpClient._get_retry_policy() assert False except BadUserConfigurationException: assert True
def test_get_all_session_logs(): http_client = MagicMock() livy_client = LivyReliableHttpClient(http_client, None) out = livy_client.get_all_session_logs(42) assert_equals(out, http_client.get.return_value.json.return_value) http_client.get.assert_called_once_with("/sessions/42/log?from=0", [200])
def test_delete_session(): http_client = MagicMock() livy_client = LivyReliableHttpClient(http_client, None) livy_client.delete_session(99) http_client.delete.assert_called_once_with("/sessions/99", [200, 404])
def test_get_all_session_logs(): http_client = MagicMock() livy_client = LivyReliableHttpClient(http_client) out = livy_client.get_all_session_logs(42) assert_equals(out, http_client.get.return_value.json.return_value) http_client.get.assert_called_once_with("/sessions/42/log?from=0", [200])
def test_delete_session(): http_client = MagicMock() livy_client = LivyReliableHttpClient(http_client) livy_client.delete_session(99) http_client.delete.assert_called_once_with("/sessions/99", [200, 404])
def test_get_sessions(): http_client = MagicMock() livy_client = LivyReliableHttpClient(http_client) out = livy_client.get_sessions() assert_equals(out, http_client.get.return_value.json.return_value) http_client.get.assert_called_once_with("/sessions", [200])
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])