From 3327ca47c233937c822656211a34c58e9d42884d Mon Sep 17 00:00:00 2001
From: Jonathan Kim <jkimbo@gmail.com>
Date: Thu, 25 Jun 2020 10:57:18 +0100
Subject: [PATCH] Fix tests

---
 graphene/types/definitions.py     |  3 ++-
 graphene/types/tests/test_enum.py | 18 ++++++++++--------
 2 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/graphene/types/definitions.py b/graphene/types/definitions.py
index 37fa78f5..908cc7c8 100644
--- a/graphene/types/definitions.py
+++ b/graphene/types/definitions.py
@@ -7,6 +7,7 @@ from graphql import (
     GraphQLObjectType,
     GraphQLScalarType,
     GraphQLUnionType,
+    Undefined,
 )
 
 
@@ -49,7 +50,7 @@ class GrapheneEnumType(GrapheneGraphQLType, GraphQLEnumType):
                 try:
                     value = enum[value]
                 except KeyError:
-                    value = None
+                    return Undefined
         return super(GrapheneEnumType, self).serialize(value)
 
 
diff --git a/graphene/types/tests/test_enum.py b/graphene/types/tests/test_enum.py
index 95e712b1..57d3cfee 100644
--- a/graphene/types/tests/test_enum.py
+++ b/graphene/types/tests/test_enum.py
@@ -234,8 +234,10 @@ def test_enum_types():
     from enum import Enum as PyEnum
 
     class Color(PyEnum):
+        """Primary colors"""
+
         RED = 1
-        GREEN = 2
+        YELLOW = 2
         BLUE = 3
 
     GColor = Enum.from_enum(Color)
@@ -250,16 +252,16 @@ def test_enum_types():
 
     assert str(schema) == dedent(
         '''\
-        """An enumeration."""
-        enum Color {
-          RED
-          GREEN
-          BLUE
-        }
-
         type Query {
           color: Color!
         }
+
+        """Primary colors"""
+        enum Color {
+          RED
+          YELLOW
+          BLUE
+        }
     '''
     )