Switch using pytest

aaltat · aaltat · commit bfdbf1ea12f1 · 2019-09-28T13:01:36.000+03:00
diff --git a/.gitignore b/.gitignore
@@ -10,6 +10,7 @@ MANIFEST
 *.egg-info
 *.egg
 chromedriver.log
+.pytest_cache
 
 dist
 build
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -4,6 +4,7 @@
 mockito >= 1.0.0
 robotstatuschecker >= 1.4
 approvaltests >= 0.2.4
+pytest
 requests
 
 # Include normal dependencies from requirements.txt. Makes it possible to use
diff --git a/utest/README.rst b/utest/README.rst
@@ -8,7 +8,7 @@ Before running the test, install the dependencies::
 
 Unit Tests
 ----------
-Units tests are written by using the Python default `unittest`_ framework.
+Units tests are written by using the `pytest`_  framework.
 Unit test can be executed by running::
 
     python utest/run.py
@@ -33,7 +33,7 @@ must handled with `try/except ImportError:` and skipped with:
 imported from `from robot.utils import JYTHON`
 
 
-.. _unittest: https://docs.python.org/3/library/unittest.html
+.. _pytest: https://docs.pytest.org/en/latest/
 .. _ApprovalTests: https://github.com/approvals/ApprovalTests.Python
 .. _ApprovalTests blog post: http://blog.approvaltests.com/
 .. _Jython: http://www.jython.org/
diff --git a/utest/run.py b/utest/run.py
@@ -4,10 +4,12 @@
 import shutil
 import sys
 from os.path import abspath, dirname, join
-from unittest import defaultTestLoader, TextTestRunner
+
+from pytest import main as py_main
 
 
 CURDIR = dirname(abspath(__file__))
+SRC = join(CURDIR, os.pardir, 'src')
 
 
 def remove_output_dir():
@@ -18,13 +20,12 @@ def remove_output_dir():
 
 
 def run_unit_tests():
-    sys.path.insert(0, join(CURDIR, os.pardir, 'src'))
+    sys.path.insert(0, SRC)
     try:
-        suite = defaultTestLoader.discover(join(CURDIR, 'test'), 'test_*.py')
-        result = TextTestRunner().run(suite)
+        result = py_main(['--rootdir=%s' % CURDIR])
     finally:
         sys.path.pop(0)
-    return min(len(result.failures) + len(result.errors), 255)
+    return result
 
 
 if __name__ == '__main__':
