246 lines
9.3 KiB
Python
246 lines
9.3 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 twisted.internet import defer
|
|
|
|
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"""
|
|
def test_method(self):
|
|
return 'manager'
|
|
|
|
|
|
class ExtendedService(service.Service):
|
|
def test_method(self):
|
|
return 'service'
|
|
|
|
|
|
class ServiceManagerTestCase(test.TrialTestCase):
|
|
"""Test cases for Services"""
|
|
|
|
def test_attribute_error_for_no_manager(self):
|
|
serv = service.Service('test',
|
|
'test',
|
|
'test',
|
|
'nova.tests.service_unittest.FakeManager')
|
|
self.assertRaises(AttributeError, getattr, serv, 'test_method')
|
|
|
|
def test_message_gets_to_manager(self):
|
|
serv = service.Service('test',
|
|
'test',
|
|
'test',
|
|
'nova.tests.service_unittest.FakeManager')
|
|
serv.startService()
|
|
self.assertEqual(serv.test_method(), 'manager')
|
|
|
|
def test_override_manager_method(self):
|
|
serv = ExtendedService('test',
|
|
'test',
|
|
'test',
|
|
'nova.tests.service_unittest.FakeManager')
|
|
serv.startService()
|
|
self.assertEqual(serv.test_method(), 'service')
|
|
|
|
|
|
class ServiceTestCase(test.TrialTestCase):
|
|
"""Test cases for Services"""
|
|
|
|
def setUp(self):
|
|
super(ServiceTestCase, self).setUp()
|
|
self.mox.StubOutWithMock(service, 'db')
|
|
|
|
def test_create(self):
|
|
host = 'foo'
|
|
binary = 'nova-fake'
|
|
topic = 'fake'
|
|
|
|
# NOTE(vish): Create was moved out of mox replay to make sure that
|
|
# the looping calls are created in StartService.
|
|
app = service.Service.create(host=host, binary=binary)
|
|
|
|
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()
|
|
|
|
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
|
|
@defer.inlineCallbacks
|
|
def test_report_state_no_service(self):
|
|
host = 'foo'
|
|
binary = 'bar'
|
|
topic = 'test'
|
|
service_create = {'host': host,
|
|
'binary': binary,
|
|
'topic': topic,
|
|
'report_count': 0}
|
|
service_ref = {'host': host,
|
|
'binary': binary,
|
|
'topic': topic,
|
|
'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)
|
|
service.db.service_get(mox.IgnoreArg(),
|
|
service_ref['id']).AndReturn(service_ref)
|
|
service.db.service_update(mox.IgnoreArg(), service_ref['id'],
|
|
mox.ContainsKeyValue('report_count', 1))
|
|
|
|
self.mox.ReplayAll()
|
|
serv = service.Service(host,
|
|
binary,
|
|
topic,
|
|
'nova.tests.service_unittest.FakeManager')
|
|
serv.startService()
|
|
yield serv.report_state()
|
|
|
|
@defer.inlineCallbacks
|
|
def test_report_state_newly_disconnected(self):
|
|
host = 'foo'
|
|
binary = 'bar'
|
|
topic = 'test'
|
|
service_create = {'host': host,
|
|
'binary': binary,
|
|
'topic': topic,
|
|
'report_count': 0}
|
|
service_ref = {'host': host,
|
|
'binary': binary,
|
|
'topic': topic,
|
|
'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)
|
|
service.db.service_get(mox.IgnoreArg(),
|
|
mox.IgnoreArg()).AndRaise(Exception())
|
|
|
|
self.mox.ReplayAll()
|
|
serv = service.Service(host,
|
|
binary,
|
|
topic,
|
|
'nova.tests.service_unittest.FakeManager')
|
|
serv.startService()
|
|
yield serv.report_state()
|
|
self.assert_(serv.model_disconnected)
|
|
|
|
@defer.inlineCallbacks
|
|
def test_report_state_newly_connected(self):
|
|
host = 'foo'
|
|
binary = 'bar'
|
|
topic = 'test'
|
|
service_create = {'host': host,
|
|
'binary': binary,
|
|
'topic': topic,
|
|
'report_count': 0}
|
|
service_ref = {'host': host,
|
|
'binary': binary,
|
|
'topic': topic,
|
|
'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)
|
|
service.db.service_get(mox.IgnoreArg(),
|
|
service_ref['id']).AndReturn(service_ref)
|
|
service.db.service_update(mox.IgnoreArg(), service_ref['id'],
|
|
mox.ContainsKeyValue('report_count', 1))
|
|
|
|
self.mox.ReplayAll()
|
|
serv = service.Service(host,
|
|
binary,
|
|
topic,
|
|
'nova.tests.service_unittest.FakeManager')
|
|
serv.startService()
|
|
serv.model_disconnected = True
|
|
yield serv.report_state()
|
|
|
|
self.assert_(not serv.model_disconnected)
|