depot_tools/tests/my_activity_test.py

#!/usr/bin/env vpython3
# Copyright 2020 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os
import sys
import unittest

from datetime import datetime

DEPOT_TOOLS_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, DEPOT_TOOLS_ROOT)

import my_activity


class MyActivityTest(unittest.TestCase):
    def test_datetime_to_midnight(self):
        self.assertEqual(
            datetime(2020, 9, 12),
            my_activity.datetime_to_midnight(datetime(2020, 9, 12, 13, 0, 0)))
        self.assertEqual(
            datetime(2020, 12, 31),
            my_activity.datetime_to_midnight(datetime(2020, 12, 31, 23, 59,
                                                      59)))
        self.assertEqual(
            datetime(2020, 12, 31),
            my_activity.datetime_to_midnight(datetime(2020, 12, 31)))

    def test_get_quarter_of(self):
        self.assertEqual((datetime(2020, 7, 1), datetime(2020, 10, 1)),
                         my_activity.get_quarter_of(datetime(2020, 9, 12)))
        # Quarter range includes beginning
        self.assertEqual((datetime(2020, 10, 1), datetime(2021, 1, 1)),
                         my_activity.get_quarter_of(datetime(2020, 10, 1)))
        # Quarter range excludes end
        self.assertEqual((datetime(2021, 1, 1), datetime(2021, 4, 1)),
                         my_activity.get_quarter_of(datetime(2021, 1, 1)))
        self.assertEqual(
            (datetime(2020, 10, 1), datetime(2021, 1, 1)),
            my_activity.get_quarter_of(datetime(2020, 12, 31, 23, 59, 59)))

    def test_get_year_of(self):
        self.assertEqual((datetime(2020, 1, 1), datetime(2021, 1, 1)),
                         my_activity.get_year_of(datetime(2020, 9, 12)))
        # Year range includes beginning
        self.assertEqual((datetime(2020, 1, 1), datetime(2021, 1, 1)),
                         my_activity.get_year_of(datetime(2020, 1, 1)))
        # Year range excludes end
        self.assertEqual((datetime(2021, 1, 1), datetime(2022, 1, 1)),
                         my_activity.get_year_of(datetime(2021, 1, 1)))

    def test_get_week_of(self):
        self.assertEqual((datetime(2020, 9, 7), datetime(2020, 9, 14)),
                         my_activity.get_week_of(datetime(2020, 9, 12)))
        # Week range includes beginning
        self.assertEqual((datetime(2020, 9, 7), datetime(2020, 9, 14)),
                         my_activity.get_week_of(datetime(2020, 9, 7)))
        # Week range excludes beginning
        self.assertEqual((datetime(2020, 9, 14), datetime(2020, 9, 21)),
                         my_activity.get_week_of(datetime(2020, 9, 14)))

    def _get_issue_with_description(self, description):
        return {
            'current_revision': 'rev',
            'revisions': {
                'rev': {
                    'commit': {
                        'message': description
                    }
                }
            },
        }

    def test_extract_bug_numbers_from_description(self):
        issue = self._get_issue_with_description(
            'Title\n'
            '\n'
            'Description\n'
            'A comment:\n'
            '> Bug: 1234, another:5678\n'
            '\n'
            'Bug: another:1234, 5678\n'
            'BUG=project:13141516\n'
            'Fixed: fixed:9101112\n'
            'Change-Id: Iabcdef1234567890\n')
        self.assertEqual([
            'another:1234', 'chromium:5678', 'fixed:9101112', 'project:13141516'
        ], my_activity.extract_bug_numbers_from_description(issue))


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