diff --git a/tests/storm_control_test.py b/tests/storm_control_test.py index f54e88774e..2f2320576a 100644 --- a/tests/storm_control_test.py +++ b/tests/storm_control_test.py @@ -18,7 +18,7 @@ def test_add_broadcast_storm(self): db = Db() obj = {'db':db.cfgdb} - result = runner.invoke(config.config.commands["storm_control"].commands["add"], ["Ethernet0", "broadcast", "10000"], obj = obj) + result = runner.invoke(config.config.commands["storm-control"].commands["add"], ["Ethernet0", "broadcast", "10000"], obj = obj) print (result.exit_code) print (result.output) assert result.exit_code == 0 @@ -28,7 +28,7 @@ def test_add_uucast_storm(self): db = Db() obj = {'db':db.cfgdb} - result = runner.invoke(config.config.commands["storm_control"].commands["add"], ["Ethernet0", "unknown-unicast", "20000"], obj = obj) + result = runner.invoke(config.config.commands["storm-control"].commands["add"], ["Ethernet0", "unknown-unicast", "20000"], obj = obj) print (result.exit_code) print (result.output) assert result.exit_code == 0 @@ -38,7 +38,7 @@ def test_add_umcast_storm(self): db = Db() obj = {'db':db.cfgdb} - result = runner.invoke(config.config.commands["storm_control"].commands["add"], ["Ethernet0", "unknown-multicast", "30000"], obj = obj) + result = runner.invoke(config.config.commands["storm-control"].commands["add"], ["Ethernet0", "unknown-multicast", "30000"], obj = obj) print (result.exit_code) print (result.output) assert result.exit_code == 0 @@ -48,7 +48,7 @@ def test_del_broadcast_storm(self): db = Db() obj = {'db':db.cfgdb} - result = runner.invoke(config.config.commands["storm_control"].commands["del"], ["Ethernet0", "broadcast"], obj = obj) + result = runner.invoke(config.config.commands["storm-control"].commands["del"], ["Ethernet0", "broadcast"], obj = obj) print (result.exit_code) print (result.output) assert result.exit_code == 0 @@ -58,7 +58,7 @@ def test_del_uucast_storm(self): db = Db() obj = {'db':db.cfgdb} - result = runner.invoke(config.config.commands["storm_control"].commands["del"], ["Ethernet0", "unknown-unicast"], obj = obj) + result = runner.invoke(config.config.commands["storm-control"].commands["del"], ["Ethernet0", "unknown-unicast"], obj = obj) print (result.exit_code) print (result.output) assert result.exit_code == 0 @@ -68,7 +68,7 @@ def test_del_umcast_storm(self): db = Db() obj = {'db':db.cfgdb} - result = runner.invoke(config.config.commands["storm_control"].commands["del"], ["Ethernet0", "unknown-multicast"], obj = obj) + result = runner.invoke(config.config.commands["storm-control"].commands["del"], ["Ethernet0", "unknown-multicast"], obj = obj) print (result.exit_code) print (result.output) assert result.exit_code == 0