Added tests

Author: dbrgn

.gitignore

@@ -4,3 +4,5 @@ __pycache__
 build/
 dist/
 *.egg-info/
+.cache/
+.coverage

pytest.ini

@@ -0,0 +1,12 @@
+[pytest]
+addopts = --pep8 --tb=short --cov
+python_files = test_*.py
+norecursedirs = *.egg tmp* build .tox
+pep8ignore =
+    *.py E126 E127 E128
+    setup.py ALL
+    */tests/* ALL
+    */docs/* ALL
+    */build/* ALL
+    */TOXENV/* ALL
+pep8maxlinelength = 99

requirements-dev.txt

@@ -0,0 +1,4 @@
+pytest==2.8.2
+pytest-cache==1.0
+pytest-cov==2.2.0
+pytest-pep8==1.0.6

tests/conftest.py

@@ -0,0 +1,9 @@
+# -*- coding: utf-8 -*-
+from __future__ import print_function, division, absolute_import, unicode_literals
+
+import sys
+import os
+
+current_dir = os.path.dirname(os.path.abspath(__file__))
+parent_dir = os.path.normpath(os.path.join(current_dir, '..'))
+sys.path.insert(0, parent_dir)

tests/test_output.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+from __future__ import print_function, division, absolute_import, unicode_literals
+
+import sys
+
+import pytest
+
+from coverage_badge import __main__
+
+
+@pytest.fixture
+def cb(monkeypatch):
+    """
+    Return a monkey patched coverage_badge module that always returns a percentage of 79.
+    """
+    def get_fake_total():
+        return '79'
+    monkeypatch.setattr(__main__, 'get_total', get_fake_total)
+    return __main__
+
+
+def test_version(cb, capsys):
+    """
+    Test the version output.
+    """
+    with pytest.raises(SystemExit) as se:
+        cb.main(['-v'])
+    out, _ = capsys.readouterr()
+    assert out == 'coverage-badge v%s\n' % __main__.__version__
+
+
+def test_svg_output(cb, capsys):
+    """
+    Test the SVG output.
+    """
+    cb.main([])
+    out, _ = capsys.readouterr()
+    assert out.startswith('<svg xmlns="http://www.w3.org/2000/svg" width="99" height="20">')
+    assert '<text x="80" y="14">79%</text>' in out
+    assert out.endswith('</svg>\n')