aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--test/mocks/mass-pnf-sim/test_cli.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/test/mocks/mass-pnf-sim/test_cli.py b/test/mocks/mass-pnf-sim/test_cli.py
index f574a725d..0f464167a 100644
--- a/test/mocks/mass-pnf-sim/test_cli.py
+++ b/test/mocks/mass-pnf-sim/test_cli.py
@@ -39,12 +39,30 @@ def test_validate_trigger_custom(parser, caplog):
'status'
])
def test_count_option(parser, capsys, subcommand):
+ '''Test case where no arg passed to '--count' opt'''
try:
parser.parse_args([subcommand, '--count'])
except SystemExit:
pass
assert f"{subcommand}: error: argument --count: expected one argument" in capsys.readouterr().err
+@pytest.mark.parametrize(("subcommand"), [
+ 'start',
+ 'stop',
+ 'trigger',
+ 'status'
+ ])
+def test_count_option_bad_value(parser, caplog, subcommand):
+ '''Test case where invalid value passed to '--count' opt'''
+ try:
+ args = parser.parse_args([subcommand, '--count', str(SIM_INSTANCES + 1)])
+ m = getattr(MassPnfSim(args), subcommand)
+ m()
+ except SystemExit:
+ pass
+ assert '--count value greater that existing instance count' in caplog.text
+ caplog.clear()
+
def test_empty(parser, capsys):
try:
parser.parse_args([])