mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 1 | #!/usr/bin/python |
| 2 | # |
| 3 | # Copyright 2008 Google Inc. All Rights Reserved. |
| 4 | |
| 5 | """Tests for test.""" |
| 6 | |
Allen Li | b4572db | 2017-02-01 14:46:38 -0800 | [diff] [blame] | 7 | import unittest |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 8 | |
| 9 | import common |
Allen Li | b4572db | 2017-02-01 14:46:38 -0800 | [diff] [blame] | 10 | from autotest_lib.cli import cli_mock |
Aviv Keshet | 3dd8beb | 2013-05-13 17:36:04 -0700 | [diff] [blame] | 11 | from autotest_lib.client.common_lib import control_data |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 12 | |
Aviv Keshet | 3dd8beb | 2013-05-13 17:36:04 -0700 | [diff] [blame] | 13 | CLIENT = control_data.CONTROL_TYPE_NAMES.CLIENT |
| 14 | SERVER = control_data.CONTROL_TYPE_NAMES.SERVER |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 15 | |
| 16 | class test_list_unittest(cli_mock.cli_unittest): |
| 17 | values = [{u'description': u'unknown', |
Aviv Keshet | 3dd8beb | 2013-05-13 17:36:04 -0700 | [diff] [blame] | 18 | u'test_type': CLIENT, |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 19 | u'test_class': u'Canned Test Sets', |
| 20 | u'path': u'client/tests/test0/control', |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 21 | u'id': 138, |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 22 | u'name': u'test0', |
| 23 | u'experimental': False}, |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 24 | {u'description': u'unknown', |
Aviv Keshet | 3dd8beb | 2013-05-13 17:36:04 -0700 | [diff] [blame] | 25 | u'test_type': SERVER, |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 26 | u'test_class': u'Kernel', |
| 27 | u'path': u'server/tests/test1/control', |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 28 | u'id': 139, |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 29 | u'name': u'test1', |
| 30 | u'experimental': False}, |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 31 | {u'description': u'unknown', |
Aviv Keshet | 3dd8beb | 2013-05-13 17:36:04 -0700 | [diff] [blame] | 32 | u'test_type': CLIENT, |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 33 | u'test_class': u'Canned Test Sets', |
| 34 | u'path': u'client/tests/test2/control.readprofile', |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 35 | u'id': 140, |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 36 | u'name': u'test2', |
| 37 | u'experimental': False}, |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 38 | {u'description': u'unknown', |
Aviv Keshet | 3dd8beb | 2013-05-13 17:36:04 -0700 | [diff] [blame] | 39 | u'test_type': SERVER, |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 40 | u'test_class': u'Canned Test Sets', |
| 41 | u'path': u'server/tests/test3/control', |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 42 | u'id': 142, |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 43 | u'name': u'test3', |
| 44 | u'experimental': False}, |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 45 | {u'description': u'Random stuff to check that things are ok', |
Aviv Keshet | 3dd8beb | 2013-05-13 17:36:04 -0700 | [diff] [blame] | 46 | u'test_type': CLIENT, |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 47 | u'test_class': u'Hardware', |
| 48 | u'path': u'client/tests/test4/control.export', |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 49 | u'id': 143, |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 50 | u'name': u'test4', |
| 51 | u'experimental': True}] |
| 52 | |
| 53 | |
| 54 | def test_test_list_tests_default(self): |
| 55 | self.run_cmd(argv=['atest', 'test', 'list'], |
| 56 | rpcs=[('get_tests', {'experimental': False}, |
| 57 | True, self.values)], |
| 58 | out_words_ok=['test0', 'test1', 'test2', |
| 59 | 'test3', 'test4'], |
| 60 | out_words_no=['Random', 'control.export']) |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 61 | |
| 62 | |
| 63 | def test_test_list_tests_all(self): |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 64 | self.run_cmd(argv=['atest', 'test', 'list', '--all'], |
| 65 | rpcs=[('get_tests', {}, |
| 66 | True, self.values)], |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 67 | out_words_ok=['test0', 'test1', 'test2', |
| 68 | 'test3', 'test4'], |
| 69 | out_words_no=['Random', 'control.export']) |
| 70 | |
| 71 | |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 72 | def test_test_list_tests_exp(self): |
| 73 | self.run_cmd(argv=['atest', 'test', 'list', '--experimental'], |
| 74 | rpcs=[('get_tests', {'experimental': True}, |
| 75 | True, |
| 76 | [{u'description': u'Random stuff', |
Aviv Keshet | 3dd8beb | 2013-05-13 17:36:04 -0700 | [diff] [blame] | 77 | u'test_type': CLIENT, |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 78 | u'test_class': u'Hardware', |
| 79 | u'path': u'client/tests/test4/control.export', |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 80 | u'id': 143, |
| 81 | u'name': u'test4', |
| 82 | u'experimental': True}])], |
| 83 | out_words_ok=['test4'], |
| 84 | out_words_no=['Random', 'control.export']) |
| 85 | |
| 86 | |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 87 | def test_test_list_tests_select_one(self): |
| 88 | filtered = [val for val in self.values if val['name'] in ['test3']] |
| 89 | self.run_cmd(argv=['atest', 'test', 'list', 'test3'], |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 90 | rpcs=[('get_tests', {'name__in': ['test3'], |
| 91 | 'experimental': False}, |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 92 | True, filtered)], |
| 93 | out_words_ok=['test3'], |
| 94 | out_words_no=['test0', 'test1', 'test2', 'test4', |
| 95 | 'unknown']) |
| 96 | |
| 97 | |
| 98 | def test_test_list_tests_select_two(self): |
| 99 | filtered = [val for val in self.values |
| 100 | if val['name'] in ['test3', 'test1']] |
| 101 | self.run_cmd(argv=['atest', 'test', 'list', 'test3,test1'], |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 102 | rpcs=[('get_tests', {'name__in': ['test1', 'test3'], |
| 103 | 'experimental': False}, |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 104 | True, filtered)], |
Aviv Keshet | 3dd8beb | 2013-05-13 17:36:04 -0700 | [diff] [blame] | 105 | out_words_ok=['test3', 'test1', SERVER], |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 106 | out_words_no=['test0', 'test2', 'test4', |
Aviv Keshet | 3dd8beb | 2013-05-13 17:36:04 -0700 | [diff] [blame] | 107 | 'unknown', CLIENT]) |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 108 | |
| 109 | |
| 110 | def test_test_list_tests_select_two_space(self): |
| 111 | filtered = [val for val in self.values |
| 112 | if val['name'] in ['test3', 'test1']] |
| 113 | self.run_cmd(argv=['atest', 'test', 'list', 'test3', 'test1'], |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 114 | rpcs=[('get_tests', {'name__in': ['test1', 'test3'], |
| 115 | 'experimental': False}, |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 116 | True, filtered)], |
Aviv Keshet | 3dd8beb | 2013-05-13 17:36:04 -0700 | [diff] [blame] | 117 | out_words_ok=['test3', 'test1', SERVER], |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 118 | out_words_no=['test0', 'test2', 'test4', |
Aviv Keshet | 3dd8beb | 2013-05-13 17:36:04 -0700 | [diff] [blame] | 119 | 'unknown', CLIENT]) |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 120 | |
| 121 | |
| 122 | def test_test_list_tests_all_verbose(self): |
| 123 | self.run_cmd(argv=['atest', 'test', 'list', '-v'], |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 124 | rpcs=[('get_tests', {'experimental': False}, |
| 125 | True, self.values)], |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 126 | out_words_ok=['test0', 'test1', 'test2', |
| 127 | 'test3', 'test4', 'client/tests', |
| 128 | 'server/tests'], |
| 129 | out_words_no=['Random']) |
| 130 | |
| 131 | |
| 132 | def test_test_list_tests_all_desc(self): |
| 133 | self.run_cmd(argv=['atest', 'test', 'list', '-d'], |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 134 | rpcs=[('get_tests', {'experimental': False}, |
| 135 | True, self.values)], |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 136 | out_words_ok=['test0', 'test1', 'test2', |
| 137 | 'test3', 'test4', 'unknown', 'Random'], |
| 138 | out_words_no=['client/tests', 'server/tests']) |
| 139 | |
| 140 | |
| 141 | def test_test_list_tests_all_desc_verbose(self): |
| 142 | self.run_cmd(argv=['atest', 'test', 'list', '-d', '-v'], |
mbligh | 140a23c | 2008-10-29 16:55:21 +0000 | [diff] [blame] | 143 | rpcs=[('get_tests', {'experimental': False}, |
| 144 | True, self.values)], |
mbligh | be630eb | 2008-08-01 16:41:48 +0000 | [diff] [blame] | 145 | out_words_ok=['test0', 'test1', 'test2', |
| 146 | 'test3', 'test4', 'client/tests', |
| 147 | 'server/tests', 'unknown', 'Random' ]) |
| 148 | |
| 149 | |
| 150 | if __name__ == '__main__': |
| 151 | unittest.main() |