Qubes/core-admin
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

tests: fix get_label mockup to return meaningful values

Browse Source 
Reject invalid argument.
This commit is contained in:
Marek Marczykowski-Górecki 2016-05-21 03:46:17 +02:00 committed by Wojtek Porczyk
parent 98115eb541
commit bf0966bfcf
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
qubes/tests/vm/__init__.py
View File

@ -46,7 +46,12 @@ class TestApp(qubes.tests.TestEmitter):
def get_label(self, label): def get_label(self, label):
# pylint: disable=unused-argument # pylint: disable=unused-argument
return self.labels[1] if label in self.labels:
return self.labels[label]
for l in self.labels.values():
if l.name == label:
return l
raise KeyError(label)
def __init__(self): def __init__(self):
super(TestApp, self).__init__() super(TestApp, self).__init__()