class TestContextContainerRegisteringDataSource(unittest.TestCase):
"""
Tests for `ContextContainerRegisteringDataSource`.
"""
def setUp(self):
self._temp_directory = tempfile.mkdtemp()
self._context = StubContext()
self._data_source = StubContextContainerRegisteringDataSource(
self._temp_directory, StubContextContainer, self._context)
def test_extract_data_from_file(self):
self._data_source.start()
# Don't have to wait to start
extracted = self._data_source.extract_data_from_file(EXAMPLE_CONTEXT_CONTAINER_FILE_LOCATION)
self.assertEqual(len(list(extracted)), 3)
for context_container in extracted:
self.assertEqual(context_container.context, self._context)
def tearDown(self):
os.rmdir(self._temp_directory)
def setUp(self):
self._temp_directory = tempfile.mkdtemp()
self._context = StubContext()
self._data_source = StubContextContainerRegisteringDataSource(
self._temp_directory, StubContextContainer, self._context)
