Add more tests + introduce the option to pass the DB to crm main

Author: judyjoseph

crm/main.py

@@ -7,31 +7,35 @@
 from sonic_py_common import multi_asic
 
 class Crm:
-    def __init__(self):
+    def __init__(self, db=None):
         self.cli_mode = None
         self.addr_family = None
         self.res_type = None
         self.db = None
-        self.config_db = None
+        self.cfgdb = db
         self.multi_asic = multi_asic_util.MultiAsic()
 
     @multi_asic_util.run_on_multi_asic
     def config(self, attr, val):
         """
         CRM handler for 'config' CLI commands.
         """
+        if self.cfgdb:
+            self.config_db = self.cfgdb
         self.config_db.mod_entry("CRM", 'Config', {attr: val})
 
     def show_summary(self):
         """
         CRM Handler to display general information.
         """
 
-        # Get the namespace list
-        namespaces = multi_asic.get_namespace_list()
+        configdb = self.cfgdb
+        if configdb is None:
+            # Get the namespace list
+            namespaces = multi_asic.get_namespace_list()
 
-        configdb = swsssdk.ConfigDBConnector(namespace=namespaces[0])
-        configdb.connect()
+            configdb = swsssdk.ConfigDBConnector(namespace=namespaces[0])
+            configdb.connect()
 
         crm_info = configdb.get_entry('CRM', 'Config')
 
@@ -45,11 +49,13 @@ def show_thresholds(self, resource):
         CRM Handler to display thresholds information.
         """
 
-        # Get the namespace list
-        namespaces = multi_asic.get_namespace_list()
+        configdb = self.cfgdb
+        if configdb is None:
+            # Get the namespace list
+            namespaces = multi_asic.get_namespace_list()
 
-        configdb = swsssdk.ConfigDBConnector(namespace=namespaces[0])
-        configdb.connect()
+            configdb = swsssdk.ConfigDBConnector(namespace=namespaces[0])
+            configdb.connect()
 
         crm_info = configdb.get_entry('CRM', 'Config')
 
@@ -190,8 +196,11 @@ def cli(ctx):
     """
     Utility entry point.
     """
+    # Use the db object if given as input.
+    db = None if ctx.obj is None else ctx.obj.cfgdb
+
     context = {
-        "crm": Crm()
+        "crm": Crm(db)
     }
 
     ctx.obj = context