def test_add_module_not_copied(self, mock_shutil):
module_path = '/tmp/mod.tar.gz'
feed_url = 'http://host/root/PULP_MANAFEST'
unit_key = {'name': 'puppet-module'}
unit_metadata = {'A': 1, 'B': 2}
unit = Mock()
unit.storage_path = os.path.join(os.getcwd(), __file__)
mock_conduit = Mock()
mock_conduit.init_unit = Mock(return_value=unit)
config = {constants.CONFIG_FEED: feed_url}
mock_module = Mock()
mock_module.unit_key = Mock(return_value=unit_key)
mock_module.unit_metadata = Mock(return_value=unit_metadata)
mock_module.filename = Mock(return_value='puppet-module')
# test
method = SynchronizeWithDirectory(mock_conduit, config)
method._add_module(module_path, mock_module)
# validation
self.assertFalse(mock_shutil.copy.called)
def test_add_module_not_copied(self, mock_shutil):
module_path = '/tmp/mod.tar.gz'
feed_url = 'http://host/root/PULP_MANAFEST'
unit_key = {'name': 'puppet-module'}
unit_metadata = {'A': 1, 'B': 2}
unit = Mock()
unit.storage_path = os.path.join(os.getcwd(), __file__)
mock_conduit = Mock()
mock_conduit.init_unit = Mock(return_value=unit)
config = {constants.CONFIG_FEED: feed_url}
mock_module = Mock()
mock_module.unit_key = Mock(return_value=unit_key)
mock_module.unit_metadata = Mock(return_value=unit_metadata)
mock_module.filename = Mock(return_value='puppet-module')
# test
method = SynchronizeWithDirectory(mock_conduit, config)
method._add_module(module_path, mock_module)
# validation
self.assertFalse(mock_shutil.copy.called)
def test_add_module(self, mock_shutil):
module_path = '/tmp/mod.tar.gz'
feed_url = 'http://host/root/PULP_MANAFEST'
unit_key = {'name': 'puppet-module'}
unit_metadata = {'A': 1, 'B': 2}
unit = Mock()
unit.storage_path = '/tmp/%s' % uuid4()
mock_conduit = Mock()
mock_conduit.init_unit = Mock(return_value=unit)
config = {constants.CONFIG_FEED: feed_url}
mock_module = Mock()
mock_module.unit_key = Mock(return_value=unit_key)
mock_module.unit_metadata = Mock(return_value=unit_metadata)
mock_module.filename = Mock(return_value='puppet-module')
# test
method = SynchronizeWithDirectory(mock_conduit, config)
method._add_module(module_path, mock_module)
# validation
mock_conduit.init_unit.assert_called_with(constants.TYPE_PUPPET_MODULE,
unit_key, unit_metadata,
mock_module.filename())
mock_shutil.copy.assert_called_with(module_path, unit.storage_path)
def test_add_module(self, mock_shutil):
module_path = '/tmp/mod.tar.gz'
feed_url = 'http://host/root/PULP_MANAFEST'
unit_key = {'name': 'puppet-module'}
unit_metadata = {'A': 1, 'B': 2}
unit = Mock()
unit.storage_path = '/tmp/%s' % uuid4()
mock_conduit = Mock()
mock_conduit.init_unit = Mock(return_value=unit)
config = {constants.CONFIG_FEED: feed_url}
mock_module = Mock()
mock_module.unit_key = Mock(return_value=unit_key)
mock_module.unit_metadata = Mock(return_value=unit_metadata)
mock_module.filename = Mock(return_value='puppet-module')
# test
method = SynchronizeWithDirectory(mock_conduit, config)
method._add_module(module_path, mock_module)
# validation
mock_conduit.init_unit.assert_called_with(
constants.TYPE_PUPPET_MODULE, unit_key, unit_metadata, mock_module.filename())
mock_shutil.copy.assert_called_with(module_path, unit.storage_path)
