test_instance.py 文件源码

def test_create_success(self):
        req = dict(
            container_name=test_cfg.CONTAINER_NAME,
            image_id=test_cfg.CONTAINER_IMAGE_ID,
            user_name=test_cfg.USER_NAME
        )
        with mock.patch('pika.BlockingConnection') as mock_block,\
            mock.patch.object(pika.BlockingConnection,'channel') as mock_block_channel,\
            mock.patch.object(UiQueue, 'send') as mock_queue_send:
            worker_res = dict(
                code='0x1',
                message='pass',
                ins={}
            )
            mock_block.return_value = mock.Mock()
            mock_block_channel.return_value = mock.Mock()
            mock_queue_send.return_value = json.dumps(worker_res)
            response = self.app.post('/create_ins', data = json.dumps(req), follow_redirects=True)
            mock_queue_send.assert_called_once()
            res_dict = json.loads(response.data)
            self.assertEqual(res_dict.get('code'), '0x1')

    # Container name occupied by others