Python LivyReliableHttpClient Examples

Example #1

File: test_livyreliablehttpclient.py Project: cfregly/sparkmagic

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)

Example #2

File: test_livyreliablehttpclient.py Project: mindis/sparkmagic

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)

Example #3

File: test_livyreliablehttpclient.py Project: cfregly/sparkmagic

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)

Example #4

File: test_livyreliablehttpclient.py Project: mindis/sparkmagic

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])

Example #5

File: test_livyreliablehttpclient.py Project: mindis/sparkmagic

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)

Example #6

File: test_livyreliablehttpclient.py Project: cfregly/sparkmagic

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])

Example #7

File: test_livyreliablehttpclient.py Project: cfregly/sparkmagic

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])

Example #8

File: test_livyreliablehttpclient.py Project: cfregly/sparkmagic

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])

Example #9

File: test_livyreliablehttpclient.py Project: mindis/sparkmagic

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])

Example #10

File: test_livyreliablehttpclient.py Project: mindis/sparkmagic

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])