Source code
Revision control
Copy as Markdown
Other Tools
#!/usr/bin/env vpython3
# Copyright (c) 2022 The WebRTC project authors. All Rights Reserved.
#
# Use of this source code is governed by a BSD-style license
# that can be found in the LICENSE file in the root of the source
# tree. An additional intellectual property rights grant can be found
# in the file PATENTS. All contributing project authors may
# be found in the AUTHORS file in the root of the source tree.
import os
import sys
import unittest
from unittest.mock import MagicMock
# This tests requires the webrtc_dashboard_upload target to be built before
# running the tests.
def _ConfigurePythonPath():
# We just yank the python scripts we require into the PYTHONPATH. You could
# also imagine a solution where we use for instance
# protobuf:py_proto_runtime to copy catapult and protobuf code to out/.
# This is the convention in Chromium and WebRTC python scripts. We do need
# to build histogram_pb2 however, so that's why we add out/ to sys.path
# below.
#
# It would be better if there was an equivalent to py_binary in GN, but
# there's not.
script_dir = os.path.dirname(os.path.realpath(__file__))
checkout_root = os.path.abspath(os.path.join(script_dir, os.pardir,
os.pardir))
sys.path.insert(
0, os.path.join(checkout_root, 'third_party', 'catapult', 'tracing'))
sys.path.insert(
0, os.path.join(checkout_root, 'third_party', 'protobuf', 'python'))
# The webrtc_dashboard_upload gn rule will build the protobuf stub for
# python, so put it in the path for this script before we attempt to import
# it.
histogram_proto_path = os.path.join(os.path.join('../../out/Default'),
'pyproto', 'tracing', 'tracing', 'proto')
sys.path.insert(0, histogram_proto_path)
# Fail early in case the proto hasn't been built.
from tracing.proto import histogram_proto
if not histogram_proto.HAS_PROTO:
raise ImportError('Could not find histogram_pb2. You need to build the '
'webrtc_dashboard_upload target before invoking this '
'script. Expected to find '
'histogram_pb2.py in %s.' % histogram_proto_path)
def _CreateHistogram(name='hist',
master=None,
bot=None,
benchmark=None,
benchmark_description=None,
commit_position=None,
samples=None):
hists = [catapult_uploader.histogram.Histogram(name, 'count')]
if samples:
for s in samples:
hists[0].AddSample(s)
histograms = catapult_uploader.histogram_set.HistogramSet(hists)
if master:
histograms.AddSharedDiagnosticToAllHistograms(
catapult_uploader.reserved_infos.MASTERS.name,
catapult_uploader.generic_set.GenericSet([master]))
if bot:
histograms.AddSharedDiagnosticToAllHistograms(
catapult_uploader.reserved_infos.BOTS.name,
catapult_uploader.generic_set.GenericSet([bot]))
if commit_position:
histograms.AddSharedDiagnosticToAllHistograms(
catapult_uploader.reserved_infos.CHROMIUM_COMMIT_POSITIONS.name,
catapult_uploader.generic_set.GenericSet([commit_position]))
if benchmark:
histograms.AddSharedDiagnosticToAllHistograms(
catapult_uploader.reserved_infos.BENCHMARKS.name,
catapult_uploader.generic_set.GenericSet([benchmark]))
if benchmark_description:
histograms.AddSharedDiagnosticToAllHistograms(
catapult_uploader.reserved_infos.BENCHMARK_DESCRIPTIONS.name,
catapult_uploader.generic_set.GenericSet([benchmark_description]))
return histograms
class CatapultUploaderTest(unittest.TestCase):
def setUp(self):
mock = MagicMock(return_value=[200, None])
catapult_uploader.httplib2.Http.request = mock
self.histogram = _CreateHistogram(
master='master',
bot='bot',
benchmark='benchmark',
commit_position=123,
benchmark_description='Benchmark description.',
samples=[1, 2, 3])
def testSendHistogramsSet(self):
# pylint: disable=protected-access
response, content = catapult_uploader._SendHistogramSet(url, self.histogram)
self.assertEqual(response, 200)
self.assertEqual(content, None)
def testSendHistogramsSetLocalhost(self):
# pylint: disable=protected-access
response, content = catapult_uploader._SendHistogramSet(url, self.histogram)
self.assertEqual(response, 200)
self.assertEqual(content, None)
if (__name__) == '__main__':
_ConfigurePythonPath()
import catapult_uploader
unittest.main()