Correctly resolve Enum

examples/starwars_newenum/data.py

@@ -0,0 +1,97 @@
+human_data = {}
+droid_data = {}
+
+
+def setup():
+    from .schema import Human, Droid, Episode
+
+    global human_data, droid_data
+    luke = Human(
+        id="1000",
+        name="Luke Skywalker",
+        friends=["1002", "1003", "2000", "2001"],
+        appears_in=[Episode.NEWHOPE, Episode.EMPIRE, Episode.JEDI],
+        home_planet="Tatooine",
+    )
+
+    vader = Human(
+        id="1001",
+        name="Darth Vader",
+        friends=["1004"],
+        appears_in=[Episode.NEWHOPE, Episode.EMPIRE, Episode.JEDI],
+        home_planet="Tatooine",
+    )
+
+    han = Human(
+        id="1002",
+        name="Han Solo",
+        friends=["1000", "1003", "2001"],
+        appears_in=[Episode.NEWHOPE, Episode.EMPIRE, Episode.JEDI],
+        home_planet=None,
+    )
+
+    leia = Human(
+        id="1003",
+        name="Leia Organa",
+        friends=["1000", "1002", "2000", "2001"],
+        appears_in=[Episode.NEWHOPE, Episode.EMPIRE, Episode.JEDI],
+        home_planet="Alderaan",
+    )
+
+    tarkin = Human(
+        id="1004",
+        name="Wilhuff Tarkin",
+        friends=["1001"],
+        appears_in=[Episode.NEWHOPE],
+        home_planet=None,
+    )
+
+    human_data = {
+        "1000": luke,
+        "1001": vader,
+        "1002": han,
+        "1003": leia,
+        "1004": tarkin,
+    }
+
+    c3po = Droid(
+        id="2000",
+        name="C-3PO",
+        friends=["1000", "1002", "1003", "2001"],
+        appears_in=[Episode.NEWHOPE, Episode.EMPIRE, Episode.JEDI],
+        primary_function="Protocol",
+    )
+
+    r2d2 = Droid(
+        id="2001",
+        name="R2-D2",
+        friends=["1000", "1002", "1003"],
+        appears_in=[Episode.NEWHOPE, Episode.EMPIRE, Episode.JEDI],
+        primary_function="Astromech",
+    )
+
+    droid_data = {"2000": c3po, "2001": r2d2}
+
+
+def get_character(id):
+    return human_data.get(id) or droid_data.get(id)
+
+
+def get_friends(character):
+    return map(get_character, character.friends)
+
+
+def get_hero(episode):
+    from .schema import Episode
+
+    if episode == Episode.EMPIRE:
+        return human_data["1000"]
+    return droid_data["2001"]
+
+
+def get_human(id):
+    return human_data.get(id)
+
+
+def get_droid(id):
+    return droid_data.get(id)

examples/starwars_newenum/schema.py

@@ -0,0 +1,57 @@
+from enum import Enum
+
+import graphene
+
+from .data import get_character, get_droid, get_hero, get_human
+
+
+class Episode(Enum):
+    NEWHOPE = 4
+    EMPIRE = 5
+    JEDI = 6
+
+
+GrapheneEpisode = graphene.Enum.from_enum(Episode, legacy_enum_resolver=False)
+
+
+class Character(graphene.Interface):
+    id = graphene.ID()
+    name = graphene.String()
+    friends = graphene.List(lambda: Character)
+    appears_in = graphene.List(GrapheneEpisode)
+
+    def resolve_friends(self, info):
+        # The character friends is a list of strings
+        return [get_character(f) for f in self.friends]
+
+
+class Human(graphene.ObjectType):
+    class Meta:
+        interfaces = (Character,)
+
+    home_planet = graphene.String()
+
+
+class Droid(graphene.ObjectType):
+    class Meta:
+        interfaces = (Character,)
+
+    primary_function = graphene.String()
+
+
+class Query(graphene.ObjectType):
+    hero = graphene.Field(Character, episode=GrapheneEpisode())
+    human = graphene.Field(Human, id=graphene.String())
+    droid = graphene.Field(Droid, id=graphene.String())
+
+    def resolve_hero(self, info, episode=None):
+        return get_hero(episode)
+
+    def resolve_human(self, info, id):
+        return get_human(id)
+
+    def resolve_droid(self, info, id):
+        return get_droid(id)
+
+
+schema = graphene.Schema(query=Query)

