forked from autotest/autotest
/
label_unittest.py
executable file
·153 lines (125 loc) · 6.14 KB
/
label_unittest.py
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
#!/usr/bin/python
#
# Copyright 2008 Google Inc. All Rights Reserved.
"""Tests for label."""
import unittest, sys, os
try:
import autotest.common as common
except ImportError:
import common
from autotest.cli import cli_mock, topic_common
class label_list_unittest(cli_mock.cli_unittest):
values = [{u'id': 180, # Valid label
u'platform': False,
u'name': u'label0',
u'invalid': False,
u'kernel_config': u'',
u'only_if_needed': False},
{u'id': 338, # Valid label
u'platform': False,
u'name': u'label1',
u'invalid': False,
u'kernel_config': u'',
u'only_if_needed': False},
{u'id': 340, # Invalid label
u'platform': False,
u'name': u'label2',
u'invalid': True,
u'kernel_config': u'',
u'only_if_needed': False},
{u'id': 350, # Valid platform
u'platform': True,
u'name': u'plat0',
u'invalid': False,
u'kernel_config': u'',
u'only_if_needed': False},
{u'id': 420, # Invalid platform
u'platform': True,
u'name': u'plat1',
u'invalid': True,
u'kernel_config': u'',
u'only_if_needed': False}]
def test_label_list_labels_only(self):
self.run_cmd(argv=['atest', 'label', 'list', '--ignore_site_file'],
rpcs=[('get_labels', {}, True, self.values)],
out_words_ok=['label0', 'label1', 'label2'],
out_words_no=['plat0', 'plat1'])
def test_label_list_labels_only_valid(self):
self.run_cmd(argv=['atest', 'label', 'list', '-d',
'--ignore_site_file'],
rpcs=[('get_labels', {}, True, self.values)],
out_words_ok=['label0', 'label1'],
out_words_no=['label2', 'plat0', 'plat1'])
def test_label_list_labels_and_platforms(self):
self.run_cmd(argv=['atest', 'label', 'list', '--all',
'--ignore_site_file'],
rpcs=[('get_labels', {}, True, self.values)],
out_words_ok=['label0', 'label1', 'label2',
'plat0', 'plat1'])
def test_label_list_platforms_only(self):
self.run_cmd(argv=['atest', 'label', 'list', '-t',
'--ignore_site_file'],
rpcs=[('get_labels', {}, True, self.values)],
out_words_ok=['plat0', 'plat1'],
out_words_no=['label0', 'label1', 'label2'])
def test_label_list_platforms_only_valid(self):
self.run_cmd(argv=['atest', 'label', 'list',
'-t', '--valid-only', '--ignore_site_file'],
rpcs=[('get_labels', {}, True, self.values)],
out_words_ok=['plat0'],
out_words_no=['label0', 'label1', 'label2',
'plat1'])
class label_create_unittest(cli_mock.cli_unittest):
def test_execute_create_two_labels(self):
self.run_cmd(argv=['atest', 'label', 'create', 'label0', 'label1',
'--ignore_site_file'],
rpcs=[('add_label',
{'name': 'label0', 'platform': False,
'only_if_needed': False},
True, 42),
('add_label',
{'name': 'label1', 'platform': False,
'only_if_needed': False},
True, 43)],
out_words_ok=['Created', 'label0', 'label1'])
def test_execute_create_two_labels_bad(self):
self.run_cmd(argv=['atest', 'label', 'create', 'label0', 'label1',
'--ignore_site_file'],
rpcs=[('add_label',
{'name': 'label0', 'platform': False,
'only_if_needed': False},
True, 3),
('add_label',
{'name': 'label1', 'platform': False,
'only_if_needed': False},
False,
'''ValidationError: {'name':
'This value must be unique (label0)'}''')],
out_words_ok=['Created', 'label0'],
out_words_no=['label1'],
err_words_ok=['label1', 'ValidationError'])
class label_delete_unittest(cli_mock.cli_unittest):
def test_execute_delete_labels(self):
self.run_cmd(argv=['atest', 'label', 'delete', 'label0', 'label1',
'--ignore_site_file'],
rpcs=[('delete_label', {'id': 'label0'}, True, None),
('delete_label', {'id': 'label1'}, True, None)],
out_words_ok=['Deleted', 'label0', 'label1'])
class label_add_unittest(cli_mock.cli_unittest):
def test_execute_add_labels_to_hosts(self):
self.run_cmd(argv=['atest', 'label', 'add', 'label0',
'--machine', 'host0,host1', '--ignore_site_file'],
rpcs=[('label_add_hosts', {'id': 'label0',
'hosts': ['host1', 'host0']},
True, None)],
out_words_ok=['Added', 'label0', 'host0', 'host1'])
class label_remove_unittest(cli_mock.cli_unittest):
def test_execute_remove_labels_from_hosts(self):
self.run_cmd(argv=['atest', 'label', 'remove', 'label0',
'--machine', 'host0,host1', '--ignore_site_file'],
rpcs=[('label_remove_hosts', {'id': 'label0',
'hosts': ['host1', 'host0']},
True, None)],
out_words_ok=['Removed', 'label0', 'host0', 'host1'])
if __name__ == '__main__':
unittest.main()