From 6e136a0d7f4e93e6557a17dabb62823f1b420248 Mon Sep 17 00:00:00 2001
From: Rida Hanif <rida.hanif@xflowresearch.com>
Date: Tue, 14 May 2024 11:25:51 +0000
Subject: [PATCH] Modified shpw command

---
 tests/dhcp_rate_test.py | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/tests/dhcp_rate_test.py b/tests/dhcp_rate_test.py
index b029ab3b80..45d8f79793 100644
--- a/tests/dhcp_rate_test.py
+++ b/tests/dhcp_rate_test.py
@@ -1,14 +1,16 @@
 import os
-import imp
-
+import traceback
+import pytest
+from unittest import mock
 from click.testing import CliRunner
+from utilities_common.db import Db
 
 import config.main as config
 import show.main as show
-from utilities_common.db import Db
 
 
-test_config_dhcp_rate_add_del_output = """\
+
+show_interface_dhcp_rate_limit_output ="""\
 Interface        DHCP Mitigation Rate
 -----------     ----------------------
 Ethernet0                       300
@@ -17,7 +19,7 @@
 Ethernet12                      300
 Ethernet16                      300
 Ethernet20                      300
-Ethernet24                      
+Ethernet24                      300
 Ethernet28                      300
 Ethernet32                      45
 Ethernet36                      300
@@ -47,9 +49,6 @@
 
 
 class TestDHCPRate(object):
-
-    _old_run_bgp_command = None
-
     @classmethod
     def setup_class(cls):
         os.environ['UTILITIES_UNIT_TESTING'] = "1"
@@ -172,12 +171,16 @@ def test_config_dhcp_rate_add_del(self):
         print(result.output)
         assert result.exit_code == 0
 
-        # show output
-        result = runner.invoke(show.cli.commands["interfaces"].commands["dhcp-mitigation-rate"], [], obj=obj)
+
+    def test_show_dhcp_rate_limit(self):
+        runner = CliRunner()
+        result = runner.invoke(show.cli.commands["interfaces"].commands["dhcp-mitigation-rate"], [])
         print(result.exit_code)
         print(result.output)
         assert result.exit_code == 0
-        assert result.output == test_config_dhcp_rate_add_del_output
+        assert result.output == show_interface_dhcp_rate_limit_output
+
+
 
     @classmethod
     def teardown_class(cls):