File: process_perf_results_test.py

package info (click to toggle)
thunderbird 1%3A143.0.1-1
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 4,703,968 kB
  • sloc: cpp: 7,770,492; javascript: 5,943,842; ansic: 3,918,754; python: 1,418,263; xml: 653,354; asm: 474,045; java: 183,079; sh: 111,238; makefile: 20,410; perl: 14,359; objc: 13,059; yacc: 4,583; pascal: 3,405; lex: 1,720; ruby: 999; exp: 762; sql: 715; awk: 580; php: 436; lisp: 430; sed: 69; csh: 10
file content (70 lines) | stat: -rw-r--r-- 2,624 bytes parent folder | download | duplicates (29)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#!/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 import mock

_SCRIPT_DIR = os.path.dirname(__file__)
_SRC_DIR = os.path.normpath(os.path.join(_SCRIPT_DIR, '..', '..'))

sys.path.insert(0, os.path.join(_SRC_DIR, 'third_party', 'protobuf', 'python'))
import process_perf_results


class ProcessPerfResultsTest(unittest.TestCase):
  def testConfigurePythonPath(self):
    # pylint: disable=protected-access
    self.assertEqual(
        0,
        process_perf_results._ConfigurePythonPath(
            os.path.join(_SRC_DIR, 'out/Default')))

  def testUploadToDasboard(self):
    outdir = os.path.join(_SRC_DIR, 'out/Default')
    args = mock.Mock(
        build_properties='{' + '"outdir":"' + outdir + '", ' +
        '"perf_dashboard_machine_group":"mock_machine_group", ' +
        '"bot":"mock_bot", ' + '"webrtc_git_hash":"mock_webrtc_git_hash", ' +
        '"commit_position":"123456", ' +
        '"build_page_url":"mock_build_page_url", ' +
        '"dashboard_url":"mock_dashboard_url"' + '}',
        summary_json='mock_sumary_json',
        task_output_dir='mock_task_output_dir',
        test_suite='mock_test_suite',
    )
    perftest_output = mock.Mock(
        absolute=lambda: 'dummy_path/perftest-output.pb',
        is_file=lambda: True,
    )
    with mock.patch('pathlib.Path.rglob') as mocked_rglob:
      with mock.patch('catapult_uploader.UploadToDashboard') as mocked_upload:
        mocked_rglob.return_value = [perftest_output]
        mocked_upload.return_value = 0
        # pylint: disable=protected-access
        self.assertEqual(0, process_perf_results._UploadToDasboard(args))

        import catapult_uploader
        mocked_upload.assert_called_once_with(
            catapult_uploader.UploaderOptions(
                perf_dashboard_machine_group='mock_machine_group',
                bot='mock_bot',
                test_suite='mock_test_suite',
                webrtc_git_hash='mock_webrtc_git_hash',
                commit_position='123456',
                build_page_url='mock_build_page_url',
                dashboard_url='mock_dashboard_url',
                input_results_file=perftest_output.absolute()))


if (__name__) == '__main__':
  unittest.main()