examples/starwars_newenum/tests/snapshots/snap_test_query.py

@@ -0,0 +1,100 @@
+# -*- coding: utf-8 -*-
+# snapshottest: v1 - https://goo.gl/zC4yUc
+from __future__ import unicode_literals
+
+from snapshottest import Snapshot
+
+snapshots = Snapshot()
+
+snapshots["test_hero_name_query 1"] = {"data": {"hero": {"name": "R2-D2"}}}
+
+snapshots["test_hero_name_and_friends_query 1"] = {
+    "data": {
+        "hero": {
+            "id": "2001",
+            "name": "R2-D2",
+            "friends": [
+                {"name": "Luke Skywalker"},
+                {"name": "Han Solo"},
+                {"name": "Leia Organa"},
+            ],
+        }
+    }
+}
+
+snapshots["test_nested_query 1"] = {
+    "data": {
+        "hero": {
+            "name": "R2-D2",
+            "friends": [
+                {
+                    "name": "Luke Skywalker",
+                    "appearsIn": ["NEWHOPE", "EMPIRE", "JEDI"],
+                    "friends": [
+                        {"name": "Han Solo"},
+                        {"name": "Leia Organa"},
+                        {"name": "C-3PO"},
+                        {"name": "R2-D2"},
+                    ],
+                },
+                {
+                    "name": "Han Solo",
+                    "appearsIn": ["NEWHOPE", "EMPIRE", "JEDI"],
+                    "friends": [
+                        {"name": "Luke Skywalker"},
+                        {"name": "Leia Organa"},
+                        {"name": "R2-D2"},
+                    ],
+                },
+                {
+                    "name": "Leia Organa",
+                    "appearsIn": ["NEWHOPE", "EMPIRE", "JEDI"],
+                    "friends": [
+                        {"name": "Luke Skywalker"},
+                        {"name": "Han Solo"},
+                        {"name": "C-3PO"},
+                        {"name": "R2-D2"},
+                    ],
+                },
+            ],
+        }
+    }
+}
+
+snapshots["test_fetch_luke_query 1"] = {"data": {"human": {"name": "Luke Skywalker"}}}
+
+snapshots["test_fetch_some_id_query 1"] = {
+    "data": {"human": {"name": "Luke Skywalker"}}
+}
+
+snapshots["test_fetch_some_id_query2 1"] = {"data": {"human": {"name": "Han Solo"}}}
+
+snapshots["test_invalid_id_query 1"] = {"data": {"human": None}}
+
+snapshots["test_fetch_luke_aliased 1"] = {"data": {"luke": {"name": "Luke Skywalker"}}}
+
+snapshots["test_fetch_luke_and_leia_aliased 1"] = {
+    "data": {"luke": {"name": "Luke Skywalker"}, "leia": {"name": "Leia Organa"}}
+}
+
+snapshots["test_duplicate_fields 1"] = {
+    "data": {
+        "luke": {"name": "Luke Skywalker", "homePlanet": "Tatooine"},
+        "leia": {"name": "Leia Organa", "homePlanet": "Alderaan"},
+    }
+}
+
+snapshots["test_use_fragment 1"] = {
+    "data": {
+        "luke": {"name": "Luke Skywalker", "homePlanet": "Tatooine"},
+        "leia": {"name": "Leia Organa", "homePlanet": "Alderaan"},
+    }
+}
+
+snapshots["test_check_type_of_r2 1"] = {
+    "data": {"hero": {"__typename": "Droid", "name": "R2-D2"}}
+}
+
+snapshots["test_check_type_of_luke 1"] = {
+    "data": {"hero": {"__typename": "Human", "name": "Luke Skywalker"}}
+}