| #!/usr/bin/env python3 |
| # Copyright 2021 The Chromium Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import unittest |
| |
| from models import TestPlatform |
| |
| |
| class ModelsTest(unittest.TestCase): |
| def test_platforms_from_chars(self): |
| self.assertCountEqual({TestPlatform.LINUX}, |
| TestPlatform.get_platforms_from_chars("L")) |
| self.assertCountEqual({TestPlatform.MAC}, |
| TestPlatform.get_platforms_from_chars("M")) |
| self.assertCountEqual({TestPlatform.WINDOWS}, |
| TestPlatform.get_platforms_from_chars("W")) |
| self.assertCountEqual({TestPlatform.CHROME_OS}, |
| TestPlatform.get_platforms_from_chars("C")) |
| self.assertCountEqual({TestPlatform.WINDOWS, TestPlatform.CHROME_OS}, |
| TestPlatform.get_platforms_from_chars("WC")) |
| |
| def test_platforms_from_browsertest_filename(self): |
| self.assertCountEqual( |
| {TestPlatform.LINUX}, |
| TestPlatform.get_platforms_from_browsertest_filename( |
| "browsertest_name_linux.cc")) |
| self.assertCountEqual( |
| {TestPlatform.MAC}, |
| TestPlatform.get_platforms_from_browsertest_filename( |
| "browsertest_name_mac.cc")) |
| self.assertCountEqual( |
| {TestPlatform.WINDOWS}, |
| TestPlatform.get_platforms_from_browsertest_filename( |
| "browsertest_name_win.cc")) |
| self.assertCountEqual( |
| {TestPlatform.CHROME_OS}, |
| TestPlatform.get_platforms_from_browsertest_filename( |
| "browsertest_name_cros.cc")) |
| self.assertCountEqual( |
| {TestPlatform.WINDOWS, TestPlatform.CHROME_OS}, |
| TestPlatform.get_platforms_from_browsertest_filename( |
| "browsertest_name_win_cros.cc")) |
| # No suffixes should apply to all platforms |
| self.assertCountEqual( |
| TestPlatform, |
| TestPlatform.get_platforms_from_browsertest_filename( |
| "browsertest_name.cc")) |
| |
| def test_platforms_to_fixture_suffix(self): |
| self.assertEqual( |
| "MacWinLinux", |
| TestPlatform.get_test_fixture_suffix( |
| {TestPlatform.MAC, TestPlatform.WINDOWS, TestPlatform.LINUX})) |
| self.assertEqual( |
| "Cros", |
| TestPlatform.get_test_fixture_suffix({TestPlatform.CHROME_OS})) |
| self.assertEqual( |
| "", |
| TestPlatform.get_test_fixture_suffix({ |
| TestPlatform.MAC, TestPlatform.WINDOWS, TestPlatform.LINUX, |
| TestPlatform.CHROME_OS |
| })) |
| self.assertEqual("", TestPlatform.get_test_fixture_suffix({})) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
