Python unittest passing arguments

by

So the doctors here that are saying “You say that hurts? Then don’t do that!” are probably right. But if you really want to, here’s one way of passing arguments to a unittest test:

import sys
import unittest

class MyTest(unittest.TestCase):
    USERNAME = "jemima"
    PASSWORD = "password"

    def test_logins_or_something(self):
        print('username:', self.USERNAME)
        print('password:', self.PASSWORD)


if __name__ == "__main__":
    if len(sys.argv) > 1:
        MyTest.USERNAME = sys.argv.pop()
        MyTest.PASSWORD = sys.argv.pop()
    unittest.main()

That will let you run with:

python mytests.py myusername mypassword

You need the argv.pops, so your command line parameters don’t mess with unittest’s own…

The other thing you might want to look into is using environment variables:

import os
import unittest

class MyTest(unittest.TestCase):
    USERNAME = "jemima"
    PASSWORD = "password"

    def test_logins_or_something(self):
        print('username:', self.USERNAME)
        print('password:', self.PASSWORD)

if __name__ == "__main__":
    MyTest.USERNAME = os.environ.get('TEST_USERNAME', MyTest.USERNAME)
    MyTest.PASSWORD = os.environ.get('TEST_PASSWORD', MyTest.PASSWORD)
    unittest.main()

That will let you run with:

TEST_USERNAME=ausername TEST_PASSWORD=apassword python mytests.py

And it has the advantage that you’re not messing with unittest’s own argument parsing. The downside is it won’t work quite like that on Windows…

More Related Contents:

  1. Why doesn’t Python run my bash code?
  2. PATH issue with pytest ‘ImportError: No module named YadaYadaYada’
  3. Running unittest with typical test directory structure
  4. python mocking raw input in unittests
  5. Run / Debug a Django application’s UnitTests from the mouse right click context menu in PyCharm Community Edition?
  6. Can I patch a Python decorator before it wraps a function?
  7. changing order of unit tests in Python
  8. Running a single test from unittest.TestCase via the command line
  9. Python unit test with base and sub class
  10. How to assert output with nosetest/unittest in python?
  11. Getting Python’s unittest results in a tearDown() method
  12. Why python mock patch doesn’t work?
  13. Python mock multiple return values
  14. Python Mock object with method called multiple times
  15. How can I create a ramdisk in Python?
  16. How to test Python 3.4 asyncio code?
  17. Continuing in Python’s unittest when an assertion fails
  18. How can I disable logging while running unit tests in Python Django?
  19. Mock Python’s built in print function
  20. Unittest tests order
  21. Can Python’s unittest test in parallel, like nose can?
  22. Mocking Functions Using Python Mock
  23. Python mock Patch os.environ and return value
  24. Customizing unittest.mock.mock_open for iteration
  25. What is the difference between setUp() and setUpClass() in Python unittest?
  26. Outputting data from unit test in Python
  27. Is it possible for a unit test to assert that a method calls sys.exit()?
  28. How do I run unittest on a Tkinter app?
  29. pydev breakpoints not working
  30. How to mock a readonly property with mock?

Leave a Comment