dbccbb5cb9
We will eventually need the compute agent to accept instructions to poll the status of an instance before nova deletes it (for details see bug 1005944). This change prepares for that fix by making the compute agent pollsters take an instance as an argument provided by the caller, instead of accessing the database directly to look up the instances known to be running on the current host. Change-Id: I6a16405fd65bfb3c190d02f3d70b2bde7fc0fc83 Signed-off-by: Doug Hellmann <doug.hellmann@dreamhost.com>
90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
# -*- encoding: utf-8 -*-
|
|
#
|
|
# Copyright © 2012 New Dream Network, LLC (DreamHost)
|
|
#
|
|
# Author: Doug Hellmann <doug.hellmann@dreamhost.com>
|
|
#
|
|
# 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.
|
|
"""Tests for ceilometer/agent/manager.py
|
|
"""
|
|
|
|
import datetime
|
|
|
|
from nova import context
|
|
from nova import test
|
|
|
|
from ceilometer.compute import manager
|
|
from ceilometer import counter
|
|
from ceilometer import publish
|
|
|
|
|
|
def test_load_plugins():
|
|
mgr = manager.AgentManager()
|
|
mgr.init_host()
|
|
assert mgr.pollsters, 'Failed to load any plugins'
|
|
return
|
|
|
|
|
|
class TestRunTasks(test.TestCase):
|
|
|
|
class Pollster:
|
|
counters = []
|
|
test_data = counter.Counter(
|
|
source='test',
|
|
name='test',
|
|
type='cumulative',
|
|
volume=1,
|
|
user_id='test',
|
|
project_id='test',
|
|
resource_id='test_run_tasks',
|
|
timestamp=datetime.datetime.utcnow().isoformat(),
|
|
duration=0,
|
|
resource_metadata={'name': 'Pollster',
|
|
},
|
|
)
|
|
|
|
def get_counters(self, manager, instance):
|
|
self.counters.append((manager, instance))
|
|
return [self.test_data]
|
|
|
|
def faux_notify(self, context, msg):
|
|
self.notifications.append(msg)
|
|
|
|
def setUp(self):
|
|
super(TestRunTasks, self).setUp()
|
|
self.notifications = []
|
|
self.stubs.Set(publish, 'publish_counter', self.faux_notify)
|
|
self.mgr = manager.AgentManager()
|
|
self.mgr.pollsters = [('test', self.Pollster())]
|
|
self.ctx = context.RequestContext("user", "project")
|
|
# Set up a fake instance value to be returned by
|
|
# instance_get_all_by_host() so when the manager gets the list
|
|
# of instances to poll we can control the results.
|
|
self.instance = 'faux instance'
|
|
self.mox.StubOutWithMock(self.mgr.db, 'instance_get_all_by_host')
|
|
self.mgr.db.instance_get_all_by_host(
|
|
self.ctx,
|
|
self.mgr.host,
|
|
).AndReturn([self.instance])
|
|
|
|
self.mox.ReplayAll()
|
|
# Invoke the periodic tasks to call the pollsters.
|
|
self.mgr.periodic_tasks(self.ctx)
|
|
|
|
def test_message(self):
|
|
assert self.Pollster.counters[0][1] is self.instance
|
|
|
|
def test_notifications(self):
|
|
assert self.notifications[0] is self.Pollster.test_data
|
|
assert len(self.notifications) == 1
|