ff5a7e6377
https://fedoraproject.org/wiki/Changes/RemovePythonMockUsage Prefer using unittest.mock to a standalone package. The separate packages should only really be needed on Python 2.7 these days. The test requirements file is updated to only require mock on old Python, and the dependency is removed from setup.py to avoid issues there. Relates: https://src.fedoraproject.org/rpms/pungi/pull-request/9 Signed-off-by: Lubomír Sedlář <lsedlar@redhat.com>
29 lines
937 B
Python
29 lines
937 B
Python
try:
|
|
from unittest import mock
|
|
except ImportError:
|
|
import mock
|
|
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
import unittest
|
|
import six
|
|
|
|
from pungi.scripts.pungi_koji import cli_main
|
|
|
|
|
|
class PungiKojiTestCase(unittest.TestCase):
|
|
@mock.patch("sys.argv", new=["prog", "--version"])
|
|
@mock.patch("sys.stderr", new_callable=six.StringIO)
|
|
@mock.patch("sys.stdout", new_callable=six.StringIO)
|
|
@mock.patch("pungi.scripts.pungi_koji.get_full_version", return_value="a-b-c.111")
|
|
def test_version(self, get_full_version, stdout, stderr):
|
|
with self.assertRaises(SystemExit) as cm:
|
|
cli_main()
|
|
self.assertEqual(cm.exception.code, 0)
|
|
# Python 2.7 prints the version to stderr, 3.4+ to stdout.
|
|
if six.PY3:
|
|
self.assertMultiLineEqual(stdout.getvalue(), "a-b-c.111\n")
|
|
else:
|
|
self.assertMultiLineEqual(stderr.getvalue(), "a-b-c.111\n")
|