def test_inject_parent_settings(self, _pulp_conf, _read_config, _open):
host = 'test-host'
cert_path = '/path/cert'
_pulp_conf.get.return_value = host
_read_config.return_value.main = Mock(node_certificate=cert_path)
fp = Mock()
fp.__enter__ = Mock(return_value=fp)
fp.__exit__ = Mock()
_open.return_value = fp
# test
options = {}
NodeProfiler._inject_parent_settings(options)
# validation
_pulp_conf.get.assert_called_once_with('server', 'server_name')
_read_config.assert_called_once_with()
_open.assert_called_once_with(cert_path)
settings = options[constants.PARENT_SETTINGS]
self.assertEqual(settings[constants.HOST], host)
self.assertEqual(settings[constants.PORT], 443)
self.assertEqual(
settings[constants.NODE_CERTIFICATE],
_open.return_value.read.return_value)
def test_inject_parent_settings(self, _pulp_conf, _read_config, _open):
host = 'test-host'
cert_path = '/path/cert'
_pulp_conf.get.return_value = host
_read_config.return_value.main = Mock(node_certificate=cert_path)
fp = Mock()
fp.__enter__ = Mock(return_value=fp)
fp.__exit__ = Mock()
_open.return_value = fp
# test
options = {}
NodeProfiler._inject_parent_settings(options)
# validation
_pulp_conf.get.assert_called_once_with('server', 'server_name')
_read_config.assert_called_once_with()
_open.assert_called_once_with(cert_path)
settings = options[constants.PARENT_SETTINGS]
self.assertEqual(settings[constants.HOST], host)
self.assertEqual(settings[constants.PORT], 443)
self.assertEqual(settings[constants.NODE_CERTIFICATE],
_open.return_value.read.return_value)
def test_inject_strategy(self, managers):
consumer_id = 'test-consumer'
strategy = 'test-strategy'
notes = {constants.STRATEGY_NOTE_KEY: strategy}
manager = Mock()
manager.get_consumer.return_value = {'notes': notes}
managers.consumer_manager.return_value = manager
# test
options = {}
NodeProfiler._inject_strategy(consumer_id, options)
# validation
manager.get_consumer.assert_called_once_with(consumer_id)
self.assertEqual(options[constants.STRATEGY_KEYWORD], strategy)
def test_inject_strategy(self, managers):
consumer_id = 'test-consumer'
strategy = 'test-strategy'
notes = {constants.STRATEGY_NOTE_KEY: strategy}
manager = Mock()
manager.get_consumer.return_value = {'notes': notes}
managers.consumer_manager.return_value = manager
# test
options = {}
NodeProfiler._inject_strategy(consumer_id, options)
# validation
manager.get_consumer.assert_called_once_with(consumer_id)
self.assertEqual(options[constants.STRATEGY_KEYWORD], strategy)
def test_metadata(self, gettext):
self.assertEqual(
NodeProfiler.metadata(), {
'id': constants.PROFILER_ID,
'display_name': gettext.return_value,
'types': [constants.NODE_SCOPE, constants.REPOSITORY_SCOPE]
})
def test_update_units(self):
consumer = Mock(id='123')
units = [1, 2, 3]
options = {}
config = Mock()
conduit = Mock()
# test
profiler = NodeProfiler()
profiler._inject_parent_settings = Mock()
profiler._inject_strategy = Mock()
# validation
_units = profiler.update_units(consumer, units, options, config, conduit)
profiler._inject_parent_settings.assert_called_once_with(options)
profiler._inject_strategy.assert_called_once_with(consumer.id, options)
self.assertEqual(_units, units)
def test_metadata(self, gettext):
self.assertEqual(
NodeProfiler.metadata(),
{
'id': constants.PROFILER_ID,
'display_name': gettext.return_value,
'types': [constants.NODE_SCOPE, constants.REPOSITORY_SCOPE]
})
def test_update_units(self):
consumer = Mock(id='123')
units = [1, 2, 3]
options = {}
config = Mock()
conduit = Mock()
# test
profiler = NodeProfiler()
profiler._inject_parent_settings = Mock()
profiler._inject_strategy = Mock()
# validation
_units = profiler.update_units(consumer, units, options, config, conduit)
profiler._inject_parent_settings.assert_called_once_with(options)
profiler._inject_strategy.assert_called_once_with(consumer.id, options)
self.assertEqual(_units, units)
def test_update_units(self, mock_read_config):
# Setup
mock_read_config.return_value = self.node_configuration()
# Test
host = 'abc'
port = 443
units = [1, 2, 3]
options = {}
p = NodeProfiler()
pulp_conf.set('server', 'server_name', host)
_units = p.update_units(None, units, options, None, None)
# Verify
self.assertTrue(constants.PARENT_SETTINGS in options)
settings = options[constants.PARENT_SETTINGS]
self.assertEqual(settings[constants.HOST], host)
self.assertEqual(settings[constants.PORT], port)
self.assertEqual(settings[constants.NODE_CERTIFICATE], NODE_CERTIFICATE)
self.assertEqual(units, _units)
def test_update_units(self, mock_get_node_conf):
# Setup
mock_get_node_conf.return_value = self.node_configuration()
# Test
host = 'abc'
port = 443
units = [1, 2, 3]
options = {}
p = NodeProfiler()
pulp_conf.set('server', 'server_name', host)
_units = p.update_units(None, units, options, None, None)
# Verify
self.assertTrue(constants.PARENT_SETTINGS in options)
settings = options[constants.PARENT_SETTINGS]
self.assertEqual(settings[constants.HOST], host)
self.assertEqual(settings[constants.PORT], port)
self.assertEqual(settings[constants.NODE_CERTIFICATE],
NODE_CERTIFICATE)
self.assertEqual(units, _units)