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)