mirror of
https://github.com/davidhalter/django-stubs.git
synced 2025-12-07 20:54:29 +08:00
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from mypy import api
|
|
from mypy.test.config import test_temp_dir
|
|
from mypy.test.data import DataSuite, DataDrivenTestCase
|
|
from mypy.test.helpers import assert_string_arrays_equal
|
|
|
|
ROOT_DIR = Path(__file__).parent.parent
|
|
TEST_DATA_DIR = ROOT_DIR / 'test' / 'test-data'
|
|
MYPY_INI_PATH = ROOT_DIR / 'test' / 'plugins.ini'
|
|
|
|
|
|
class DjangoTestSuite(DataSuite):
|
|
files = [
|
|
'check-model-fields.test',
|
|
'check-postgres-fields.test',
|
|
'check-model-relations.test'
|
|
]
|
|
data_prefix = str(TEST_DATA_DIR)
|
|
|
|
def run_case(self, testcase: DataDrivenTestCase) -> None:
|
|
assert testcase.old_cwd is not None, "test was not properly set up"
|
|
|
|
mypy_cmdline = [
|
|
'--show-traceback',
|
|
'--no-silence-site-packages',
|
|
'--config-file={}'.format(MYPY_INI_PATH)
|
|
]
|
|
mypy_cmdline.append('--python-version={}'.format('.'.join(map(str,
|
|
sys.version_info[:2]))))
|
|
|
|
program_path = os.path.join(test_temp_dir, 'main.py')
|
|
mypy_cmdline.append(program_path)
|
|
|
|
with open(program_path, 'w') as file:
|
|
for s in testcase.input:
|
|
file.write('{}\n'.format(s))
|
|
|
|
output = []
|
|
# Type check the program.
|
|
out, err, returncode = api.run(mypy_cmdline)
|
|
# split lines, remove newlines, and remove directory of test case
|
|
for line in (out + err).splitlines():
|
|
if line.startswith(test_temp_dir + os.sep):
|
|
output.append(line[len(test_temp_dir + os.sep):].rstrip("\r\n").replace('.py', ''))
|
|
else:
|
|
output.append(line.rstrip("\r\n"))
|
|
# Remove temp file.
|
|
os.remove(program_path)
|
|
|
|
assert_string_arrays_equal(testcase.output, output,
|
|
'Invalid output ({}, line {})'.format(
|
|
testcase.file, testcase.line))
|