get 'test_case' option form environment variable

[doctor.git] / doctor_tests / scenario / __init__.py

diff --git a/doctor_tests/scenario/__init__.py b/doctor_tests/scenario/__init__.py
index b9dfa30..4278bb2 100644 (file)
--- a/doctor_tests/scenario/__init__.py
+++ b/doctor_tests/scenario/__init__.py
@@ -6,12 +6,15 @@
 # which accompanies this distribution, and is available at
 # http://www.apache.org/licenses/LICENSE-2.0
 ##############################################################################
+import os
+
 from oslo_config import cfg
 
 
 OPTS = [
     cfg.StrOpt('test_case',
-               default='fault_management',
+               default=os.environ.get('TEST_CASE', 'fault_management'),
+               choices=['all', 'fault_management', 'maintenance'],
                help="A name of test case to be executed,"
                     " choices are 'all', 'fault_management' or 'maintenance'."
                     " Set 'all' to execute all the test cases existing in"