def test_get_random_topic_host_with_domain(self, mock_get_coordinator): mock_coordinator = mock.Mock(spec=['get_members']) members = ['ironic_inspector.conductor.' 'local.domain'.encode('ascii')] mock_coordinator.get_members.return_value = set(members) mock_get_coordinator.return_value = mock_coordinator topic = main.get_random_topic() self.assertEqual(topic, 'ironic_inspector.conductor.local.domain')
def test_get_random_topic(self, mock_get_coordinator): mock_coordinator = mock.Mock(spec=['get_members']) members = [('ironic_inspector.conductor.host%s' % i).encode('ascii') for i in range(5)] topics = [('ironic_inspector.conductor.host%s' % i) for i in range(5)] mock_coordinator.get_members.return_value = set(members) mock_get_coordinator.return_value = mock_coordinator for i in range(10): topic = main.get_random_topic() self.assertIn(topic, topics)