Add some testing infrastructure (#13)

This includes only basic infra

Author: ilevkivskyi

.gitignore

@@ -0,0 +1,39 @@
+build/
+__pycache__
+*.py[cod]
+*~
+@*
+/build
+/env
+docs/build/
+*.iml
+/out/
+.venv/
+.mypy_cache/
+.incremental_checker_cache.json
+.cache
+dmypy.json
+.dmypy.json
+
+# Packages
+*.egg
+*.egg-info
+*.eggs
+
+# IDEs
+.idea
+*.swp
+.vscode
+
+# Operating Systems
+.DS_Store
+
+# Coverage Files
+htmlcov
+.coverage*
+
+# pytest cache
+.pytest_cache/
+
+.tox
+

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "external/mypy"]
+	path = external/mypy
+	url = https://github.com/python/mypy.git

.travis.yml

@@ -0,0 +1,24 @@
+language: python
+# cache package wheels (1 cache per python version)
+cache: pip
+# newer python versions are available only on xenial (while some older only on trusty) Ubuntu distribution
+dist: xenial
+sudo: required
+
+jobs:
+  include:
+  - name: "run test suite with python 3.6"
+    python: 3.6
+
+before_install: |
+  # Upgrade pip, setuptools, and wheel
+  pip install -U pip setuptools wheel
+
+install:
+  - pip install -r external/mypy/test-requirements.txt
+
+script: |
+  set -e
+  export PYTHONPATH=`pwd`/external/mypy
+  pytest
+ 

conftest.py

@@ -0,0 +1,4 @@
+pytest_plugins = [
+    'mypy.test.data',
+]
+

external/mypy

@@ -0,0 +1,5 @@
+[pytest]
+testpaths = test
+python_files = test*.py
+addopts = -nauto
+

pytest.ini

@@ -0,0 +1,31 @@
+[case testColumnFieldsInferred]
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy import Column, Integer, String
+
+Base = declarative_base()
+
+class User(Base):
+    __tablename__ = 'users'
+    id = Column(Integer, primary_key=True)
+    name = Column(String)
+
+user = None  # type: User
+reveal_type(user.id)
+reveal_type(User.name)
+[out]
+
+[case testColumnFieldsInferredInstance]
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy import Column, Integer, String
+
+Base = declarative_base()
+
+class User(Base):
+    __tablename__ = 'users'
+    id = Column(Integer(), primary_key=True)
+    name = Column(String())
+
+user = None  # type: User
+reveal_type(user.id)
+reveal_type(User.name)
+[out]

sqlalchemy-plugin/sqlmypy.py

@@ -0,0 +1,58 @@
+"""Mypy style test cases for SQLAlchemy stubs and plugin."""
+
+import os
+import os.path
+import sys
+
+import pytest  # type: ignore  # no pytest in typeshed
+
+from mypy.test.config import test_temp_dir
+from mypy.test.data import DataDrivenTestCase, DataSuite
+from mypy.test.helpers import assert_string_arrays_equal
+from mypy import api
+
+this_file_dir = os.path.dirname(os.path.realpath(__file__))
+prefix = os.path.dirname(this_file_dir)
+
+# Locations of test data files such as test case descriptions (.test).
+test_data_prefix = os.path.join(prefix, 'test', 'test-data')
+
+
+class SQLDataSuite(DataSuite):
+    files = ['sqlalchemy-basics.test']
+    data_prefix = test_data_prefix
+
+    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',
+        ]
+        py2 = testcase.name.lower().endswith('python2')
+        if py2:
+            mypy_cmdline.append('--py2')
+        else:
+            mypy_cmdline.append('--python-version={}'.format('.'.join(map(str,
+                                                                          sys.version_info[:2]))))
+
+        # Write the program to a file.
+        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"))
+            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))