def setUp(self): super(RPCAPITestCase, self).setUp() self.fake_bay = dbutils.get_test_bay(driver='fake-driver') self.fake_container = dbutils.get_test_container(driver='fake-driver') self.fake_pod = dbutils.get_test_pod(driver='fake-driver') self.fake_rc = dbutils.get_test_rc(driver='fake-driver') self.fake_service = dbutils.get_test_service(driver='fake-driver')
def setUp(self): super(RPCAPITestCase, self).setUp() self.fake_bay = dbutils.get_test_bay(driver='fake-driver') self.fake_container = dbutils.get_test_container(driver='fake-driver') self.fake_rc = dbutils.get_test_rc(driver='fake-driver') self.fake_certificate = objects.Certificate.from_db_bay(self.fake_bay) self.fake_certificate.csr = 'fake-csr'
def test_get_logs_put_fails(self, mock_get_by_uuid, mock_container_logs): test_container = utils.get_test_container() test_container_obj = objects.Container(self.context, **test_container) mock_get_by_uuid.return_value = test_container_obj container_uuid = test_container.get('uuid') self.assertRaises(AppError, self.app.put, '/v1/containers/%s/logs' % container_uuid) self.assertFalse(mock_container_logs.called)
def test_policy_disallow_update(self): test_container = utils.get_test_container() container_uuid = test_container.get('uuid') params = [{'path': '/name', 'value': 'new_name', 'op': 'replace'}] self._common_policy_check( 'container:update', self.app.patch_json, '/v1/containers/%s' % container_uuid, params)
def test_get_logs_by_uuid(self, mock_get_by_uuid, mock_container_logs): test_container = utils.get_test_container() test_container_obj = objects.Container(self.context, **test_container) mock_get_by_uuid.return_value = test_container_obj container_uuid = test_container.get('uuid') response = self.app.get('/v1/containers/%s/logs' % container_uuid) self.assertEqual(response.status_int, 200) mock_container_logs.assert_called_once_with(container_uuid)
def setUp(self): super(RPCAPITestCase, self).setUp() self.fake_bay = dbutils.get_test_bay(driver='fake-driver') self.fake_container = dbutils.get_test_container(driver='fake-driver') self.fake_rc = dbutils.get_test_rc(driver='fake-driver') self.fake_service = dbutils.get_test_service(driver='fake-driver') self.fake_x509keypair = dbutils.get_test_x509keypair( driver='fake-driver') self.fake_certificate = objects.Certificate.from_db_bay(self.fake_bay) self.fake_certificate.csr = 'fake-csr'
def test_get_logs_by_uuid(self, mock_get_by_uuid, mock_container_logs): mock_container_logs.return_value = "" test_container = utils.get_test_container() test_container_obj = objects.Container(self.context, **test_container) mock_get_by_uuid.return_value = test_container_obj container_uuid = test_container.get('uuid') response = self.app.get('/v1/containers/%s/logs' % container_uuid) self.assertEqual(response.status_int, 200) mock_container_logs.assert_called_once_with(container_uuid)
def get_test_container(context, **kw): """Return a test container object with appropriate attributes. NOTE: The object leaves the attributes marked as changed, such that a create() could be used to commit it to the DB. """ db_container = db_utils.get_test_container(**kw) container = objects.Container(context) for key in db_container: setattr(container, key, db_container[key]) return container
def test_execute_command_by_uuid(self, mock_get_by_uuid, mock_container_execute): test_container = utils.get_test_container() test_container_obj = objects.Container(self.context, **test_container) mock_get_by_uuid.return_value = test_container_obj container_uuid = test_container.get('uuid') url = '/v1/containers/%s/%s' % (container_uuid, 'execute') cmd = {'command': 'ls'} response = self.app.put(url, cmd) self.assertEqual(response.status_int, 200) mock_container_execute.assert_called_one_with(container_uuid, cmd)
def test_get_logs_by_name(self, mock_get_by_name, mock_container_logs): mock_container_logs.return_value = "" test_container = utils.get_test_container() test_container_obj = objects.Container(self.context, **test_container) mock_get_by_name.return_value = test_container_obj container_name = test_container.get('name') container_uuid = test_container.get('uuid') response = self.app.get('/v1/containers/%s/logs' % container_name) self.assertEqual(200, response.status_int) mock_container_logs.assert_called_once_with(container_uuid)
def test_get_one_by_name(self, mock_container_get_by_name, mock_container_show): test_container = utils.get_test_container() test_container_obj = objects.Container(self.context, **test_container) mock_container_get_by_name.return_value = test_container_obj mock_container_show.return_value = test_container_obj response = self.app.get('/v1/containers/%s' % test_container['name']) mock_container_get_by_name.assert_called_once_with( mock.ANY, test_container['name']) self.assertEqual(response.status_int, 200) self.assertEqual(response.json['uuid'], test_container['uuid'])
def test_delete_container_by_uuid(self, mock_get_by_uuid, mock_container_delete): test_container = utils.get_test_container() test_container_obj = objects.Container(self.context, **test_container) mock_get_by_uuid.return_value = test_container_obj with patch.object(test_container_obj, 'destroy') as mock_destroy: container_uuid = test_container.get('uuid') response = self.app.delete('/v1/containers/%s' % container_uuid) self.assertEqual(response.status_int, 204) mock_container_delete.assert_called_once_with(container_uuid) mock_destroy.assert_called_once_with()
def test_get_one_by_name(self, mock_container_get_by_name, mock_container_show): test_container = utils.get_test_container() test_container_obj = objects.Container(self.context, **test_container) mock_container_get_by_name.return_value = test_container_obj mock_container_show.return_value = test_container_obj response = self.app.get('/v1/containers/%s' % test_container['name']) mock_container_get_by_name.assert_called_once_with(mock.ANY, test_container['name']) self.assertEqual(response.status_int, 200) self.assertEqual(response.json['uuid'], test_container['uuid'])
def test_execute_command_by_uuid(self, mock_get_by_uuid, mock_container_exec): mock_container_exec.return_value = "" test_container = utils.get_test_container() test_container_obj = objects.Container(self.context, **test_container) mock_get_by_uuid.return_value = test_container_obj container_uuid = test_container.get('uuid') url = '/v1/containers/%s/%s' % (container_uuid, 'execute') cmd = {'command': 'ls'} response = self.app.put(url, cmd) self.assertEqual(response.status_int, 200) mock_container_exec.assert_called_once_with(container_uuid, cmd['command'])
def test_patch_by_name(self, mock_container_get_by_name): test_container = utils.get_test_container() test_container_obj = objects.Container(self.context, **test_container) mock_container_get_by_name.return_value = test_container_obj with patch.object(test_container_obj, 'save') as mock_save: params = [{'path': '/name', 'value': 'new_name', 'op': 'replace'}] container_name = test_container.get('name') response = self.app.patch_json('/v1/containers/%s' % container_name, params=params) mock_save.assert_called_once_with() self.assertEqual(response.status_int, 200) self.assertEqual(test_container_obj.name, 'new_name')
def test_execute_command_by_name(self, mock_get_by_name, mock_container_exec): mock_container_exec.return_value = "" test_container = utils.get_test_container() test_container_obj = objects.Container(self.context, **test_container) mock_get_by_name.return_value = test_container_obj container_name = test_container.get('name') container_uuid = test_container.get('uuid') url = '/v1/containers/%s/%s' % (container_name, 'execute') cmd = {'command': 'ls'} response = self.app.put(url, cmd) self.assertEqual(200, response.status_int) mock_container_exec.assert_called_once_with(container_uuid, cmd['command'])
def test_get_all_containers(self, mock_container_list, mock_container_show): test_container = utils.get_test_container() containers = [objects.Container(self.context, **test_container)] mock_container_list.return_value = containers mock_container_show.return_value = containers[0] response = self.app.get('/v1/containers') mock_container_list.assert_called_once_with(mock.ANY, 1000, None, 'id', 'asc', filters=None) self.assertEqual(200, response.status_int) actual_containers = response.json['containers'] self.assertEqual(1, len(actual_containers)) self.assertEqual(test_container['uuid'], actual_containers[0].get('uuid'))
def test_patch_by_name(self, mock_container_get_by_name): test_container = utils.get_test_container() test_container_obj = objects.Container(self.context, **test_container) mock_container_get_by_name.return_value = test_container_obj with patch.object(test_container_obj, 'save') as mock_save: params = [{'path': '/name', 'value': 'new_name', 'op': 'replace'}] container_name = test_container.get('name') response = self.app.patch_json( '/v1/containers/%s' % container_name, params=params) mock_save.assert_called_once_with() self.assertEqual(response.status_int, 200) self.assertEqual(test_container_obj.name, 'new_name')
def test_get_all_containers(self, mock_container_list, mock_container_show): test_container = utils.get_test_container() containers = [objects.Container(self.context, **test_container)] mock_container_list.return_value = containers mock_container_show.return_value = containers[0] response = self.app.get('/v1/containers') mock_container_list.assert_called_once_with(mock.ANY, 1000, None, sort_dir='asc', sort_key='id') self.assertEqual(response.status_int, 200) actual_containers = response.json['containers'] self.assertEqual(len(actual_containers), 1) self.assertEqual(actual_containers[0].get('uuid'), test_container['uuid'])
def test_get_all_containers_with_exception(self, mock_container_list, mock_container_show): test_container = utils.get_test_container() containers = [objects.Container(self.context, **test_container)] mock_container_list.return_value = containers mock_container_show.side_effect = Exception response = self.app.get('/v1/containers') mock_container_list.assert_called_once_with(mock.ANY, 1000, None, sort_dir='asc', sort_key='id') self.assertEqual(200, response.status_int) actual_containers = response.json['containers'] self.assertEqual(1, len(actual_containers)) self.assertEqual(test_container['uuid'], actual_containers[0].get('uuid')) self.assertEqual(fields.ContainerStatus.UNKNOWN, actual_containers[0].get('status'))
def setUp(self): super(TestContainerObject, self).setUp() self.fake_container = utils.get_test_container()
def test_start_by_name(self, mock_container_start): mock_container_start.return_value = "" test_container = utils.get_test_container() self._action_test(test_container, 'start', 'name') mock_container_start.assert_called_once_with( test_container.get('uuid'))
def test_stop_by_uuid(self, mock_container_stop): mock_container_stop.return_value = "" test_container = utils.get_test_container() self._action_test(test_container, 'stop', 'uuid') mock_container_stop.assert_called_once_with( test_container.get('uuid'))
def test_unpause_by_uuid(self, mock_container_unpause): mock_container_unpause.return_value = "" test_container = utils.get_test_container() self._action_test(test_container, 'unpause', 'uuid') mock_container_unpause.assert_called_once_with( test_container.get('uuid'))
def test_reboot_by_name(self, mock_container_reboot): mock_container_reboot.return_value = "" test_container = utils.get_test_container() self._action_test(test_container, 'reboot', 'name') mock_container_reboot.assert_called_once_with( test_container.get('uuid'))
def test_unpause_by_name(self, mock_container_unpause): mock_container_unpause.return_value = "" test_container = utils.get_test_container() self._action_test(test_container, 'unpause', 'name') mock_container_unpause.assert_called_once_with( test_container.get('uuid'))
def test_stop_by_name(self, mock_container_stop): mock_container_stop.return_value = "" test_container = utils.get_test_container() self._action_test(test_container, 'stop', 'name') mock_container_stop.assert_called_once_with(test_container.get('uuid'))
def test_start_by_uuid(self, mock_container_start): mock_container_start.return_value = "" test_container = utils.get_test_container() self._action_test(test_container, 'start', 'uuid') mock_container_start.assert_called_once_with( test_container.get('uuid'))
def test_reboot_by_uuid(self, mock_container_reboot): mock_container_reboot.return_value = "" test_container = utils.get_test_container() self._action_test(test_container, 'reboot', 'uuid') mock_container_reboot.assert_called_once_with( test_container.get('uuid'))
def test_start_by_uuid(self, mock_container_start): test_container = utils.get_test_container() self._action_test(test_container, 'start', 'uuid') mock_container_start.assert_called_once_with( test_container.get('uuid'))
def test_stop_by_name(self, mock_container_stop): test_container = utils.get_test_container() self._action_test(test_container, 'stop', 'name') mock_container_stop.assert_called_once_with( test_container.get('uuid'))
def test_unpause_by_name(self, mock_container_unpause): test_container = utils.get_test_container() self._action_test(test_container, 'unpause', 'name') mock_container_unpause.assert_called_once_with( test_container.get('uuid'))
def test_reboot_by_name(self, mock_container_reboot): test_container = utils.get_test_container() self._action_test(test_container, 'reboot', 'name') mock_container_reboot.assert_called_once_with( test_container.get('uuid'))