test/mocks/mass-pnf-sim/test_cli.py

   1 import pytest
   2 from MassPnfSim import MassPnfSim
   3 from test_settings import SIM_INSTANCES
   4
   5 @pytest.mark.parametrize(('expect_string, cli_opts'), [
   6     ("bootstrap: error: the following arguments are required: --urlves, --ipfileserver, --typefileserver, --ipstart",
   7      ['bootstrap']),
   8     ("bootstrap: error: argument --typefileserver: invalid choice: 'dummy' (choose from 'sftp', 'ftps')",
   9      ['bootstrap', '--typefileserver', 'dummy']),
  10     ("bootstrap: error: argument --urlves: invalid_url is not a valid URL",
  11      ['bootstrap', '--urlves', 'invalid_url']),
  12     ("bootstrap: error: argument --ipstart: x.x.x.x is not a valid IP address",
  13      ['bootstrap', '--ipstart', 'x.x.x.x']),
  14     ("trigger_custom: error: the following arguments are required: --triggerstart, --triggerend",
  15      ['trigger_custom'])
  16     ])
  17 def test_subcommands(parser, capsys, expect_string, cli_opts):
  18     try:
  19         parser.parse_args(cli_opts)
  20     except SystemExit:
  21         pass
  22     assert expect_string in capsys.readouterr().err
  23
  24 def test_validate_trigger_custom(parser, caplog):
  25     args = parser.parse_args(['trigger_custom', '--triggerstart', '0',
  26                              '--triggerend', str(SIM_INSTANCES)])
  27     try:
  28         MassPnfSim(args).trigger_custom()
  29     except SystemExit as e:
  30         assert e.code == 1
  31     assert "--triggerend value greater than existing instance count" in caplog.text
  32     caplog.clear()
  33
  34 @pytest.mark.parametrize(("subcommand"), [
  35     'bootstrap',
  36     'start',
  37     'stop',
  38     'trigger',
  39     'status'
  40     ])
  41 def test_count_option(parser, capsys, subcommand):
  42     try:
  43         parser.parse_args([subcommand, '--count'])
  44     except SystemExit:
  45         pass
  46     assert f"{subcommand}: error: argument --count: expected one argument" in capsys.readouterr().err
  47
  48 def test_empty(parser, capsys):
  49     try:
  50         parser.parse_args([])
  51     except SystemExit:
  52         pass
  53     assert '' is capsys.readouterr().err
  54     assert '' is capsys.readouterr().out