commit 34997d599dc496b5362ebe029c446fc1f87c0142
author nelsonli <nelsonli@google.com> Fri Aug 17 09:43:28 2018 +0800
committer mikehoran <mikehoran@google.com> Wed Aug 22 10:22:22 2018 -0700
tree 36d1998a9981a4e7ba7c7f8bc1d8c222b55aecd0
parent 92496773f5bb69a4bfbea9903bd5c4770bc3a618

Atest: output revamp - colorize result reporting

Bug: 112155105

Test: ATEST_PRETTY_RESULT=foo atest hello_world_test
      ATEST_PRETTY_RESULT=foo atest com.android.server.wm.ScreenDecorWindowTests
      ATEST_PRETTY_RESULT=foo atest FrameworksUiServicesTests:NotificationStatsTest
      ATEST_PRETTY_RESULT=foo atest hello_world_test com.android.server.wm.ScreenDecorWindowTests FrameworksUiServicesTests:NotificationStatsTest

Change-Id: Ib95a088d13c44803eaa1d6f30558349714a72edb

atest/cli_translator_unittest.py

diff --git a/atest/cli_translator_unittest.py b/atest/cli_translator_unittest.py
index 893afe6..899c57c 100755
--- a/atest/cli_translator_unittest.py
+++ b/atest/cli_translator_unittest.py
@@ -91,20 +91,21 @@
         # Let's make sure we return what we expect.
         expected_test_infos = {uc.MODULE_INFO}
         mock_getfindmethods.return_value = [
-            test_finder_base.Finder(None, find_method_return_module_info)]
+            test_finder_base.Finder(None, find_method_return_module_info, None)]
         unittest_utils.assert_strict_equal(
             self, ctr._get_test_infos(one_test), expected_test_infos)
 
         # Check we receive multiple test infos.
         expected_test_infos = {uc.MODULE_INFO, uc.CLASS_INFO}
         mock_getfindmethods.return_value = [
-            test_finder_base.Finder(None, find_method_return_module_class_info)]
+            test_finder_base.Finder(None, find_method_return_module_class_info,
+                                    None)]
         unittest_utils.assert_strict_equal(
             self, ctr._get_test_infos(mult_test), expected_test_infos)
 
         # Let's make sure we raise an error when we have no tests found.
         mock_getfindmethods.return_value = [
-            test_finder_base.Finder(None, find_method_return_nothing)]
+            test_finder_base.Finder(None, find_method_return_nothing, None)]
         self.assertRaises(atest_error.NoTestFoundError, ctr._get_test_infos,
                           one_test)
 
@@ -113,7 +114,8 @@
         test_detail2 = test_mapping.TestDetail(uc.TEST_MAPPING_TEST_WITH_OPTION)
         expected_test_infos = {uc.MODULE_INFO, uc.CLASS_INFO}
         mock_getfindmethods.return_value = [
-            test_finder_base.Finder(None, find_method_return_module_class_info)]
+            test_finder_base.Finder(None, find_method_return_module_class_info,
+                                    None)]
         test_infos = ctr._get_test_infos(
             mult_test, [test_detail1, test_detail2])
         unittest_utils.assert_strict_equal(