Fix: Added Enum to GraphQLDescriptionAttribute (#2430)

ChilliCream · Oct 14, 2020 · 9ef8f5f · 9ef8f5f
1 parent 8299a94
commit 9ef8f5f
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/src/HotChocolate/Core/src/Abstractions/GraphQLDescriptionAttribute.cs b/src/HotChocolate/Core/src/Abstractions/GraphQLDescriptionAttribute.cs
@@ -7,6 +7,7 @@ namespace HotChocolate
         | AttributeTargets.Interface
         | AttributeTargets.Property
         | AttributeTargets.Method
+        | AttributeTargets.Enum
         | AttributeTargets.Parameter
         | AttributeTargets.Field)]
     public sealed class GraphQLDescriptionAttribute

diff --git a/src/HotChocolate/Core/test/Types.Tests/Types/EnumTypeTests.cs b/src/HotChocolate/Core/test/Types.Tests/Types/EnumTypeTests.cs
@@ -32,6 +32,22 @@ public void EnumType_DynamicName()
             Assert.NotNull(type);
         }
 
+        [Fact]
+        public void EnumType_GraphQLDescriptionAttribute()
+        {
+            // act
+            var schema = Schema.Create(c =>
+            {
+                c.RegisterType(new EnumType<DescriptionTestEnum>());
+
+                c.Options.StrictValidation = false;
+            });
+
+            // assert
+            EnumType type = schema.GetType<EnumType>("DescriptionTestEnum");
+            Assert.Equal("TestDescription", type.Description);
+        }
+
         [Fact]
         public void EnumType_DynamicName_NonGeneric()
         {
@@ -590,5 +606,11 @@ protected override void Configure(IEnumTypeDescriptor<string> descriptor)
                 descriptor.Value("ABC").Name("DEF");
             }
         }
+
+        [GraphQLDescription("TestDescription")]
+        public enum DescriptionTestEnum
+        {
+            Foo, Bar
+        }
     }
 }