X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=doctor_tests%2Finspector%2F__init__.py;h=31291baf7347e161f5976c8ee101308b8c041581;hb=9e36d918fef054b8d23692c0ec8c25d7b0640c07;hp=a9a86ecef883cd64f8da641b12ba064e16ac46df;hpb=f71afa73f5a56474036215b598e45123578c38d9;p=doctor.git

diff --git a/doctor_tests/inspector/__init__.py b/doctor_tests/inspector/__init__.py
index a9a86ece..31291baf 100644
--- a/doctor_tests/inspector/__init__.py
+++ b/doctor_tests/inspector/__init__.py
@@ -11,10 +11,12 @@ import os
 from oslo_config import cfg
 from oslo_utils import importutils
 
+from doctor_tests.common.constants import Inspector
+
 
 OPTS = [
     cfg.StrOpt('type',
-               default=os.environ.get('INSPECTOR_TYPE', 'sample'),
+               default=os.environ.get('INSPECTOR_TYPE', Inspector.SAMPLE),
                choices=['sample', 'congress', 'vitrage'],
                help='the component of doctor inspector',
                required=True),
@@ -34,8 +36,9 @@ OPTS = [
 
 
 _inspector_name_class_mapping = {
-    'sample': 'doctor_tests.inspector.sample.SampleInspector',
-    'congress': 'doctor_tests.inspector.congress.CongressInspector',
+    Inspector.SAMPLE: 'doctor_tests.inspector.sample.SampleInspector',
+    Inspector.CONGRESS: 'doctor_tests.inspector.congress.CongressInspector',
+    Inspector.VITRAGE: 'doctor_tests.inspector.vitrage.VitrageInspector',
 }