Run test files from stubs directory (#92)

Author: MaicoTimmerman

test/files/.keep

@@ -0,0 +1,52 @@
+from sqlalchemy import Column
+from sqlalchemy import ForeignKey
+from sqlalchemy import Integer
+from sqlalchemy import Sequence
+from sqlalchemy import String
+from sqlalchemy.orm import Mapped
+from sqlalchemy.orm import registry
+
+reg: registry = registry()
+
+
+@reg.mapped
+class InferredNames:
+    __tablename__ = "infered"
+
+    # Instantiated type
+    id = Column(Integer(), primary_key=True)
+
+    # Non-instantiated type
+    name = Column(String, nullable=False)
+
+    # SchemaItem
+    foreign: Mapped[int] = Column(ForeignKey("infered.id"))
+
+    # SchemaItem without type
+    foreign1: Mapped[int] = Column(ForeignKey("infered.id"))
+
+    # SchemaItem without type
+    foreign2: Mapped[int] = Column(Integer, ForeignKey("infered.id"))
+
+    # sequence
+    seq = Column(Integer, Sequence("my_sequence"), primary_key=True)
+
+
+@reg.mapped
+class ExplicitNames:
+    __tablename__ = "explicit"
+
+    # Instantiated type
+    id = Column("id", Integer(), primary_key=True)
+
+    # Non-instantiated type
+    name = Column("name", String, nullable=False)
+
+    # SchemaItem without type
+    foreign1: int = Column("foreign", ForeignKey("infered.id"))
+
+    # SchemaItem without type
+    foreign2: int = Column("foreign", Integer, ForeignKey("infered.id"))
+
+    # sequence
+    seq = Column(Integer, Sequence("my_sequence"), primary_key=True)

test/files/column_arguments.py

@@ -0,0 +1,85 @@
+from sqlalchemy import Column
+from sqlalchemy import ForeignKey
+from sqlalchemy import Integer
+from sqlalchemy import String
+
+Column("name", Integer, index=True)
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "None", "str"  # noqa E501
+Column(None, name="name")
+Column(Integer, name="name", index=True)
+Column("name", ForeignKey("a.id"))
+Column(ForeignKey("a.id"), type_=None, index=True)
+Column(ForeignKey("a.id"), name="name", type_=Integer())
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "str", "None"  # noqa E501
+Column("name", None)
+Column("name", index=True)
+Column(ForeignKey("a.id"), name="name", index=True)
+Column(type_=None, index=True)
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "None", "ForeignKey"  # noqa E501
+Column(None, ForeignKey("a.id"))
+Column("name")
+Column(name="name", type_=None, index=True)
+Column(ForeignKey("a.id"), name="name", type_=None)
+Column(Integer)
+Column(ForeignKey("a.id"), type_=Integer())
+Column("name", Integer, ForeignKey("a.id"), index=True)
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "str", "None", "ForeignKey", "bool  # noqa E501
+Column("name", None, ForeignKey("a.id"), index=True)
+Column(ForeignKey("a.id"), index=True)
+Column("name", Integer)
+Column(Integer, name="name")
+Column(Integer, ForeignKey("a.id"), name="name", index=True)
+Column(ForeignKey("a.id"), type_=None)
+Column(ForeignKey("a.id"), name="name")
+Column(name="name", index=True)
+Column(type_=None)
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "None", "bool"  # noqa E501
+Column(None, index=True)
+Column(name="name", type_=None)
+Column(type_=Integer(), index=True)
+Column("name", Integer, ForeignKey("a.id"))
+Column(name="name", type_=Integer(), index=True)
+Column(Integer, ForeignKey("a.id"), index=True)
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "str", "None", "ForeignKey"  # noqa E501
+Column("name", None, ForeignKey("a.id"))
+Column(index=True)
+Column("name", type_=None, index=True)
+Column("name", ForeignKey("a.id"), type_=Integer(), index=True)
+Column(ForeignKey("a.id"))
+Column(Integer, ForeignKey("a.id"))
+Column(Integer, ForeignKey("a.id"), name="name")
+Column("name", ForeignKey("a.id"), index=True)
+Column("name", type_=Integer(), index=True)
+Column(ForeignKey("a.id"), name="name", type_=Integer(), index=True)
+Column(name="name")
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "str", "None", "bool"  # noqa E501
+Column("name", None, index=True)
+Column("name", ForeignKey("a.id"), type_=None, index=True)
+Column("name", type_=Integer())
+# EXPECTED_MYPY: No overload variant of "Column" matches argument type "None"
+Column(None)
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "None", "ForeignKey", "bool"  # noqa E501
+Column(None, ForeignKey("a.id"), index=True)
+Column("name", ForeignKey("a.id"), type_=None)
+Column(type_=Integer())
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "None", "ForeignKey", "str", "bool"  # noqa E501
+Column(None, ForeignKey("a.id"), name="name", index=True)
+Column(Integer, index=True)
+Column(ForeignKey("a.id"), name="name", type_=None, index=True)
+Column(ForeignKey("a.id"), type_=Integer(), index=True)
+Column(name="name", type_=Integer())
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "None", "str", "bool"  # noqa E501
+Column(None, name="name", index=True)
+Column()
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "None", "ForeignKey", "str"  # noqa E501
+Column(None, ForeignKey("a.id"), name="name")
+Column("name", type_=None)
+Column("name", ForeignKey("a.id"), type_=Integer())
+
+
+# We are no longer able to detect there, as the kwargs must stay in place.
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "Type[Integer]", "ForeignKey", "Type[String]"  # noqa E501
+Column(Integer, ForeignKey("a.id"), type_=String)
+
+# EXPECTED_MYPY: No overload variant of "Column" matches argument types "str", "ForeignKey", "str"  # noqa E501
+Column("name", ForeignKey("a.id"), name="String")

test/files/column_arguments2.py

@@ -0,0 +1,31 @@
+from datetime import datetime
+
+from sqlalchemy import Column
+from sqlalchemy import create_engine
+from sqlalchemy import DateTime
+from sqlalchemy import Integer
+from sqlalchemy.orm import Mapped
+from sqlalchemy.orm import registry
+from sqlalchemy.orm import Session
+from sqlalchemy.sql.functions import now
+
+mapper_registry: registry = registry()
+e = create_engine("sqlite:///database.db", echo=True)
+
+
+@mapper_registry.mapped
+class A:
+    __tablename__ = "a"
+    id = Column(Integer, primary_key=True)
+    date_time: Mapped[datetime] = Column(DateTime())
+    date_time2 = Column(DateTime())
+
+
+mapper_registry.metadata.create_all(e)
+
+with Session(e) as s:
+    a = A()
+    a.date_time = now()
+    a.date_time2 = now()
+    s.add(a)
+    s.commit()

test/files/mapped_assign_expression.py

@@ -29,4 +29,4 @@ commands_pre=
 
 
 commands =
-    pytest {env:WORKERS} -v test/ext/mypy/test_mypy_plugin_py3k.py {posargs}
+    pytest {env:WORKERS} --mypy-extra-test-path={toxinidir}/test/ -v test/ext/mypy/test_mypy_plugin_py3k.py {posargs}