def test_make_gen3_butler(self, *mocks):
converter = DC2Converter.load(TEST_CONFIG_FILE)
dc2gen3.Butler()
butler = converter.gen3_butler
self.assertIsInstance(butler, Butler)
second_butler = converter.gen3_butler
self.assertIs(second_butler, butler)
def test_delete_gen3_repo(self, *mocks):
converter = DC2Converter.load(TEST_CONFIG_FILE)
connection_uri = 'postgresql:///bogus?host=example.org&port=0&user=nobody'
registry_namespace = 'bogusnamespace'
converter.gen3_root, parent_dir = make_name_of_nonexistant_dir()
converter._delete_gen3_repo(connection_uri, registry_namespace)
dc2gen3._empty_registry.assert_called_once()
dc2gen3._delete_data.assert_called_once()
def test_convert_with_delete(self, *mocks):
converter = DC2Converter.load(TEST_CONFIG_FILE)
converter.gen3_root, parent_dir = make_name_of_nonexistant_dir()
task = converter.convert_new(delete=True)
self.assertIsInstance(converter.gen3_butler, Butler)
task.run.assert_called_once()
dc2gen3.Butler.makeRepo.assert_called_once()
dc2gen3._empty_registry.assert_called_once()
dc2gen3._delete_data.assert_called_once()
def test_load(self, *mocks):
converter = DC2Converter.load(TEST_CONFIG_FILE)
self.assertIsInstance(converter.gen2_root, str)
self.assertIsInstance(converter.gen3_root, str)
self.assertIsInstance(converter.instrument, lsst.obs.base.Instrument)
self.assertIsInstance(
converter.convert_repo_config,
lsst.obs.base.gen2to3.ConvertRepoTask.ConfigClass)
self.assertIsInstance(converter.butler_seed, str)
self.assertIsInstance(converter.reruns, list)
self.assertIsInstance(converter.reruns[0], lsst.obs.base.gen2to3.Rerun)
self.assertIsInstance(converter.calibs, dict)
self.assertIsInstance(converter.delete_old_gen3, bool)
self.assertIsInstance(converter.create_new_gen3, bool)
self.assertIsInstance(converter.convert_skymap, bool)
def test_converter_call_create(self, *mocks):
converter = DC2Converter.load(TEST_CONFIG_FILE)
converter.create_new_gen3 = False
converter()
converter.run_conversion_task.assert_called_once()
converter.convert_new.assert_not_called()
def test_run_conversion_task(self, *mocks):
converter = DC2Converter.load(TEST_CONFIG_FILE)
task = converter.run_conversion_task()
self.assertIsInstance(converter.gen3_butler, Butler)
task.run.assert_called_once()
def test_registry_namespace(self):
converter = DC2Converter.load(TEST_CONFIG_FILE)
registry_namespace = converter.registry_namespace
self.assertIsInstance(registry_namespace, str)
def test_registry_connection_uri(self):
converter = DC2Converter.load(TEST_CONFIG_FILE)
uri = converter.registry_connection_uri
self.assertIsInstance(uri, str)
def test_put_skymap(self, *macks):
converter = DC2Converter.load(TEST_CONFIG_FILE)
converter.instrument = MagicMock(spec=converter.instrument)
converter._put_skymap()
converter.gen3_butler.put.assert_called_once()