def test_models():
expected = {
"django.contrib.auth": ("Group", "Permission", "User"),
"django.contrib.contenttypes": ("ContentType", ),
"django.contrib.sessions": ("Session", ),
"django.contrib.sites": ("Site", ),
"tests.app_a": ("InterAppSubclass", ),
"tests.app_b": ("InterAppForeignKey", ),
"tests.app_c": ("InterAppOneToOne", ),
"tests.app_d": ("InterAppManyToMany", "InterAppProxy"),
"tests.basic": (
"OutgoingForeignKey",
"OutgoingManyToMany",
"OutgoingOneToOne",
"SelfReference",
"Target",
),
"tests.generic": ("GenericFK", ),
"tests.inheritance": (
"AbstractMultipleInheritance",
"Concrete",
"ConcreteBase",
"ConcreteInheritance",
"ConcreteSubclass1",
"ConcreteSubclass2",
"MixedMultipleInheritance",
"SubSubclass",
"Subclass",
),
"tests.installed": ("ConcreteInstalled", ),
"tests.proxy": ("ProxyNode", "ProxyNode2", "Target"),
}
assert get_schema().models == expected
def test_proxy():
expected = [
(("tests.app_d", "InterAppProxy"), ("tests.app_c",
"InterAppOneToOne")),
(("tests.proxy", "ProxyNode"), ("tests.proxy", "Target")),
(("tests.proxy", "ProxyNode2"), ("tests.proxy", "Target")),
]
assert get_schema().proxies == expected
def test_one_to_one():
expected = [
(("tests.app_c", "InterAppOneToOne"), ("tests.app_b",
"InterAppForeignKey")),
(("tests.basic", "OutgoingOneToOne"), ("tests.basic", "Target")),
(
("tests.inheritance", "ConcreteSubclass2"),
("tests.inheritance", "ConcreteBase"),
),
]
assert get_schema().one_to_ones == expected
def test_context():
"""The schema should be available in the template context."""
request = create_request()
view = Schema(request=request)
context = view.get_context_data()
schema = get_schema()
assert context["models"] == json.dumps(schema.models)
assert context["foreign_keys"] == json.dumps(schema.foreign_keys)
assert context["many_to_manys"] == json.dumps(schema.many_to_manys)
assert context["one_to_ones"] == json.dumps(schema.one_to_ones)
assert context["inheritance"] == json.dumps(schema.inheritance)
assert context["proxies"] == json.dumps(schema.proxies)
def test_many_to_many():
expected = [
(("django.contrib.auth", "Group"), ("django.contrib.auth",
"Permission")),
(("django.contrib.auth", "User"), ("django.contrib.auth", "Group")),
(("django.contrib.auth", "User"), ("django.contrib.auth",
"Permission")),
(("tests.app_d", "InterAppManyToMany"), ("tests.app_b",
"InterAppForeignKey")),
(("tests.basic", "OutgoingManyToMany"), ("tests.basic", "Target")),
]
assert get_schema().many_to_manys == expected
def get_context_data(self, **kwargs):
schema = get_schema()
kwargs.update({
"abstract_models": json.dumps(schema.abstract_models),
"models": json.dumps(schema.models),
"foreign_keys": json.dumps(schema.foreign_keys),
"many_to_manys": json.dumps(schema.many_to_manys),
"one_to_ones": json.dumps(schema.one_to_ones),
"inheritance": json.dumps(schema.inheritance),
"proxies": json.dumps(schema.proxies),
})
return super(Schema, self).get_context_data(**kwargs)
def test_abstract_models():
expected = {
"django.contrib.auth":
("AbstractBaseUser", "AbstractUser", "PermissionsMixin"),
"django.contrib.sessions": ("AbstractBaseSession", ),
"tests.inheritance": (
"Abstract",
"AbstractBase",
"AbstractSubclass1",
"AbstractSubclass2",
),
"tests.not_installed.models": ("AbstractNotInstalled", ),
}
if DJANGO_LT_19:
expected.pop("django.contrib.sessions")
assert get_schema().abstract_models == expected
def test_foreign_key():
expected = [
(
("django.contrib.auth", "Permission"),
("django.contrib.contenttypes", "ContentType"),
),
(("tests.app_b", "InterAppForeignKey"), ("django.contrib.auth",
"User")),
(("tests.basic", "OutgoingForeignKey"), ("tests.basic", "Target")),
(("tests.basic", "SelfReference"), ("tests.basic", "SelfReference")),
(
("tests.generic", "GenericFK"),
("django.contrib.contenttypes", "ContentType"),
),
]
assert get_schema().foreign_keys == expected
def test_inheritance():
expected = [
(
("django.contrib.auth", "AbstractUser"),
("django.contrib.auth", "AbstractBaseUser"),
),
(
("django.contrib.auth", "AbstractUser"),
("django.contrib.auth", "PermissionsMixin"),
),
(("django.contrib.auth", "User"), ("django.contrib.auth",
"AbstractUser")),
(
("django.contrib.sessions", "Session"),
("django.contrib.sessions", "AbstractBaseSession"),
),
(("tests.app_a", "InterAppSubclass"), ("django.contrib.auth",
"Group")),
(
("tests.inheritance", "AbstractMultipleInheritance"),
("tests.inheritance", "AbstractSubclass1"),
),
(
("tests.inheritance", "AbstractMultipleInheritance"),
("tests.inheritance", "AbstractSubclass2"),
),
(
("tests.inheritance", "AbstractSubclass1"),
("tests.inheritance", "AbstractBase"),
),
(
("tests.inheritance", "AbstractSubclass2"),
("tests.inheritance", "AbstractBase"),
),
(("tests.inheritance", "Concrete"), ("tests.inheritance", "Abstract")),
(
("tests.inheritance", "ConcreteInheritance"),
("tests.inheritance", "ConcreteSubclass1"),
),
(
("tests.inheritance", "ConcreteInheritance"),
("tests.inheritance", "ConcreteSubclass2"),
),
(
("tests.inheritance", "ConcreteSubclass1"),
("tests.inheritance", "ConcreteBase"),
),
(
("tests.inheritance", "ConcreteSubclass2"),
("tests.inheritance", "ConcreteBase"),
),
(
("tests.inheritance", "MixedMultipleInheritance"),
("tests.inheritance", "AbstractBase"),
),
(
("tests.inheritance", "MixedMultipleInheritance"),
("tests.inheritance", "ConcreteBase"),
),
(("tests.inheritance", "SubSubclass"), ("tests.inheritance",
"Subclass")),
(("tests.inheritance", "Subclass"), ("tests.inheritance",
"ConcreteBase")),
(
("tests.installed", "ConcreteInstalled"),
("tests.not_installed.models", "AbstractNotInstalled"),
),
]
if DJANGO_LT_19:
expected.remove((
("django.contrib.sessions", "Session"),
("django.contrib.sessions", "AbstractBaseSession"),
))
assert get_schema().inheritance == expected