graphene-django/graphene_django/tests/test_registry.py

from pytest import raises

from ..registry import Registry, get_global_registry, reset_global_registry
from ..types import DjangoObjectType
from .models import Reporter as ReporterModel


def setup_function(function):
  reset_global_registry()


def test_registry_basic():
    global_registry = get_global_registry()
    
    class Reporter(DjangoObjectType):
        '''Reporter description'''
        class Meta:
            model = ReporterModel

    assert Reporter._meta.registry == global_registry
    assert global_registry.get_type_for_model(ReporterModel) == Reporter


def test_registry_multiple_types():
    class Reporter(DjangoObjectType):
        '''Reporter description'''
        class Meta:
            model = ReporterModel

    with raises(Exception) as exc_info:
        class Reporter2(DjangoObjectType):
            '''Reporter2 description'''
            class Meta:
                model = ReporterModel

    assert str(exc_info.value) == (
      'Django Model "tests.Reporter" already associated with <class \'graphene_django.tests.test_registry.Reporter\'>. '
      'You can use a different registry for <class \'graphene_django.tests.test_registry.Reporter2\'> '
      'or skip the global Registry with "Reporter2.Meta.skip_global_registry = True".'
    )


def test_registry_multiple_types_dont_collision_if_skip_global_registry():
    class Reporter(DjangoObjectType):
        '''Reporter description'''
        class Meta:
            model = ReporterModel

    class Reporter2(DjangoObjectType):
        '''Reporter2 description'''
        class Meta:
            model = ReporterModel
            skip_global_registry = True

    assert Reporter._meta.registry != Reporter2._meta.registry
    assert Reporter2._meta.registry != get_global_registry()