From 2f2119aa1f7300072ed0e586b89db42507ffcf4e Mon Sep 17 00:00:00 2001
From: Pekka Heikura <pekkah@gmail.com>
Date: Fri, 8 Mar 2019 18:32:34 +0200
Subject: [PATCH] Improve test coverage

---
 tests/graphql.tests/EventsModel.cs   |  4 +++-
 tests/graphql.tests/ExecutorFacts.cs | 16 +++++++++++++---
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/tests/graphql.tests/EventsModel.cs b/tests/graphql.tests/EventsModel.cs
index 3d281ff6b..1c1681986 100644
--- a/tests/graphql.tests/EventsModel.cs
+++ b/tests/graphql.tests/EventsModel.cs
@@ -53,12 +53,14 @@ public ISourceBlock<object> Subscribe(CancellationToken unsubscribe)
 
         public class Success
         {
-            public Success(int id)
+            public Success(int id, Event ev)
             {
                 Id = id;
+                Event = ev;
             }
 
             public int Id { get; set; }
+            public Event Event { get; }
         }
 
         public class Failure
diff --git a/tests/graphql.tests/ExecutorFacts.cs b/tests/graphql.tests/ExecutorFacts.cs
index 9a15b42be..6cc139356 100644
--- a/tests/graphql.tests/ExecutorFacts.cs
+++ b/tests/graphql.tests/ExecutorFacts.cs
@@ -1,4 +1,5 @@
 using System.Collections.Generic;
+using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
 using System.Threading.Tasks.Dataflow;
@@ -22,6 +23,7 @@ enum EventType {
 
                 type Success {
                     id: ID!
+                    event: Event
                 }
 
                 type Failure {
@@ -68,7 +70,8 @@ public ExecutorFacts()
                 {
                     "Success", new FieldResolverMap
                     {
-                        {"id", Resolve.PropertyOf<EventsModel.Success>(m => m.Id)}
+                        {"id", Resolve.PropertyOf<EventsModel.Success>(m => m.Id)},
+                        {"event", Resolve.PropertyOf<EventsModel.Success>(m => m.Event)}
                     }
                 },
                 {
@@ -112,10 +115,11 @@ public ExecutorFacts()
                                         new EventsModel.Failure("Payload should be given"));
 
                                 var id = await Model.AddAsync(newEvent);
+                                var ev = Model.Events.Single(e => e.Id == id);
 
                                 return Resolve.As(
                                     context.Schema.GetNamedType<ObjectType>("Success"),
-                                    new EventsModel.Success(id));
+                                    new EventsModel.Success(id, ev));
                             }
                         }
                     }
@@ -175,6 +179,9 @@ public async Task Mutation1()
                         __typename
                         ...on Success {
                             id
+                            event {
+                                payload
+                            }
                         }
                         
                         ...on Failure {
@@ -200,7 +207,10 @@ ...on Failure {
                   ""data"": {
                     ""create"": {
                       ""__typename"": ""Success"",
-                      ""id"": ""1""
+                      ""id"": ""1"",
+                      ""event"": {
+                        ""payload"": ""payload""
+                      }
                     }
                   }
                 }");