change ci from base-on-pod to base-on-scenario (in progress)

[yardstick.git] / tests / unit / cmd / commands / test_task.py

diff --git a/tests/unit/cmd/commands/test_task.py b/tests/unit/cmd/commands/test_task.py
index 5b404c4..ccb0f2a 100644 (file)
--- a/tests/unit/cmd/commands/test_task.py
+++ b/tests/unit/cmd/commands/test_task.py
@@ -62,11 +62,12 @@ class TaskCommandsTestCase(unittest.TestCase):
         cfg = \
             {'precondition':
                  {'installer_type': 'compass',
-                  'deploy_scenarios': 'os-nosdn'
+                  'deploy_scenarios': 'os-nosdn',
+                  'pod_name': 'huawei-pod1'
                  }
             }
 
         t = task.TaskParser('/opt')
-        mock_os.environ.get.side_effect = ['compass', 'os-nosdn']
+        mock_os.environ.get.side_effect = ['compass', 'os-nosdn', 'huawei-pod1']
         result = t._check_precondition(cfg)
         self.assertTrue(result)