mirror of
https://github.com/davidhalter/django-stubs.git
synced 2025-12-06 20:24:31 +08:00
104 lines
3.3 KiB
Python
104 lines
3.3 KiB
Python
from test.pytest_plugin import reveal_type, output
|
|
from test.pytest_tests.base import BaseDjangoPluginTestCase
|
|
|
|
|
|
class TestForeignKey(BaseDjangoPluginTestCase):
|
|
def test_foreign_key_field(self):
|
|
from django.db import models
|
|
|
|
class Publisher(models.Model):
|
|
pass
|
|
|
|
class Book(models.Model):
|
|
publisher = models.ForeignKey(to=Publisher, on_delete=models.CASCADE,
|
|
related_name='books')
|
|
|
|
book = Book()
|
|
reveal_type(book.publisher) # E: Revealed type is 'main.Publisher*'
|
|
|
|
publisher = Publisher()
|
|
reveal_type(publisher.books) # E: Revealed type is 'django.db.models.query.QuerySet[main.Book]'
|
|
|
|
def test_every_foreign_key_creates_field_name_with_appended_id(self):
|
|
from django.db import models
|
|
|
|
class Publisher(models.Model):
|
|
pass
|
|
|
|
class Book(models.Model):
|
|
publisher = models.ForeignKey(to=Publisher, on_delete=models.CASCADE,
|
|
related_name='books')
|
|
|
|
book = Book()
|
|
reveal_type(book.publisher_id) # E: Revealed type is 'builtins.int'
|
|
|
|
def test_foreign_key_different_order_of_params(self):
|
|
from django.db import models
|
|
|
|
class Publisher(models.Model):
|
|
pass
|
|
|
|
class Book(models.Model):
|
|
publisher = models.ForeignKey(on_delete=models.CASCADE, to=Publisher,
|
|
related_name='books')
|
|
|
|
book = Book()
|
|
reveal_type(book.publisher) # E: Revealed type is 'main.Publisher*'
|
|
|
|
publisher = Publisher()
|
|
reveal_type(publisher.books) # E: Revealed type is 'django.db.models.query.QuerySet[main.Book]'
|
|
|
|
@output(
|
|
"""
|
|
main:4: warning: to= parameter type Instance is not supported
|
|
"""
|
|
)
|
|
def test_to_parameter_as_string_fallbacks_to_any(self):
|
|
from django.db import models
|
|
|
|
class Book(models.Model):
|
|
publisher = models.ForeignKey(to='Publisher', on_delete=models.CASCADE)
|
|
|
|
book = Book()
|
|
reveal_type(book.publisher) # E: Revealed type is 'Any'
|
|
|
|
|
|
class TestOneToOneField(BaseDjangoPluginTestCase):
|
|
def test_onetoone_field(self):
|
|
from django.db import models
|
|
|
|
class User(models.Model):
|
|
pass
|
|
|
|
class Profile(models.Model):
|
|
user = models.OneToOneField(to=User, on_delete=models.CASCADE, related_name='profile')
|
|
|
|
profile = Profile()
|
|
reveal_type(profile.user) # E: Revealed type is 'main.User*'
|
|
|
|
user = User()
|
|
reveal_type(user.profile) # E: Revealed type is 'main.Profile'
|
|
|
|
def test_onetoone_field_with_underscore_id(self):
|
|
from django.db import models
|
|
|
|
class User(models.Model):
|
|
pass
|
|
|
|
class Profile(models.Model):
|
|
user = models.OneToOneField(to=User, on_delete=models.CASCADE, related_name='profile')
|
|
|
|
profile = Profile()
|
|
reveal_type(profile.user_id) # E: Revealed type is 'builtins.int'
|
|
|
|
def test_parameter_to_keyword_may_be_absent(self):
|
|
from django.db import models
|
|
|
|
class User(models.Model):
|
|
pass
|
|
|
|
class Profile(models.Model):
|
|
user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='profile')
|
|
|
|
reveal_type(User().profile) # E: Revealed type is 'main.Profile'
|