Source code
Revision control
Copy as Markdown
Other Tools
#!/usr/bin/env vpython3
# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""File for testing measures.py."""
import os
import tempfile
import unittest
import google.protobuf.json_format as json_format
from google.protobuf import any_pb2
import measures
from measure import Measure
from test_script_metrics_pb2 import TestScriptMetric, TestScriptMetrics
class MeasuresTest(unittest.TestCase):
"""Test measure.py."""
def test_create_average(self) -> None:
ave = measures.average('a', 'b', 'c')
ave.record(1)
ave.record(2)
exp = TestScriptMetric()
exp.name = 'a/b/c'
exp.value = 1.5
self.assertEqual(ave.dump(), exp)
def test_create_single_name_piece(self) -> None:
self.assertEqual(measures.average('a')._name, 'a')
def test_create_no_name_piece(self) -> None:
self.assertRaises(ValueError, lambda: measures.average())
def test_create_none_in_name_pieces(self) -> None:
self.assertRaises(TypeError, lambda: measures.average('a', None))
def test_create_count(self) -> None:
self.assertIsInstance(measures.count('a'), Measure)
def test_create_data_points(self) -> None:
self.assertIsInstance(measures.data_points('a'), Measure)
def test_register(self) -> None:
before = len(measures._metric._metrics)
for x in range(3):
measures.average('a')
self.assertEqual(len(measures._metric._metrics), before + 3)
def test_dump(self) -> None:
before = len(measures._metric._metrics)
count = measures.count('test', 'dump')
for _ in range(101):
count.record()
with tempfile.TemporaryDirectory() as tmpdir:
measures.dump(tmpdir)
with open(os.path.join(tmpdir,
measures.TEST_SCRIPT_METRICS_JSONPB_FILENAME),
'r',
encoding='utf-8') as rf:
any_msg = json_format.Parse(rf.read(), any_pb2.Any())
message = TestScriptMetrics()
self.assertTrue(any_msg.Unpack(message))
self.assertEqual(len(message.metrics), before + 1)
exp = TestScriptMetric()
exp.name = 'test/dump'
exp.value = 101
self.assertEqual(message.metrics[-1], exp)
def test_dump_with_type(self) -> None:
with tempfile.TemporaryDirectory() as tmpdir:
measures.dump(tmpdir)
with open(os.path.join(tmpdir,
measures.TEST_SCRIPT_METRICS_JSONPB_FILENAME),
'r',
encoding='utf-8') as rf:
txt = rf.read()
self.assertTrue('"@type":' in txt)
self.assertTrue(TestScriptMetrics().DESCRIPTOR.full_name in txt)
def test_dump_create_dir(self) -> None:
with tempfile.TemporaryDirectory() as tmpdir:
dir_path = os.path.join(tmpdir, 'hello', 'this', 'dir', 'should', 'not',
'exist')
measures.dump(dir_path)
self.assertTrue(os.path.isdir(dir_path))
if __name__ == '__main__':
unittest.main()