django/tests/model_options/models/tablespaces.py

54 lines
1.8 KiB
Python

from django.db import models
# Since the test database doesn't have tablespaces, it's impossible for Django
# to create the tables for models where db_tablespace is set. To avoid this
# problem, we mark the models as unmanaged, and temporarily revert them to
# managed during each test. We also set them to use the same tables as the
# "reference" models to avoid errors when other tests run 'migrate'
# (proxy_models_inheritance does).
class ScientistRef(models.Model):
name = models.CharField(max_length=50)
class ArticleRef(models.Model):
title = models.CharField(max_length=50, unique=True)
code = models.CharField(max_length=50, unique=True)
authors = models.ManyToManyField(ScientistRef, related_name="articles_written_set")
reviewers = models.ManyToManyField(
ScientistRef, related_name="articles_reviewed_set"
)
class Scientist(models.Model):
name = models.CharField(max_length=50)
class Meta:
db_table = "model_options_scientistref"
db_tablespace = "tbl_tbsp"
managed = False
class Article(models.Model):
title = models.CharField(max_length=50, unique=True)
code = models.CharField(max_length=50, unique=True, db_tablespace="idx_tbsp")
authors = models.ManyToManyField(Scientist, related_name="articles_written_set")
reviewers = models.ManyToManyField(
Scientist, related_name="articles_reviewed_set", db_tablespace="idx_tbsp"
)
class Meta:
db_table = "model_options_articleref"
db_tablespace = "tbl_tbsp"
managed = False
# Also set the tables for automatically created models
Authors = Article._meta.get_field("authors").remote_field.through
Authors._meta.db_table = "model_options_articleref_authors"
Reviewers = Article._meta.get_field("reviewers").remote_field.through
Reviewers._meta.db_table = "model_options_articleref_reviewers"