def test_reads_config_on_first_use(self, yaml_load):
fy_stream = mock.Mock()
fy = filters.FilterYaml(fy_stream)
self.assertFalse(yaml_load.called)
fy.get_filter_yaml()
self.assertTrue(yaml_load.called)
def test_returns_tenant_id_if_provided(self):
tenant_id = 'some-tenant'
filters_file = """
tenants:
tenant_id:
- {tenant_id}
""".format(tenant_id=tenant_id)
fy = filters.FilterYaml(filters_file)
self.assertEqual(tenant_id, fy.get_tenant())
def get_filter(self):
if self.volume_filter is None:
with open(self.config.migrate.filter_path, 'r') as f:
filter_yaml = filters.FilterYaml(f)
filter_yaml.read()
self.volume_filter = cinder_filters.CinderFilters(
self.cinder_client, filter_yaml)
return self.volume_filter
def get_image_filter(self):
if self._image_filter is None:
with open(self.config.migrate.filter_path, 'r') as f:
filter_yaml = filters.FilterYaml(f)
filter_yaml.read()
self._image_filter = glance_filters.GlanceFilters(
self.glance_client, filter_yaml)
return self._image_filter
def test_date_filter_returns_datetime_object(self):
filters_file = """
images:
date: 2000-01-01
"""
fy = filters.FilterYaml(filters_file)
filtered_date = fy.get_image_date()
self.assertTrue(isinstance(filtered_date, datetime.date))
def test_returns_images_from_excluded_image_ids(self):
image1 = 'image1'
image2 = 'image2'
filters_file = """
images:
exclude_images_list:
- {image1}
- {image2}
""".format(image1=image1, image2=image2)
fy = filters.FilterYaml(filters_file)
filtered_images = fy.get_excluded_image_ids()
self.assertTrue(isinstance(filtered_images, list))
self.assertIn(image1, filtered_images)
self.assertIn(image2, filtered_images)
def test_returns_instances_from_instance_ids(self):
instance1 = 'inst1'
instance2 = 'inst2'
filters_file = """
instances:
id:
- {instance1}
- {instance2}
""".format(instance1=instance1, instance2=instance2)
fy = filters.FilterYaml(filters_file)
filtered_instances = fy.get_instance_ids()
self.assertTrue(isinstance(filtered_instances, list))
self.assertIn(instance1, filtered_instances)
self.assertIn(instance2, filtered_instances)
def test_returns_empty_list_if_nothing_in_instance_ids(self):
filters_file = ""
fy = filters.FilterYaml(filters_file)
self.assertEqual(list(), fy.get_instance_ids())
def test_returns_empty_list_for_get_excluded_image_ids(self):
filters_file = ""
fy = filters.FilterYaml(filters_file)
self.assertEqual(list(), fy.get_excluded_image_ids())
def test_returns_none_if_no_tenant_provided(self):
filters_file = ""
fy = filters.FilterYaml(filters_file)
self.assertIsNone(fy.get_tenant())
def test_returns_empty_dict_if_filter_conf_is_empty(self):
filters_file = ""
fy = filters.FilterYaml(filters_file)
self.assertEqual(dict(), fy.get_filter_yaml())
def test_date_returns_none_if_not_specified(self):
filters_file = ""
fy = filters.FilterYaml(filters_file)
self.assertIsNone(fy.get_image_date())
