omi/.github/scripts/test_check_brand_ui.py at main · ghzhost/omi · GitHub

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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
#!/usr/bin/env python3
"""Unit tests for check_brand_ui.py count_purple helper."""

from __future__ import annotations

import unittest

from check_brand_ui import count_purple, is_purple_hex, is_ui_source


class BrandUiTests(unittest.TestCase):
    def test_counts_color_purple_and_hex(self) -> None:
        text = "Color.purple\nlet x = Color(hex: 0x8B5CF6)\n#8B5CF6\npurplePrimary\n"
        # Color.purple + 0x8B5CF6 + #8B5CF6 + purplePrimary
        self.assertGreaterEqual(count_purple(text), 4)

    def test_counts_flutter_deep_purple(self) -> None:
        """`deepPurple` is the spelling most of the Flutter app actually uses.

        It went uncounted because camelCase leaves no word boundary before
        `purple`, so `\\bpurple\\b` cannot see it and no other pattern spelt it.
        """
        self.assertGreaterEqual(count_purple("color: Colors.deepPurple"), 1)
        self.assertGreaterEqual(count_purple("Colors.deepPurpleAccent"), 1)
        self.assertGreaterEqual(count_purple("Colors.deepPurple.shade300"), 1)

    def test_counts_dart_hex_literal(self) -> None:
        """Flutter writes `Color(0xFF8B5CF6)`, never `#8B5CF6`."""
        self.assertGreaterEqual(count_purple("const Color(0xFF8B5CF6)"), 1)
        self.assertGreaterEqual(count_purple("const Color(0xff7c3aed)"), 1)

    def test_ignores_colours_that_merely_contain_purple_hex_digits(self) -> None:
        """A non-purple hue must not be counted just for sharing digits."""
        self.assertEqual(count_purple("const Color(0xFF1A2B3C)"), 0)
        self.assertEqual(count_purple("const Color(0xFF00FF00)"), 0)

    def test_dart_hex_literal_must_start_a_token(self) -> None:
        """`0x` has to begin a token, or the ratchet fails files containing no purple.

        Without a leading boundary the pattern matches inside an identifier or a longer
        literal, so an unrelated constant ending in a purple hue is reported as a purple
        increase.
        """
        self.assertEqual(count_purple("foo0x8B5CF6"), 0)
        self.assertEqual(count_purple("SOME_CONST0x8B5CF6"), 0)
        # ...while a real literal in the shapes Flutter writes still counts.
        self.assertGreaterEqual(count_purple("Color(0xFF8B5CF6)"), 1)
        self.assertGreaterEqual(count_purple("value = 0x8B5CF6;"), 1)

    def test_is_ui_source(self) -> None:
        self.assertTrue(is_ui_source("desktop/macos/Desktop/Sources/Foo.swift"))
        self.assertFalse(is_ui_source("backend/main.py"))
        self.assertFalse(is_ui_source("desktop/macos/Desktop/Sources/Theme/OmiColors.swift"))

    def test_counts_swiftui_dot_purple(self) -> None:
        self.assertGreaterEqual(count_purple(".foregroundStyle(.purple)"), 1)

    def test_counts_flutter_colors_purple(self) -> None:
        self.assertGreaterEqual(count_purple("color: Colors.purple"), 1)

    def test_counts_tailwind_bg_purple_500(self) -> None:
        self.assertGreaterEqual(count_purple('className="bg-purple-500 text-purple-700"'), 2)

    def test_counts_css_color_purple(self) -> None:
        self.assertGreaterEqual(count_purple("color: purple;"), 1)

    def test_counts_tailwind_ramps_that_are_purple_by_sight_not_by_name(self) -> None:
        # These shipped past the check: the web marketplace had a violet promo
        # card, violet and indigo category themes, and violet capability chips.
        self.assertGreaterEqual(count_purple('className="bg-violet-500 text-indigo-300"'), 2)
        self.assertGreaterEqual(count_purple('className="from-fuchsia-600"'), 1)

    def test_hex_is_judged_by_hue_not_by_membership(self) -> None:
        # #6C2BD9 is one digit from the listed #6D28D9 and shipped the app-store
        # developer banner's purple gradient past a green check.
        for digits in ("6C2BD9", "2D1B69", "7C3AED", "A855F7", "C4B5FD", "D946EF"):
            self.assertTrue(is_purple_hex(digits), digits)

    def test_counts_css_swift_and_dart_indigo_literals(self) -> None:
        self.assertEqual(count_purple("#6366F1"), 1)
        self.assertEqual(count_purple("Color(hex: 0x6366F1)"), 1)
        self.assertEqual(count_purple("Color(0xFF6366F1)"), 1)

    def test_hue_test_leaves_blues_greys_and_pinks_alone(self) -> None:
        # A false positive here blocks unrelated PRs, so the negative cases
        # matter as much as the positive ones. Includes the app palette's own
        # backgrounds and the blues sitting nearest the purple boundary.
        for digits in ("3B82F6", "6C8EEF", "2563EB", "818CF6", "EC4899", "0F0F0F", "1F1F25", "35343B", "FFFFFF"):
            self.assertFalse(is_purple_hex(digits), digits)

    def test_near_black_is_neutral_whatever_hue_it_computes(self) -> None:
        # #0D0D0F is the content pane's fill and sits at a blue-purple hue with
        # almost no saturation; counting it would make the pane unfixable.
        self.assertFalse(is_purple_hex("0D0D0F"))


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