129 lines
4.2 KiB
Python
129 lines
4.2 KiB
Python
# Copyright (c) 2016 VMware, Inc.
|
|
# 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.
|
|
|
|
import logging
|
|
import os
|
|
|
|
from oslo_config import cfg
|
|
|
|
from osprofiler.drivers import base
|
|
from osprofiler import initializer
|
|
from osprofiler import opts
|
|
from osprofiler import profiler
|
|
from osprofiler.tests import test
|
|
|
|
|
|
CONF = cfg.CONF
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
@profiler.trace_cls("rpc", hide_args=True)
|
|
class Foo(object):
|
|
def bar(self, x):
|
|
return self.baz(x, x)
|
|
|
|
def baz(self, x, y):
|
|
return x * y
|
|
|
|
|
|
class DriverTestCase(test.FunctionalTestCase):
|
|
|
|
SERVICE = "service"
|
|
PROJECT = "project"
|
|
|
|
def setUp(self):
|
|
super(DriverTestCase, self).setUp()
|
|
CONF(["--config-file", os.path.dirname(__file__) + "/config.cfg"])
|
|
opts.set_defaults(CONF,
|
|
enabled=True,
|
|
trace_sqlalchemy=False,
|
|
hmac_keys="SECRET_KEY")
|
|
|
|
def _assert_dict(self, info, **kwargs):
|
|
for key in kwargs:
|
|
self.assertEqual(kwargs[key], info[key])
|
|
|
|
def _assert_child_dict(self, child, base_id, parent_id, name, fn_name):
|
|
self.assertEqual(parent_id, child["parent_id"])
|
|
|
|
exp_info = {"name": "rpc",
|
|
"service": self.SERVICE,
|
|
"project": self.PROJECT}
|
|
self._assert_dict(child["info"], **exp_info)
|
|
|
|
raw_start = child["info"]["meta.raw_payload.%s-start" % name]
|
|
self.assertEqual(fn_name, raw_start["info"]["function"]["name"])
|
|
exp_raw = {"name": "%s-start" % name,
|
|
"service": self.SERVICE,
|
|
"trace_id": child["trace_id"],
|
|
"project": self.PROJECT,
|
|
"base_id": base_id}
|
|
self._assert_dict(raw_start, **exp_raw)
|
|
|
|
raw_stop = child["info"]["meta.raw_payload.%s-stop" % name]
|
|
exp_raw["name"] = "%s-stop" % name
|
|
self._assert_dict(raw_stop, **exp_raw)
|
|
|
|
def test_get_report(self):
|
|
# initialize profiler notifier (the same way as in services)
|
|
initializer.init_from_conf(
|
|
CONF, {}, self.PROJECT, self.SERVICE, "host")
|
|
profiler.init("SECRET_KEY")
|
|
|
|
# grab base_id
|
|
base_id = profiler.get().get_base_id()
|
|
|
|
# execute profiled code
|
|
foo = Foo()
|
|
foo.bar(1)
|
|
|
|
# instantiate report engine (the same way as in osprofiler CLI)
|
|
engine = base.get_driver(CONF.profiler.connection_string,
|
|
project=self.PROJECT,
|
|
service=self.SERVICE,
|
|
host="host",
|
|
conf=CONF)
|
|
|
|
# generate the report
|
|
report = engine.get_report(base_id)
|
|
LOG.debug("OSProfiler report: %s", report)
|
|
|
|
# verify the report
|
|
self.assertEqual("total", report["info"]["name"])
|
|
self.assertEqual(2, report["stats"]["rpc"]["count"])
|
|
self.assertEqual(1, len(report["children"]))
|
|
|
|
cbar = report["children"][0]
|
|
self._assert_child_dict(
|
|
cbar, base_id, base_id, "rpc",
|
|
"osprofiler.tests.functional.test_driver.Foo.bar")
|
|
|
|
self.assertEqual(1, len(cbar["children"]))
|
|
cbaz = cbar["children"][0]
|
|
self._assert_child_dict(
|
|
cbaz, base_id, cbar["trace_id"], "rpc",
|
|
"osprofiler.tests.functional.test_driver.Foo.baz")
|
|
|
|
|
|
class RedisDriverTestCase(DriverTestCase):
|
|
def setUp(self):
|
|
super(DriverTestCase, self).setUp()
|
|
CONF([])
|
|
opts.set_defaults(CONF,
|
|
connection_string="redis://localhost:6379",
|
|
enabled=True,
|
|
trace_sqlalchemy=False,
|
|
hmac_keys="SECRET_KEY")
|