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())