Class MockIterator

tests/test_lint.py:245–259

class TestIteratorWrapper(unittest.TestCase):
    def test_close(self):
        class MockIterator:

            def __init__(self):
                self.closed = False

            def __iter__(self):
                return self

            def __next__(self):
                return None

            next = __next__

            def close(self):
                self.closed = True

        mock = MockIterator()
        wrapper = IteratorWrapper(mock, None)