conftest.py 946 B

123456789101112131415161718192021
  1. def pytest_addoption(parser):
  2. parser.addoption("--port", type=str, help="default is 2244 local, 22 remote")
  3. parser.addoption("--dbclient", type=str, default="../dbclient")
  4. parser.addoption("--dropbear", type=str, default="../dropbear")
  5. parser.addoption("--dropbearconvert", type=str, default="../dropbearconvert")
  6. parser.addoption("--dropbearkey", type=str, default="../dropbearkey")
  7. parser.addoption("--hostkey", type=str, help="required unless --remote")
  8. parser.addoption("--remote", type=str, help="remote host")
  9. parser.addoption("--user", type=str, help="optional username")
  10. parser.addoption("--ssh-keygen", type=str, default="ssh-keygen")
  11. def pytest_configure(config):
  12. opt = config.option
  13. if not opt.hostkey and not opt.remote:
  14. raise Exception("--hostkey must be given")
  15. if not opt.port:
  16. if opt.remote:
  17. opt.port = "22"
  18. else:
  19. opt.port = "2244"