Add parametrization

sonic-net · Jul 13, 2022 · f266dfb · f266dfb
1 parent cb88ec8
commit f266dfb
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/portchannel_test.py b/tests/portchannel_test.py
@@ -1,4 +1,5 @@
 import os
+import pytest
 import traceback
 
 from click.testing import CliRunner
@@ -61,24 +62,26 @@ def test_delete_non_existing_portchannel(self):
         assert result.exit_code != 0
         assert "Error: PortChannel0005 is not present." in result.output
 
-    def test_add_portchannel_with_fast_rate(self):
+    @pytest.mark.parametrize("fast_rate", ["False", "True", "false", "true"])
+    def test_add_portchannel_with_fast_rate(self, fast_rate):
         runner = CliRunner()
         db = Db()
         obj = {'db':db.cfgdb}
 
         # add a portchannel with fats rate
-        result = runner.invoke(config.config.commands["portchannel"].commands["add"], ["PortChannel0005", "--fast-rate", "True"], obj=obj)
+        result = runner.invoke(config.config.commands["portchannel"].commands["add"], ["PortChannel0005", "--fast-rate", fast_rate], obj=obj)
         print(result.exit_code)
         print(result.output)
         assert result.exit_code == 0
 
+    @pytest.mark.parametrize("fast_rate", ["Fls", "tru"])
     def test_add_portchannel_with_invalid_fast_rate(self):
         runner = CliRunner()
         db = Db()
         obj = {'db':db.cfgdb}
 
         # add a portchannel with invalid fats rate
-        result = runner.invoke(config.config.commands["portchannel"].commands["add"], ["PortChannel0005", "--fast-rate", "Tru"], obj=obj)
+        result = runner.invoke(config.config.commands["portchannel"].commands["add"], ["PortChannel0005", "--fast-rate", fast_rate], obj=obj)
         print(result.exit_code)
         print(result.output)
         assert result.exit_code != 0