def test_get_agent_bindings(self, mock_factory):
bind_payload = {'a': 1, 'b': 2}
distributor = {'distributor_type_id': '3838'}
mock_distributor_manager = Mock()
mock_distributor_manager.get_distributor = Mock(return_value=distributor)
mock_distributor_manager.create_bind_payload = Mock(return_value=bind_payload)
mock_factory.repo_distributor_manager = Mock(return_value=mock_distributor_manager)
# test manager
bindings = [
{'consumer_id': '10', 'repo_id': '20', 'distributor_id': '30', 'binding_config': {}},
{'consumer_id': '40', 'repo_id': '50', 'distributor_id': '60', 'binding_config': {}},
]
agent_bindings = AgentManager._bindings(bindings)
# validation
for binding in bindings:
mock_distributor_manager.get_distributor.assert_any_call(
binding['repo_id'], binding['distributor_id'])
mock_distributor_manager.create_bind_payload.assert_any_call(
binding['repo_id'], binding['distributor_id'], binding['binding_config'])
self.assertEqual(len(agent_bindings), 2)
for binding, agent_binding in itertools.izip(bindings, agent_bindings):
self.assertEqual(binding['repo_id'], agent_binding['repo_id'])
self.assertEqual(distributor['distributor_type_id'], agent_binding['type_id'])
self.assertEqual(bind_payload, agent_binding['details'])
def test_get_agent_bindings(self, mock_factory):
bind_payload = {'a': 1, 'b': 2}
distributor = {'distributor_type_id': '3838'}
mock_distributor_manager = Mock()
mock_distributor_manager.get_distributor = Mock(return_value=distributor)
mock_distributor_manager.create_bind_payload = Mock(return_value=bind_payload)
mock_factory.repo_distributor_manager = Mock(return_value=mock_distributor_manager)
# test manager
bindings = [
{'consumer_id': '10', 'repo_id': '20', 'distributor_id': '30', 'binding_config': {}},
{'consumer_id': '40', 'repo_id': '50', 'distributor_id': '60', 'binding_config': {}},
]
agent_bindings = AgentManager._bindings(bindings)
# validation
for binding in bindings:
mock_distributor_manager.get_distributor.assert_any_call(
binding['repo_id'], binding['distributor_id'])
mock_distributor_manager.create_bind_payload.assert_any_call(
binding['repo_id'], binding['distributor_id'], binding['binding_config'])
self.assertEqual(len(agent_bindings), 2)
for binding, agent_binding in itertools.izip(bindings, agent_bindings):
self.assertEqual(binding['repo_id'], agent_binding['repo_id'])
self.assertEqual(distributor['distributor_type_id'], agent_binding['type_id'])
self.assertEqual(bind_payload, agent_binding['details'])
def test_get_agent_bindings(self, m_model, m_dist_cont):
bind_payload = {'a': 1, 'b': 2}
distributor = model.Distributor(distributor_type_id='3838')
m_model.Distributor.objects.get_or_404.return_value = distributor
m_dist_cont.create_bind_payload = Mock(return_value=bind_payload)
bindings = [
{'consumer_id': '10', 'repo_id': '20', 'distributor_id': '30', 'binding_config': {}},
{'consumer_id': '40', 'repo_id': '50', 'distributor_id': '60', 'binding_config': {}},
]
agent_bindings = AgentManager._bindings(bindings)
for binding in bindings:
m_model.Distributor.objects.get_or_404.assert_any_call(
repo_id=binding['repo_id'], distributor_id=binding['distributor_id'])
m_dist_cont.create_bind_payload.assert_any_call(
binding['repo_id'], binding['distributor_id'], binding['binding_config'])
self.assertEqual(len(agent_bindings), 2)
for binding, agent_binding in itertools.izip(bindings, agent_bindings):
self.assertEqual(binding['repo_id'], agent_binding['repo_id'])
self.assertEqual(distributor['distributor_type_id'], agent_binding['type_id'])
self.assertEqual(bind_payload, agent_binding['details'])
def test_get_agent_bindings(self, m_model, m_dist_cont):
bind_payload = {'a': 1, 'b': 2}
distributor = model.Distributor(distributor_type_id='3838')
m_model.Distributor.objects.get_or_404.return_value = distributor
m_dist_cont.create_bind_payload = Mock(return_value=bind_payload)
bindings = [
{
'consumer_id': '10',
'repo_id': '20',
'distributor_id': '30',
'binding_config': {}
},
{
'consumer_id': '40',
'repo_id': '50',
'distributor_id': '60',
'binding_config': {}
},
]
agent_bindings = AgentManager._bindings(bindings)
for binding in bindings:
m_model.Distributor.objects.get_or_404.assert_any_call(
repo_id=binding['repo_id'],
distributor_id=binding['distributor_id'])
m_dist_cont.create_bind_payload.assert_any_call(
binding['repo_id'], binding['distributor_id'],
binding['binding_config'])
self.assertEqual(len(agent_bindings), 2)
for binding, agent_binding in itertools.izip(bindings, agent_bindings):
self.assertEqual(binding['repo_id'], agent_binding['repo_id'])
self.assertEqual(distributor['distributor_type_id'],
agent_binding['type_id'])
self.assertEqual(bind_payload, agent_binding['details'])
