Merge pull request #606 from gleybersonandrade/tags

Create tags decorator to tests

Author: hdiogenes

.coveragerc

@@ -1,2 +1,3 @@
 [run]
 source = pyof
+omit = .eggs/*,.tox/*,*tests*,setup.py

setup.cfg

@@ -14,4 +14,10 @@ add-ignore = D105
 
 [isort]
 known_first_party = pyof,tests
-multi_line_output = 4
+multi_line_output = 4
+
+[tool:pytest]
+markers =
+    small: marks tests as small
+    medium: marks tests as medium
+    large: marks tests as large

setup.py

@@ -45,6 +45,39 @@ def finalize_options(self):
         """Post-process options."""
 
 
+# pylint: disable=attribute-defined-outside-init, abstract-method
+class TestCommand(Command):
+    """Test tags decorators."""
+
+    user_options = [
+        ('size=', None, 'Specify the size of tests to be executed.'),
+        ('type=', None, 'Specify the type of tests to be executed.'),
+    ]
+
+    sizes = ('small', 'medium', 'large', 'all')
+    types = ('unit', 'integration', 'e2e')
+
+    def get_args(self):
+        """Return args to be used in test command."""
+        return '--size %s --type %s' % (self.size, self.type)
+
+    def initialize_options(self):
+        """Set default size and type args."""
+        self.size = 'all'
+        self.type = 'unit'
+
+    def finalize_options(self):
+        """Post-process."""
+        try:
+            assert self.size in self.sizes, ('ERROR: Invalid size:'
+                                             f':{self.size}')
+            assert self.type in self.types, ('ERROR: Invalid type:'
+                                             f':{self.type}')
+        except AssertionError as exc:
+            print(exc)
+            sys.exit(-1)
+
+
 class Cleaner(clean):
     """Custom clean command to tidy up the project root."""
 
@@ -58,15 +91,41 @@ def run(self):
         call('test -d docs && make -C docs/ clean', shell=True)
 
 
-class TestCoverage(SimpleCommand):
+class Test(TestCommand):
+    """Run all tests."""
+
+    description = 'run tests and display results'
+
+    def get_args(self):
+        """Return args to be used in test command."""
+        markers = self.size
+        if markers == "small":
+            markers = 'not medium and not large'
+        size_args = "" if self.size == "all" else "-m '%s'" % markers
+        return '--addopts="tests/%s %s"' % (self.type, size_args)
+
+    def run(self):
+        """Run tests."""
+        cmd = 'python setup.py pytest %s' % self.get_args()
+        try:
+            check_call(cmd, shell=True)
+        except CalledProcessError as exc:
+            print(exc)
+
+
+class TestCoverage(Test):
     """Display test coverage."""
 
-    description = 'run unit tests and display code coverage'
+    description = 'run tests and display code coverage'
 
     def run(self):
-        """Run unittest quietly and display coverage report."""
-        cmd = 'coverage3 run setup.py test && coverage3 report'
-        check_call(cmd, shell=True)
+        """Run tests quietly and display coverage report."""
+        cmd = 'coverage3 run setup.py pytest %s' % self.get_args()
+        cmd += '&& coverage3 report'
+        try:
+            check_call(cmd, shell=True)
+        except CalledProcessError as exc:
+            print(exc)
 
 
 class DocTest(SimpleCommand):
@@ -80,17 +139,6 @@ def run(self):
         check_call(cmd, shell=True)
 
 
-class CITest(SimpleCommand):
-    """Run all CI tests."""
-
-    description = 'run all CI tests: unit and doc tests, linter'
-
-    def run(self):
-        """Run unit tests with coverage, doc tests and linter."""
-        for command in TestCoverage, DocTest, Linter:
-            self.run_command(command)
-
-
 class Linter(SimpleCommand):
     """Lint Python source code."""
 
@@ -107,6 +155,20 @@ def run(self):
             sys.exit(-1)
 
 
+class CITest(TestCommand):
+    """Run all CI tests."""
+
+    description = 'run all CI tests: unit and doc tests, linter'
+
+    def run(self):
+        """Run unit tests with coverage, doc tests and linter."""
+        coverage_cmd = 'python setup.py coverage %s' % self.get_args()
+        doctest_cmd = 'python setup.py doctest'
+        lint_cmd = 'python setup.py lint'
+        cmd = '%s && %s && %s' % (coverage_cmd, doctest_cmd, lint_cmd)
+        check_call(cmd, shell=True)
+
+
 setup(name='python-openflow',
       version=__version__,
       description='Library to parse and generate OpenFlow messages',
@@ -127,7 +189,8 @@ def run(self):
           'clean': Cleaner,
           'coverage': TestCoverage,
           'doctest': DocTest,
-          'lint': Linter
+          'lint': Linter,
+          'test': Test
       },
       zip_safe=False,
       classifiers=[

tests/unit/__init__.py

@@ -0,0 +1 @@
+"""python-openflow unit tests."""