def test_close(self, mock_session):
mock_session.return_value = None
stream_cursor = StreamCursor(MOCK_STATEMENT_RESULT, mock_session,
MOCK_TRANSACTION_ID)
stream_cursor.close()
self.assertFalse(stream_cursor._is_open)
def test_next_when_closed(self, mock_session):
stream_cursor = StreamCursor(MOCK_STATEMENT_RESULT, mock_session,
MOCK_TRANSACTION_ID)
stream_cursor.close()
self.assertRaises(ResultClosedError, next, stream_cursor)