android/test_health/test_health_exporter_unittest.py

#!/usr/bin/env python3
# Copyright 2022 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Tests for test_health_exporter."""

import datetime as dt
import json
import pathlib
import tempfile
import unittest

from java_test_utils import JavaTestHealth
import test_health_exporter
from test_health_extractor import GitRepoInfo, TestHealthInfo

_TEST_TYPE = 'JAVA'
_JAVA_PACKAGE = 'org.chromium.foo_pkg'
_GIT_HEAD_HASH = 'fcd260583cb6d2c739b91f53363fcf0ad3eb3216'
_GIT_HEAD_TIME = '2022-01-18T22:45:48.000000+00:00'
_GIT_REPO_INFO = GitRepoInfo(
    git_head=_GIT_HEAD_HASH,
    git_head_time=dt.datetime.fromisoformat(_GIT_HEAD_TIME))

_JAVA_TEST_NAME = 'FooTest'
_JAVA_TEST_FILENAME = _JAVA_TEST_NAME + '.java'
_JAVA_TEST_DIR = 'javatests/org/chromium/foo_pkg'

_JAVA_TEST_HEALTH = JavaTestHealth(java_package=_JAVA_PACKAGE,
                                   disabled_tests_count=1,
                                   disable_if_tests_count=2,
                                   tests_count=10)
_TEST_HEALTH_INFO = TestHealthInfo(_JAVA_TEST_NAME,
                                   test_dir=pathlib.Path(_JAVA_TEST_DIR),
                                   test_filename=_JAVA_TEST_FILENAME,
                                   java_test_health=_JAVA_TEST_HEALTH,
                                   git_repo_info=_GIT_REPO_INFO)

_TEST_HEALTH_JSON_DICT = dict(test_name=_JAVA_TEST_NAME,
                              test_path=_JAVA_TEST_DIR,
                              test_filename=_JAVA_TEST_FILENAME,
                              test_type=_TEST_TYPE,
                              java_package=_JAVA_PACKAGE,
                              disabled_tests_count=1,
                              disable_if_tests_count=2,
                              tests_count=10,
                              git_head_hash=_GIT_HEAD_HASH,
                              git_head_timestamp=_GIT_HEAD_TIME)


class ToJsonFile(unittest.TestCase):
    """Tests for the to_json_file function."""

    def test_to_json_file_single_java_test(self):
        with tempfile.TemporaryDirectory() as tmpdir:
            json_file = pathlib.Path(tmpdir) / 'test_output'

            test_health_exporter.to_json_file([_TEST_HEALTH_INFO], json_file)

            with open(json_file) as output_file:
                json_lines = output_file.readlines()

        self.assertEqual(1, len(json_lines))

        result = json.loads(json_lines[0])

        self.assertDictEqual(_TEST_HEALTH_JSON_DICT, result)

    def test_to_json_file_multiple_java_tests(self):
        with tempfile.TemporaryDirectory() as tmpdir:
            json_file = pathlib.Path(tmpdir) / 'test_output'

            test_health_exporter.to_json_file(
                [_TEST_HEALTH_INFO, _TEST_HEALTH_INFO, _TEST_HEALTH_INFO],
                json_file)

            with open(json_file) as output_file:
                json_lines = output_file.readlines()

        self.assertEqual(3, len(json_lines))

        result1 = json.loads(json_lines[0])
        result2 = json.loads(json_lines[1])
        result3 = json.loads(json_lines[2])

        self.assertDictEqual(_TEST_HEALTH_JSON_DICT, result1)
        self.assertDictEqual(_TEST_HEALTH_JSON_DICT, result2)
        self.assertDictEqual(_TEST_HEALTH_JSON_DICT, result3)

    def test_to_json_file_java_package_omitted(self):
        java_test_health = JavaTestHealth(java_package=None,
                                          disabled_tests_count=1,
                                          disable_if_tests_count=2,
                                          tests_count=10)
        test_health_info = TestHealthInfo(
            _JAVA_TEST_NAME,
            test_dir=pathlib.Path(_JAVA_TEST_DIR),
            test_filename=_JAVA_TEST_FILENAME,
            java_test_health=java_test_health,
            git_repo_info=_GIT_REPO_INFO)
        test_health_json_dict = _TEST_HEALTH_JSON_DICT.copy()
        del test_health_json_dict['java_package']

        with tempfile.TemporaryDirectory() as tmpdir:
            json_file = pathlib.Path(tmpdir) / 'test_output'

            test_health_exporter.to_json_file([test_health_info], json_file)

            with open(json_file) as output_file:
                json_lines = output_file.readlines()

        self.assertEqual(1, len(json_lines))

        result = json.loads(json_lines[0])

        self.assertDictEqual(test_health_json_dict, result)


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