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 9e7cd11..4278bb2 100644 (file)
--- a/doctor_tests/scenario/__init__.py
+++ b/doctor_tests/scenario/__init__.py
@@ -6,12 +6,20 @@
 # 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='all',
-               help='the name of test case',
+               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"
+                    " this repo. Default is 'fault_management'. Another test"
+                    " case can be specified only if a function named"
+                    " test_<test_case>() was implemented in DoctorTest.",
                required=False),
 ]