fix the API args caused by merge, make linter happy

sonic-net · Sep 8, 2020 · d199adf · d199adf
1 parent 8359abf
commit d199adf
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/pfcwd/main.py b/pfcwd/main.py
@@ -72,7 +72,7 @@ def get_all_queues(db, namespace=None, display=constants.DISPLAY_ALL):
     return natsorted(queues)
 
 
-def get_all_ports(db):
+def get_all_ports(db, namespace=None, display=display=constants.DISPLAY_ALL):
     all_port_names = db.get_all(db.COUNTERS_DB, 'COUNTERS_PORT_NAME_MAP')
 
     # Get list of physical ports

diff --git a/sonic-utilities-tests/pfcstat_test.py b/sonic-utilities-tests/pfcstat_test.py
@@ -1,4 +1,3 @@
-import imp
 import os
 import shutil
 

diff --git a/sonic-utilities-tests/pfcwd_test.py b/sonic-utilities-tests/pfcwd_test.py
@@ -3,7 +3,7 @@
 import sys
 
 from click.testing import CliRunner
-from utilities_common.db import Db
+from swsssdk import ConfigDBConnector
 
 from pfcwd_input.pfcwd_test_vectors import (
     testData, show_pfcwd_stats_all, show_pfc_config_all,
@@ -51,7 +51,8 @@ def test_pfcwd_show_stats_invalid_queue(self):
     def executor(self, testcase):
         import pfcwd.main as pfcwd
         runner = CliRunner()
-        db = Db()
+        db = ConfigDBConnector()
+        db.connect()
 
         for input in testcase:
             exec_cmd = ""