File: test_abstracttype.py

package info (click to toggle)
python-graphene 2.1.9-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 1,024 kB
  • sloc: python: 7,295; makefile: 196; sh: 4
file content (37 lines) | stat: -rw-r--r-- 1,031 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
from .. import abstracttype
from ..abstracttype import AbstractType
from ..field import Field
from ..objecttype import ObjectType
from ..unmountedtype import UnmountedType


class MyType(ObjectType):
    pass


class MyScalar(UnmountedType):
    def get_type(self):
        return MyType


def test_abstract_objecttype_warn_deprecation(mocker):
    mocker.patch.object(abstracttype, "warn_deprecation")

    class MyAbstractType(AbstractType):
        field1 = MyScalar()

    assert abstracttype.warn_deprecation.called


def test_generate_objecttype_inherit_abstracttype():
    class MyAbstractType(AbstractType):
        field1 = MyScalar()

    class MyObjectType(ObjectType, MyAbstractType):
        field2 = MyScalar()

    assert MyObjectType._meta.description is None
    assert MyObjectType._meta.interfaces == ()
    assert MyObjectType._meta.name == "MyObjectType"
    assert list(MyObjectType._meta.fields.keys()) == ["field1", "field2"]
    assert list(map(type, MyObjectType._meta.fields.values())) == [Field, Field]