rewrite pytest-mypy-plugin extension code to use new hook

Author: mkurnikov

.travis.yml

@@ -8,8 +8,7 @@ jobs:
       python: 3.7
       script: |
         set -e
-        pip install pytest_plugin/
-        pytest -p no:pytest-mypy-plugins
+        pytest
 
     - name: Typecheck Django test suite
       python: 3.7

pytest.ini

@@ -2,7 +2,8 @@
 testpaths = ./test-data
 addopts =
     --tb=native
-    --mypy-ini-file=./test-data/plugins.ini
     -s
     -v
-    --cache-clear
+    --cache-clear
+    --mypy-ini-file=./test-data/plugins.ini
+    --mypy-extension-hook=scripts.tests_extension_hook.django_plugin_hook

pytest_plugin/collect.py

@@ -0,0 +1,28 @@
+from pytest_mypy.collect import File
+from pytest_mypy.item import YamlTestItem
+
+
+def django_plugin_hook(test_item: YamlTestItem) -> None:
+    settings = {
+        'SECRET_KEY': '"1"',
+    }
+    additional_settings = test_item.parsed_test_data.get('additional_settings')
+    if additional_settings:
+        for item in additional_settings:
+            name, _, val = item.partition('=')
+            settings[name] = val
+
+    installed_apps = test_item.parsed_test_data.get('installed_apps', None)
+    if installed_apps is not None:
+        installed_apps += ['django.contrib.contenttypes']
+        installed_apps_as_str = '(' + ','.join([repr(app) for app in installed_apps]) + ',)'
+
+        pyproject_toml_file = File(path='pyproject.toml',
+                                   content='[tool.django-stubs]\ndjango_settings_module=\'mysettings\'')
+        test_item.files.append(pyproject_toml_file)
+
+        settings_contents = f'INSTALLED_APPS={installed_apps_as_str}\n'
+        settings_contents += '\n'.join([f'{key}={val}' for key, val in settings.items()])
+
+        mysettings_file = File(path='mysettings.py', content=settings_contents)
+        test_item.files.append(mysettings_file)