diff options
Diffstat (limited to 'test/mocks/mass-pnf-sim/test_cli.py')
| -rw-r--r-- | test/mocks/mass-pnf-sim/test_cli.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test/mocks/mass-pnf-sim/test_cli.py b/test/mocks/mass-pnf-sim/test_cli.py index 70c9b6c4f..06d018f39 100644 --- a/test/mocks/mass-pnf-sim/test_cli.py +++ b/test/mocks/mass-pnf-sim/test_cli.py @@ -26,7 +26,7 @@ def test_validate_trigger_custom(parser, caplog): args = parser.parse_args(['trigger_custom', '--triggerstart', '0', '--triggerend', str(SIM_INSTANCES)]) try: - MassPnfSim(args).trigger_custom() + MassPnfSim().trigger_custom(args) except SystemExit as e: assert e.code == 1 assert "--triggerend value greater than existing instance count" in caplog.text @@ -59,8 +59,8 @@ 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() + m = getattr(MassPnfSim(), subcommand) + m(args) except SystemExit: pass assert '--count value greater that existing instance count' in caplog.text |