Fixing test failures

Author: rameshraghupathy

tests/chassis_modules_test.py

@@ -123,8 +123,10 @@ def is_transition_timed_out(db, chassis_module_name):
 sys.modules['sonic_py_common.syslogger'] = mock.Mock()
 
 # Create a mock SysLogger class
+
+
 class MockSysLogger:
-    def __init__(self, identifier):
+    def __init__(self, identifier=None):
         self.identifier = identifier
 
     def log_info(self, message):
@@ -139,6 +141,7 @@ def log_warning(self, message):
     def log_notice(self, message):
         pass
 
+
 sys.modules['sonic_py_common.syslogger'].SysLogger = MockSysLogger
 
 show_linecard0_shutdown_output="""\
@@ -293,6 +296,7 @@ def _stub_state_conn(row=None):
         row = {}
     return SimpleNamespace(STATE_DB=6, get_all=lambda _db, _key: row)
 
+
 # Mock StateDBHelper class used in chassis_modules.py
 class MockStateDBHelper:
     def __init__(self, sonic_db):
@@ -368,21 +372,23 @@ def test_config_shutdown_module(self):
         with mock.patch("config.chassis_modules.StateDBHelper", new=MockStateDBHelper):
             runner = CliRunner()
             db = Db()
-            result = runner.invoke(config.config.commands["chassis"].commands["modules"].commands["shutdown"], ["LINE-CARD0"], obj=db)
+            result = runner.invoke(config.config.commands["chassis"].commands["modules"].
+                                   commands["shutdown"], ["LINE-CARD0"], obj=db)
             print(result.exit_code)
             print(result.output)
             assert result.exit_code == 0
 
-            result = runner.invoke(show.cli.commands["chassis"].commands["modules"].commands["status"], ["LINE-CARD0"], obj=db)
+            result = runner.invoke(show.cli.commands["chassis"].commands["modules"].
+                                   commands["status"], ["LINE-CARD0"], obj=db)
             print(result.exit_code)
             print(result.output)
             result_lines = result.output.strip('\n').split('\n')
             assert result.exit_code == 0
             header_lines = 2
             result_out = " ".join((result_lines[header_lines]).split())
             assert result_out.strip('\n') == show_linecard0_shutdown_output.strip('\n')
-            #db.cfgdb.set_entry("CHASSIS_MODULE", "LINE-CARD0", { "admin_status" : "down" })
-            #db.get_data("CHASSIS_MODULE", "LINE-CARD0")
+            # db.cfgdb.set_entry("CHASSIS_MODULE", "LINE-CARD0", { "admin_status" : "down" })
+            # db.get_data("CHASSIS_MODULE", "LINE-CARD0")
 
     def test_config_shutdown_module_fabric(self):
         with mock.patch("utilities_common.cli.run_command",
@@ -430,12 +436,14 @@ def test_config_startup_module(self):
         with mock.patch("config.chassis_modules.StateDBHelper", new=MockStateDBHelper):
             runner = CliRunner()
             db = Db()
-            result = runner.invoke(config.config.commands["chassis"].commands["modules"].commands["startup"], ["LINE-CARD0"], obj=db)
+            result = runner.invoke(config.config.commands["chassis"].commands["modules"].
+                                   commands["startup"], ["LINE-CARD0"], obj=db)
             print(result.exit_code)
             print(result.output)
             assert result.exit_code == 0
 
-            result = runner.invoke(show.cli.commands["chassis"].commands["modules"].commands["status"], ["LINE-CARD0"], obj=db)
+            result = runner.invoke(show.cli.commands["chassis"].commands["modules"].
+                                   commands["status"], ["LINE-CARD0"], obj=db)
             print(result.exit_code)
             print(result.output)
             result_lines = result.output.strip('\n').split('\n')