---
 cli/action_common_unittest.py |   14 +++++++++-----
 cli/topic_common_unittest.py  |    7 +++----
 2 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/cli/action_common_unittest.py b/cli/action_common_unittest.py
index 3f3e33c..0dbeb2a 100755
--- a/cli/action_common_unittest.py
+++ b/cli/action_common_unittest.py
@@ -111,9 +111,10 @@ class atest_list_unittest(cli_mock.cli_unittest):
                          mytest.execute(op='get_labels',
                                         filters=filters,
                                         check_results=check_results))
+        errors = mytest.failed
         (out, err) = self.god.unmock_io()
         self.god.check_playback()
-        return (out, err)
+        return (out, err, errors)
 
 
     def test_atest_list_execute_no_filters(self):
@@ -125,7 +126,7 @@ class atest_list_unittest(cli_mock.cli_unittest):
         check_results = {}
         filters['name__in'] = ['label0', 'label1']
         check_results['name__in'] = 'name'
-        (out, err) = self._atest_list_execute(filters, check_results)
+        (out, err, errors) = self._atest_list_execute(filters, check_results)
         self.assertEqual(err, '')
 
 
@@ -134,8 +135,11 @@ class atest_list_unittest(cli_mock.cli_unittest):
         check_results = {}
         filters['name__in'] = ['label0', 'label1', 'label2']
         check_results['name__in'] = 'name'
-        (out, err) = self._atest_list_execute(filters, check_results)
-        self.assertWords(err, ['Unknown', 'label2'])
+        (out, err, errors) = self._atest_list_execute(filters, check_results)
+        K = errors.keys()[0]
+        V = errors.values()[0].keys()[0]
+        self.assertIn('Unknown',K)
+        self.assertIn('label2',V)
 
 
     def test_atest_list_execute_items_good_and_bad_no_check(self):
@@ -143,7 +147,7 @@ class atest_list_unittest(cli_mock.cli_unittest):
         check_results = {}
         filters['name__in'] = ['label0', 'label1', 'label2']
         check_results['name__in'] = None
-        (out, err) = self._atest_list_execute(filters, check_results)
+        (out, err, errors) = self._atest_list_execute(filters, check_results)
         self.assertEqual(err, '')
 
 
diff --git a/cli/topic_common_unittest.py b/cli/topic_common_unittest.py
index 59bc180..3d2446d 100755
--- a/cli/topic_common_unittest.py
+++ b/cli/topic_common_unittest.py
@@ -343,10 +343,9 @@ class atest_unittest(cli_mock.cli_unittest):
 
 
     def test_invalid_arg_continue(self):
-        self.god.mock_io()
-        self.atest.invalid_arg('This is sort of ok')
-        (output, err) = self.god.unmock_io()
-        self.assert_(err.find('This is sort of ok') >= 0)
+        # The string will be placed into self.atest.failed
+        self.atest.invalid_arg('This is sort of ok')      
+        self.assert_('This is sort of ok' in self.atest.failed.keys())
 
 
     def test_failure_continue(self):
-- 
1.7.7.6

_______________________________________________
Autotest mailing list
Autotest@test.kernel.org
http://test.kernel.org/cgi-bin/mailman/listinfo/autotest

Reply via email to