117 lines
4.4 KiB
Python
117 lines
4.4 KiB
Python
# Copyright 2012 Nebula, Inc.
|
|
# Copyright 2013 IBM Corp.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from oslo_config import cfg
|
|
|
|
from nova import db
|
|
from nova.db.sqlalchemy import models
|
|
from nova.tests.functional.v3 import api_sample_base
|
|
|
|
CONF = cfg.CONF
|
|
CONF.import_opt('osapi_compute_extension',
|
|
'nova.api.openstack.compute.legacy_v2.extensions')
|
|
|
|
|
|
class AgentsJsonTest(api_sample_base.ApiSampleTestBaseV3):
|
|
ADMIN_API = True
|
|
extension_name = "os-agents"
|
|
# TODO(gmann): Overriding '_api_version' till all functional tests
|
|
# are merged between v2 and v2.1. After that base class variable
|
|
# itself can be changed to 'v2'
|
|
_api_version = 'v2'
|
|
|
|
def _get_flags(self):
|
|
f = super(AgentsJsonTest, self)._get_flags()
|
|
f['osapi_compute_extension'] = CONF.osapi_compute_extension[:]
|
|
f['osapi_compute_extension'].append(
|
|
'nova.api.openstack.compute.contrib.agents.Agents')
|
|
return f
|
|
|
|
def setUp(self):
|
|
super(AgentsJsonTest, self).setUp()
|
|
|
|
fake_agents_list = [{'url': 'http://example.com/path/to/resource',
|
|
'hypervisor': 'hypervisor',
|
|
'architecture': 'x86',
|
|
'os': 'os',
|
|
'version': '8.0',
|
|
'md5hash': 'add6bb58e139be103324d04d82d8f545',
|
|
'id': 1}]
|
|
|
|
def fake_agent_build_create(context, values):
|
|
values['id'] = 1
|
|
agent_build_ref = models.AgentBuild()
|
|
agent_build_ref.update(values)
|
|
return agent_build_ref
|
|
|
|
def fake_agent_build_get_all(context, hypervisor):
|
|
agent_build_all = []
|
|
for agent in fake_agents_list:
|
|
if hypervisor and hypervisor != agent['hypervisor']:
|
|
continue
|
|
agent_build_ref = models.AgentBuild()
|
|
agent_build_ref.update(agent)
|
|
agent_build_all.append(agent_build_ref)
|
|
return agent_build_all
|
|
|
|
def fake_agent_build_update(context, agent_build_id, values):
|
|
pass
|
|
|
|
def fake_agent_build_destroy(context, agent_update_id):
|
|
pass
|
|
|
|
self.stubs.Set(db, "agent_build_create",
|
|
fake_agent_build_create)
|
|
self.stubs.Set(db, "agent_build_get_all",
|
|
fake_agent_build_get_all)
|
|
self.stubs.Set(db, "agent_build_update",
|
|
fake_agent_build_update)
|
|
self.stubs.Set(db, "agent_build_destroy",
|
|
fake_agent_build_destroy)
|
|
|
|
def test_agent_create(self):
|
|
# Creates a new agent build.
|
|
project = {'url': 'http://example.com/path/to/resource',
|
|
'hypervisor': 'hypervisor',
|
|
'architecture': 'x86',
|
|
'os': 'os',
|
|
'version': '8.0',
|
|
'md5hash': 'add6bb58e139be103324d04d82d8f545'
|
|
}
|
|
response = self._do_post('os-agents', 'agent-post-req',
|
|
project)
|
|
self._verify_response('agent-post-resp', project, response, 200)
|
|
|
|
def test_agent_list(self):
|
|
# Return a list of all agent builds.
|
|
response = self._do_get('os-agents')
|
|
self._verify_response('agents-get-resp', {}, response, 200)
|
|
|
|
def test_agent_update(self):
|
|
# Update an existing agent build.
|
|
agent_id = 1
|
|
subs = {'version': '7.0',
|
|
'url': 'http://example.com/path/to/resource',
|
|
'md5hash': 'add6bb58e139be103324d04d82d8f545'}
|
|
response = self._do_put('os-agents/%s' % agent_id,
|
|
'agent-update-put-req', subs)
|
|
self._verify_response('agent-update-put-resp', subs, response, 200)
|
|
|
|
def test_agent_delete(self):
|
|
# Deletes an existing agent build.
|
|
agent_id = 1
|
|
response = self._do_delete('os-agents/%s' % agent_id)
|
|
self.assertEqual(response.status_code, 200)
|