def test_add_agent_queues_cat_only(self, fake_get): fake_collection = Mock() fake_collection.find = Mock(return_value=[{'id': 'dog'}, {'id': 'cat'}]) fake_get.return_value = fake_collection fake_broker = Mock() fake_broker.getQueue.side_effect = [None, Mock()] # test migration = MigrationModule(MIGRATION)._module migration._add_agent_queues(fake_broker) fake_broker.getQueue.assert_any('pulp.agent.dog') fake_broker.getQueue.assert_any('pulp.agent.cat') fake_broker.addQueue.assert_calle_with('pulp.agent.cat')
def test_add_agent_queues(self, fake_get): fake_collection = Mock() fake_collection.find = Mock(return_value=[{ 'id': 'dog' }, { 'id': 'cat' }]) fake_get.return_value = fake_collection fake_broker = Mock() fake_broker.getQueue.side_effect = [None, None] # test migration = MigrationModule(MIGRATION)._module migration._add_agent_queues(fake_broker) fake_broker.getQueue.assert_any('pulp.agent.dog') fake_broker.getQueue.assert_any('pulp.agent.cat') fake_broker.addQueue.assert_any('pulp.agent.dog') fake_broker.addQueue.assert_any('pulp.agent.cat')
class TestMigration(TestCase): def setUp(self): # Remove dependencies on qpidtoollibs and qpid.messaging self.migration = MigrationModule(MIGRATION)._module if self.migration.QPIDTOOLLIBS_AVAILABLE: self.broker = self.migration.BrokerAgent if self.migration.QPID_MESSAGING_AVAILABLE: self.connection = self.migration.Connection self.qpidtoollibs = self.migration.QPIDTOOLLIBS_AVAILABLE self.qpid_messaging = self.migration.QPID_MESSAGING_AVAILABLE self.migration.QPIDTOOLLIBS_AVAILABLE = True self.migration.QPID_MESSAGING_AVAILABLE = True self.migration.BrokerAgent = Mock() self.migration.Connection = Mock() def tearDown(self): self.migration.QPIDTOOLLIBS_AVAILABLE = self.qpidtoollibs self.migration.QPID_MESSAGING_AVAILABLE = self.qpid_messaging if self.migration.QPIDTOOLLIBS_AVAILABLE: self.migration.BrokerAgent = self.broker if self.migration.QPID_MESSAGING_AVAILABLE: self.migration.Connection = self.connection @patch(MIGRATION + '._migrate_reply_queue') @patch(MIGRATION + '._migrate_agent_queues') @patch(MIGRATION + '.pulp_conf', PULP_CONF) def test_migrate(self, fake_migrate_agent_queues, fake_migrate_reply_queue): # test self.migration.migrate() # validation self.migration.Connection.assert_called_with( host='myhost', port=1234, transport='tcp', reconnect=False, ssl_certfile='TEST-CERTIFICATE', ssl_skip_hostname_check=True) self.migration.Connection().attach.assert_called_with() self.migration.BrokerAgent.assert_called_with(self.migration.Connection()) fake_migrate_reply_queue.assert_called_with(self.migration.BrokerAgent()) fake_migrate_agent_queues.assert_called_with(self.migration.BrokerAgent()) self.migration.Connection().detach.assert_called_with() @patch(MIGRATION + '._migrate_reply_queue') @patch(MIGRATION + '._migrate_agent_queues') @patch(MIGRATION + '.pulp_conf') def test_migrate_not_qpid(self, fake_conf, fake_migrate_agent_queues, fake_migrate_reply_queue,): fake_conf.get.return_value = 'not-qpid' # test self.migration.migrate() # validation self.assertFalse(self.migration.Connection.called) self.assertFalse(self.migration.BrokerAgent.called) self.assertFalse(fake_migrate_reply_queue.called) self.assertFalse(fake_migrate_agent_queues.called) @patch(MIGRATION + '._del_agent_queues') @patch(MIGRATION + '._add_agent_queues') def test_migrate_agent_queues(self, fake_add_agent_queues, fake_del_agent_queues): fake_broker = Mock() # test self.migration._migrate_agent_queues(fake_broker) # validation fake_add_agent_queues.assert_called_with(fake_broker) fake_del_agent_queues.assert_called_with(fake_broker) @patch('pulp.server.db.model.consumer.Consumer.get_collection') def test_add_agent_queues(self, fake_get): fake_collection = Mock() fake_collection.find = Mock(return_value=[{'id': 'dog'}, {'id': 'cat'}]) fake_get.return_value = fake_collection fake_broker = Mock() fake_broker.getQueue.side_effect = [None, None] # test self.migration._add_agent_queues(fake_broker) fake_broker.getQueue.assert_any('pulp.agent.dog') fake_broker.getQueue.assert_any('pulp.agent.cat') fake_broker.addQueue.assert_any('pulp.agent.dog') fake_broker.addQueue.assert_any('pulp.agent.cat') @patch('pulp.server.db.model.consumer.Consumer.get_collection') def test_add_agent_queues_cat_only(self, fake_get): fake_collection = Mock() fake_collection.find = Mock(return_value=[{'id': 'dog'}, {'id': 'cat'}]) fake_get.return_value = fake_collection fake_broker = Mock() fake_broker.getQueue.side_effect = [None, Mock()] # test self.migration._add_agent_queues(fake_broker) fake_broker.getQueue.assert_any('pulp.agent.dog') fake_broker.getQueue.assert_any('pulp.agent.cat') fake_broker.addQueue.assert_called__once_with('pulp.agent.dog', durable=True)
class TestMigration(TestCase): def setUp(self): # Remove dependencies on qpidtoollibs and qpid.messaging self.migration = MigrationModule(MIGRATION)._module if self.migration.QPIDTOOLLIBS_AVAILABLE: self.broker = self.migration.BrokerAgent if self.migration.QPID_MESSAGING_AVAILABLE: self.connection = self.migration.Connection self.qpidtoollibs = self.migration.QPIDTOOLLIBS_AVAILABLE self.qpid_messaging = self.migration.QPID_MESSAGING_AVAILABLE self.migration.QPIDTOOLLIBS_AVAILABLE = True self.migration.QPID_MESSAGING_AVAILABLE = True self.migration.BrokerAgent = Mock() self.migration.Connection = Mock() def tearDown(self): self.migration.QPIDTOOLLIBS_AVAILABLE = self.qpidtoollibs self.migration.QPID_MESSAGING_AVAILABLE = self.qpid_messaging if self.migration.QPIDTOOLLIBS_AVAILABLE: self.migration.BrokerAgent = self.broker if self.migration.QPID_MESSAGING_AVAILABLE: self.migration.Connection = self.connection @patch(MIGRATION + '._migrate_reply_queue') @patch(MIGRATION + '._migrate_agent_queues') @patch(MIGRATION + '.pulp_conf', PULP_CONF) def test_migrate(self, fake_migrate_agent_queues, fake_migrate_reply_queue): # test self.migration.migrate() # validation self.migration.Connection.assert_called_with( host='myhost', port=1234, transport='tcp', reconnect=False, ssl_certfile='TEST-CERTIFICATE', ssl_skip_hostname_check=True) self.migration.Connection().attach.assert_called_with() self.migration.BrokerAgent.assert_called_with( self.migration.Connection()) fake_migrate_reply_queue.assert_called_with( self.migration.BrokerAgent()) fake_migrate_agent_queues.assert_called_with( self.migration.BrokerAgent()) self.migration.Connection().detach.assert_called_with() @patch(MIGRATION + '._migrate_reply_queue') @patch(MIGRATION + '._migrate_agent_queues') @patch(MIGRATION + '.pulp_conf') def test_migrate_not_qpid( self, fake_conf, fake_migrate_agent_queues, fake_migrate_reply_queue, ): fake_conf.get.return_value = 'not-qpid' # test self.migration.migrate() # validation self.assertFalse(self.migration.Connection.called) self.assertFalse(self.migration.BrokerAgent.called) self.assertFalse(fake_migrate_reply_queue.called) self.assertFalse(fake_migrate_agent_queues.called) @patch(MIGRATION + '._del_agent_queues') @patch(MIGRATION + '._add_agent_queues') def test_migrate_agent_queues(self, fake_add_agent_queues, fake_del_agent_queues): fake_broker = Mock() # test self.migration._migrate_agent_queues(fake_broker) # validation fake_add_agent_queues.assert_called_with(fake_broker) fake_del_agent_queues.assert_called_with(fake_broker) @patch('pulp.server.db.model.consumer.Consumer.get_collection') def test_add_agent_queues(self, fake_get): fake_collection = Mock() fake_collection.find = Mock(return_value=[{ 'id': 'dog' }, { 'id': 'cat' }]) fake_get.return_value = fake_collection fake_broker = Mock() fake_broker.getQueue.side_effect = [None, None] # test self.migration._add_agent_queues(fake_broker) fake_broker.getQueue.assert_any('pulp.agent.dog') fake_broker.getQueue.assert_any('pulp.agent.cat') fake_broker.addQueue.assert_any('pulp.agent.dog') fake_broker.addQueue.assert_any('pulp.agent.cat') @patch('pulp.server.db.model.consumer.Consumer.get_collection') def test_add_agent_queues_cat_only(self, fake_get): fake_collection = Mock() fake_collection.find = Mock(return_value=[{ 'id': 'dog' }, { 'id': 'cat' }]) fake_get.return_value = fake_collection fake_broker = Mock() fake_broker.getQueue.side_effect = [None, Mock()] # test self.migration._add_agent_queues(fake_broker) fake_broker.getQueue.assert_any('pulp.agent.dog') fake_broker.getQueue.assert_any('pulp.agent.cat') fake_broker.addQueue.assert_called__once_with('pulp.agent.dog', durable=True)