def test_list_statements(mocker, server):
mocker.patch.object(Statement, 'from_json')
client = LivyClient(server)
statements = client.list_statements(MOCK_SESSION_ID)
assert statements == [Statement.from_json.return_value]
Statement.from_json.assert_called_once_with(MOCK_SESSION_ID,
MOCK_STATEMENT_JSON)
def test_list_statements(requests_mock, mocker):
requests_mock.get(
f"http://example.com/sessions/{MOCK_SESSION_ID}/statements",
json={"statements": [MOCK_STATEMENT_JSON]},
)
mocker.patch.object(Statement, "from_json")
client = LivyClient("http://example.com")
statements = client.list_statements(MOCK_SESSION_ID)
assert statements == [Statement.from_json.return_value]
Statement.from_json.assert_called_once_with(MOCK_SESSION_ID,
MOCK_STATEMENT_JSON)
