Update tests

Shane32 · Nov 17, 2021 · 96c9127 · 96c9127
1 parent 7f7f955
commit 96c9127
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 39 deletions.
diff --git a/src/Tests/DISchemaTypesTests.cs b/src/Tests/DISchemaTypesTests.cs
diff --git a/src/Tests/Execution/DISchemaTypes.cs b/src/Tests/Execution/DISchemaTypes.cs
@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using GraphQL;
 using GraphQL.DI;
 using GraphQL.Types;
 using Moq;
@@ -13,7 +14,7 @@ public class DISchemaTypesTests
         [Theory]
         [InlineData(typeof(int), false, typeof(IntGraphType))]
         [InlineData(typeof(int), true, typeof(IntGraphType))]
-        [InlineData(typeof(Class1), false, null)]
+        [InlineData(typeof(Class1), false, typeof(AutoObjectGraphType<Class1>))]
         [InlineData(typeof(Class1), true, typeof(AutoInputObjectGraphType<Class1>))]
         public void GetGraphTypeFromClrType(Type clrType, bool isInputType, Type graphType)
         {
@@ -31,5 +32,32 @@ public new Type GetGraphTypeFromClrType(Type clrType, bool isInputType, List<(Ty
         }
 
         private class Class1 { }
+
+        [Fact]
+        public void InputOutputTypesWork()
+        {
+            var schema = new Schema();
+            schema.Query = new ObjectGraphType();
+            var newField = new FieldType { Name = "Test", Type = typeof(GraphQLClrOutputTypeReference<Class1Model>) };
+            newField.Arguments = new QueryArguments(new QueryArgument<GraphQLClrInputTypeReference<Class2InputModel>> { Name = "arg" });
+            schema.Query.AddField(newField);
+            var schemaTypes = new DISchemaTypes(schema, new DefaultServiceProvider());
+            var class1Type = schemaTypes["Class1"].ShouldBeAssignableTo<IObjectGraphType>();
+            class1Type.Fields.Count.ShouldBe(1);
+            class1Type.Fields.Find("value").ShouldNotBeNull();
+            var class2Type = schemaTypes["Class2Input"].ShouldBeAssignableTo<IInputObjectGraphType>();
+            class2Type.Fields.Count.ShouldBe(1);
+            class2Type.Fields.Find("value").ShouldNotBeNull();
+        }
+
+        private class Class1Model
+        {
+            public int Value { get; set; }
+        }
+
+        private class Class2InputModel
+        {
+            public int Value { get; set; }
+        }
     }
 }