def test_rpm(self, mock_open, mock_package_list_generator):
rpms = model_factory.rpm_models(2)
mock_package_list_generator.return_value = rpms
fake_file = StringIO()
mock_open.return_value = fake_file
process_func = lambda x: x
ret = purge.get_remote_units(self.metadata_files, self.FAKE_TYPE, 'bar', process_func)
mock_open.assert_called_once_with('/a/b/c', 'r')
self.assertTrue(fake_file.closed)
mock_package_list_generator.assert_called_once_with(fake_file, 'bar', process_func)
self.assertEqual(len(rpms), len(ret))
for model in rpms:
self.assertTrue(model.as_named_tuple in ret)
def test_rpm(self, mock_open, mock_package_list_generator):
rpms = model_factory.rpm_models(2)
mock_package_list_generator.return_value = rpms
fake_file = StringIO()
mock_open.return_value = fake_file
def process_func(x):
return x
file_function = functools.partial(self.metadata_files.get_metadata_file_handle,
self.FAKE_TYPE)
ret = purge.get_remote_units(file_function, 'bar', process_func)
mock_open.assert_called_once_with('/a/b/c', 'r')
self.assertTrue(fake_file.closed)
mock_package_list_generator.assert_called_once_with(fake_file, 'bar', process_func)
self.assertEqual(len(rpms), len(ret))
for model in rpms:
self.assertTrue(model.as_named_tuple in ret)
def test_rpm(self, mock_open, mock_package_list_generator):
rpms = model_factory.rpm_models(2)
mock_package_list_generator.return_value = rpms
fake_file = StringIO()
mock_open.return_value = fake_file
def process_func(x):
return x
file_function = functools.partial(self.metadata_files.get_metadata_file_handle, self.FAKE_TYPE)
ret = purge.get_remote_units(file_function, "bar", process_func)
mock_open.assert_called_once_with("/a/b/c", "r")
self.assertTrue(fake_file.closed)
mock_package_list_generator.assert_called_once_with(fake_file, "bar", process_func)
self.assertEqual(len(rpms), len(ret))
for model in rpms:
self.assertTrue(model.as_named_tuple in ret)
def test_no_units(self):
file_function = mock.Mock(return_value=None)
ret = purge.get_remote_units(file_function, 'bar', lambda x: x)
self.assertEqual(ret, set())
def test_no_units(self):
file_function = mock.Mock(return_value=None)
ret = purge.get_remote_units(file_function, 'bar', lambda x: x)
self.assertEqual(ret, set())
def test_no_units(self):
ret = purge.get_remote_units(self.metadata_files, 'foo', 'bar', lambda x: x)
self.assertEqual(ret, set())
