46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import sys
|
||
|
|
from io import StringIO
|
||
|
|
|
||
|
|
import mypy.api
|
||
|
|
from mypy.test.helpers import Suite
|
||
|
|
|
||
|
|
|
||
|
|
class APISuite(Suite):
|
||
|
|
def setUp(self) -> None:
|
||
|
|
self.sys_stdout = sys.stdout
|
||
|
|
self.sys_stderr = sys.stderr
|
||
|
|
sys.stdout = self.stdout = StringIO()
|
||
|
|
sys.stderr = self.stderr = StringIO()
|
||
|
|
|
||
|
|
def tearDown(self) -> None:
|
||
|
|
sys.stdout = self.sys_stdout
|
||
|
|
sys.stderr = self.sys_stderr
|
||
|
|
assert self.stdout.getvalue() == ""
|
||
|
|
assert self.stderr.getvalue() == ""
|
||
|
|
|
||
|
|
def test_capture_bad_opt(self) -> None:
|
||
|
|
"""stderr should be captured when a bad option is passed."""
|
||
|
|
_, stderr, _ = mypy.api.run(["--some-bad-option"])
|
||
|
|
assert isinstance(stderr, str)
|
||
|
|
assert stderr != ""
|
||
|
|
|
||
|
|
def test_capture_empty(self) -> None:
|
||
|
|
"""stderr should be captured when a bad option is passed."""
|
||
|
|
_, stderr, _ = mypy.api.run([])
|
||
|
|
assert isinstance(stderr, str)
|
||
|
|
assert stderr != ""
|
||
|
|
|
||
|
|
def test_capture_help(self) -> None:
|
||
|
|
"""stdout should be captured when --help is passed."""
|
||
|
|
stdout, _, _ = mypy.api.run(["--help"])
|
||
|
|
assert isinstance(stdout, str)
|
||
|
|
assert stdout != ""
|
||
|
|
|
||
|
|
def test_capture_version(self) -> None:
|
||
|
|
"""stdout should be captured when --version is passed."""
|
||
|
|
stdout, _, _ = mypy.api.run(["--version"])
|
||
|
|
assert isinstance(stdout, str)
|
||
|
|
assert stdout != ""
|