193 lines
7.1 KiB
Python
193 lines
7.1 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2010 United States Government as represented by the
|
|
# Administrator of the National Aeronautics and Space Administration.
|
|
# All Rights Reserved.
|
|
#
|
|
# 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.
|
|
|
|
"""
|
|
Unit Tests for remote procedure calls using queue
|
|
"""
|
|
|
|
import mox
|
|
|
|
from twisted.application.app import startApplication
|
|
|
|
from nova import context
|
|
from nova import exception
|
|
from nova import flags
|
|
from nova import rpc
|
|
from nova import test
|
|
from nova import service
|
|
from nova import manager
|
|
|
|
FLAGS = flags.FLAGS
|
|
flags.DEFINE_string("fake_manager", "nova.tests.service_unittest.FakeManager",
|
|
"Manager for testing")
|
|
|
|
|
|
class FakeManager(manager.Manager):
|
|
"""Fake manager for tests"""
|
|
pass
|
|
|
|
|
|
class ServiceTestCase(test.BaseTestCase):
|
|
"""Test cases for rpc"""
|
|
|
|
def setUp(self): # pylint: disable=C0103
|
|
super(ServiceTestCase, self).setUp()
|
|
self.mox.StubOutWithMock(service, 'db')
|
|
self.context = context.get_admin_context()
|
|
|
|
def test_create(self):
|
|
host = 'foo'
|
|
binary = 'nova-fake'
|
|
topic = 'fake'
|
|
self.mox.StubOutWithMock(rpc,
|
|
'AdapterConsumer',
|
|
use_mock_anything=True)
|
|
self.mox.StubOutWithMock(
|
|
service.task, 'LoopingCall', use_mock_anything=True)
|
|
rpc.AdapterConsumer(connection=mox.IgnoreArg(),
|
|
topic=topic,
|
|
proxy=mox.IsA(service.Service)).AndReturn(
|
|
rpc.AdapterConsumer)
|
|
|
|
rpc.AdapterConsumer(connection=mox.IgnoreArg(),
|
|
topic='%s.%s' % (topic, host),
|
|
proxy=mox.IsA(service.Service)).AndReturn(
|
|
rpc.AdapterConsumer)
|
|
|
|
rpc.AdapterConsumer.attach_to_twisted()
|
|
rpc.AdapterConsumer.attach_to_twisted()
|
|
|
|
# Stub out looping call a bit needlessly since we don't have an easy
|
|
# way to cancel it (yet) when the tests finishes
|
|
service.task.LoopingCall(mox.IgnoreArg()).AndReturn(
|
|
service.task.LoopingCall)
|
|
service.task.LoopingCall.start(interval=mox.IgnoreArg(),
|
|
now=mox.IgnoreArg())
|
|
service.task.LoopingCall(mox.IgnoreArg()).AndReturn(
|
|
service.task.LoopingCall)
|
|
service.task.LoopingCall.start(interval=mox.IgnoreArg(),
|
|
now=mox.IgnoreArg())
|
|
|
|
service_create = {'host': host,
|
|
'binary': binary,
|
|
'topic': topic,
|
|
'report_count': 0}
|
|
service_ref = {'host': host,
|
|
'binary': binary,
|
|
'report_count': 0,
|
|
'id': 1}
|
|
|
|
service.db.service_get_by_args(mox.IgnoreArg(),
|
|
host,
|
|
binary).AndRaise(exception.NotFound())
|
|
service.db.service_create(mox.IgnoreArg(),
|
|
service_create).AndReturn(service_ref)
|
|
self.mox.ReplayAll()
|
|
|
|
app = service.Service.create(host=host, binary=binary)
|
|
startApplication(app, False)
|
|
self.assert_(app)
|
|
|
|
# We're testing sort of weird behavior in how report_state decides
|
|
# whether it is disconnected, it looks for a variable on itself called
|
|
# 'model_disconnected' and report_state doesn't really do much so this
|
|
# these are mostly just for coverage
|
|
def test_report_state(self):
|
|
host = 'foo'
|
|
binary = 'bar'
|
|
service_ref = {'host': host,
|
|
'binary': binary,
|
|
'report_count': 0,
|
|
'id': 1}
|
|
service.db.__getattr__('report_state')
|
|
service.db.service_get_by_args(self.context,
|
|
host,
|
|
binary).AndReturn(service_ref)
|
|
service.db.service_update(self.context, service_ref['id'],
|
|
mox.ContainsKeyValue('report_count', 1))
|
|
|
|
self.mox.ReplayAll()
|
|
s = service.Service()
|
|
rv = yield s.report_state(host, binary)
|
|
|
|
def test_report_state_no_service(self):
|
|
host = 'foo'
|
|
binary = 'bar'
|
|
service_create = {'host': host,
|
|
'binary': binary,
|
|
'report_count': 0}
|
|
service_ref = {'host': host,
|
|
'binary': binary,
|
|
'report_count': 0,
|
|
'id': 1}
|
|
|
|
service.db.__getattr__('report_state')
|
|
service.db.service_get_by_args(self.context,
|
|
host,
|
|
binary).AndRaise(exception.NotFound())
|
|
service.db.service_create(self.context,
|
|
service_create).AndReturn(service_ref)
|
|
service.db.service_get(self.context, service_ref['id']).AndReturn(service_ref)
|
|
service.db.service_update(self.context, service_ref['id'],
|
|
mox.ContainsKeyValue('report_count', 1))
|
|
|
|
self.mox.ReplayAll()
|
|
s = service.Service()
|
|
rv = yield s.report_state(host, binary)
|
|
|
|
def test_report_state_newly_disconnected(self):
|
|
host = 'foo'
|
|
binary = 'bar'
|
|
service_ref = {'host': host,
|
|
'binary': binary,
|
|
'report_count': 0,
|
|
'id': 1}
|
|
|
|
service.db.__getattr__('report_state')
|
|
service.db.service_get_by_args(self.context,
|
|
host,
|
|
binary).AndRaise(Exception())
|
|
|
|
self.mox.ReplayAll()
|
|
s = service.Service()
|
|
rv = yield s.report_state(host, binary)
|
|
|
|
self.assert_(s.model_disconnected)
|
|
|
|
def test_report_state_newly_connected(self):
|
|
host = 'foo'
|
|
binary = 'bar'
|
|
service_ref = {'host': host,
|
|
'binary': binary,
|
|
'report_count': 0,
|
|
'id': 1}
|
|
|
|
service.db.__getattr__('report_state')
|
|
service.db.service_get_by_args(self.context,
|
|
host,
|
|
binary).AndReturn(service_ref)
|
|
service.db.service_update(self.context, service_ref['id'],
|
|
mox.ContainsKeyValue('report_count', 1))
|
|
|
|
self.mox.ReplayAll()
|
|
s = service.Service()
|
|
s.model_disconnected = True
|
|
rv = yield s.report_state(host, binary)
|
|
|
|
self.assert_(not s.model_disconnected)
|