add ForeignKey

test/test-data/check-model-relations.test

@@ -0,0 +1,14 @@
+[case testForeignKeyWithClass]
+from django.db import models
+
+class User(models.Model):
+    pass
+
+class Profile(models.Model):
+    user = models.ForeignKey(to=User, on_delete=models.CASCADE)
+
+profile = Profile()
+reveal_type(profile.user)  # E: Revealed type is 'main.User*'
+[out]
+
+

test/testdjango.py

@@ -15,7 +15,8 @@ MYPY_INI_PATH = ROOT_DIR / 'test' / 'plugins.ini'
 class DjangoTestSuite(DataSuite):
     files = [
         'check-model-fields.test',
-        'check-postgres-fields.test'
+        'check-postgres-fields.test',
+        'check-model-relations.test'
     ]
     data_prefix = str(TEST_DATA_DIR)