def test_list_block(self, LoadBalancer): lb = LoadBalancer.find.return_value lb.hosts = [mock.Mock(), mock.Mock()] manager = Manager(self.config) manager.consul_manager = mock.Mock() manager.consul_manager.list_blocks.return_value = [{ u'block_name': 'server', u'content': 'something nice in server' }, { u'block_name': 'http', u'content': 'something nice in http' }] blocks = manager.list_blocks("inst") self.assertDictEqual(blocks[0], { 'block_name': 'server', 'content': 'something nice in server' }) self.assertDictEqual(blocks[1], { 'block_name': 'http', 'content': 'something nice in http' }) LoadBalancer.find.assert_called_with("inst") manager.consul_manager.list_blocks.assert_called_with("inst")
def test_list_block(self, LoadBalancer): lb = LoadBalancer.find.return_value lb.hosts = [mock.Mock(), mock.Mock()] manager = Manager(self.config) manager.consul_manager = mock.Mock() manager.consul_manager.list_blocks.return_value = [ '9796', [{u'LockIndex': 0, u'ModifyIndex': 9796, u'Value': 'something nice in server', u'Flags': 0, u'Key': u'test-suite-rpaas/myrpaas/blocks/server/ROOT', u'CreateIndex': 9796}, {u'LockIndex': 0, u'ModifyIndex': 9796, u'Value': 'something nice in http', u'Flags': 0, u'Key': u'test-suite-rpaas/myrpaas/blocks/http/ROOT', u'CreateIndex': 9796}] ] blocks = manager.list_blocks("inst") self.assertDictEqual(blocks[0], {'block_name': 'server', 'content': 'something nice in server'}) self.assertDictEqual(blocks[1], {'block_name': 'http', 'content': 'something nice in http'}) LoadBalancer.find.assert_called_with("inst") manager.consul_manager.list_blocks.assert_called_with("inst")
def test_empty_list_blocks(self, LoadBalancer): lb = LoadBalancer.find.return_value lb.hosts = [mock.Mock(), mock.Mock()] manager = Manager(self.config) manager.consul_manager = mock.Mock() manager.consul_manager.list_blocks.return_value = [] blocks = manager.list_blocks("inst") self.assertEqual(blocks, []) LoadBalancer.find.assert_called_with("inst") manager.consul_manager.list_blocks.assert_called_with("inst")
def test_list_block(self, LoadBalancer): lb = LoadBalancer.find.return_value lb.hosts = [mock.Mock(), mock.Mock()] manager = Manager(self.config) manager.consul_manager = mock.Mock() manager.consul_manager.list_blocks.return_value = [ {u'block_name': 'server', u'content': 'something nice in server'}, {u'block_name': 'http', u'content': 'something nice in http'} ] blocks = manager.list_blocks("inst") self.assertDictEqual(blocks[0], {'block_name': 'server', 'content': 'something nice in server'}) self.assertDictEqual(blocks[1], {'block_name': 'http', 'content': 'something nice in http'}) LoadBalancer.find.assert_called_with("inst") manager.consul_manager.list_blocks.assert_called_with("inst")