class FileMockTests(BaseTestCase):
def before(self):
self.mock = FileMock(DEFAULT_CONTENT)
def test_file_mock_is_iterable(self):
counter = 0
for line in self.mock:
assert_that(line, equal_to(DEFAULT_CONTENT[counter]))
counter += 1
def test_file_mock_is_readable_as_a_file(self):
expected_output = "".join(DEFAULT_CONTENT)
assert_that(self.mock.read(), equal_to(expected_output))
self.mock.close()
