X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=tests%2Fconsumer%2F__init__.py;h=ccec8644825db3de209f9c49e2f7fcf7bf5836f3;hb=162c3ae179f69bd326325b135c3831ae12843e49;hp=68cc5dc9a0740847e719fc7dffbc4b1efcd8181f;hpb=e52dccf724ee05a16b6e78e79d7045f5c0ac979f;p=doctor.git

diff --git a/tests/consumer/__init__.py b/tests/consumer/__init__.py
index 68cc5dc9..ccec8644 100644
--- a/tests/consumer/__init__.py
+++ b/tests/consumer/__init__.py
@@ -7,6 +7,7 @@
 # http://www.apache.org/licenses/LICENSE-2.0
 ##############################################################################
 from oslo_config import cfg
+from oslo_utils import importutils
 
 
 OPTS = [
@@ -24,3 +25,13 @@ OPTS = [
                help='the port of doctor consumer',
                required=True),
 ]
+
+
+_consumer_name_class_mapping = {
+    'sample': 'consumer.sample.SampleConsumer'
+}
+
+
+def get_consumer(conf, log):
+    consumer_class = _consumer_name_class_mapping.get(conf.consumer.type)
+    return importutils.import_object(consumer_class, conf, log)
\ No newline at end of file