Add singleElement() to GraphQlTester.EntityList

singleElement() is a utility method to check for lists of size one,
and return the Entity in the same statement.

Signed-off-by: Simone Conte <conte@adobe.com>

Author: sic2

spring-graphql-test/src/main/java/org/springframework/graphql/test/tester/DefaultGraphQlTester.java

@@ -655,6 +655,18 @@ public EntityList<E> hasSizeGreaterThan(int size) {
 						getEntity().size() > size));
 				return this;
 			}
+
+			@Override
+			public Entity<E, ?> singleElement() {
+				this.hasSize(1);
+
+				E element = this.get().get(0);
+				@SuppressWarnings("unchecked")
+				Class<E> elementClass = (Class<E>) element.getClass();
+
+				return new DefaultPath(DefaultPath.this.basePath, DefaultPath.this.path + "[0]", DefaultPath.this.delegate)
+						.entity(elementClass);
+			}
 		}
 
 

spring-graphql-test/src/main/java/org/springframework/graphql/test/tester/GraphQlTester.java

@@ -470,6 +470,13 @@ interface EntityList<E> extends Entity<List<E>, EntityList<E>> {
 		 */
 		EntityList<E> hasSizeGreaterThan(int size);
 
+		/**
+		 * Verify the list has a single element and return an {@code Entity} spec for it.
+		 * <p>This is a convenience method that combines {@link #hasSize(int) hasSize(1)}
+		 * with navigating to the first element in the list.
+		 * @return an {@code Entity} spec for the single element that allows further assertions
+		 */
+		Entity<E, ?> singleElement();
 	}
 
 	/**

spring-graphql-test/src/test/java/org/springframework/graphql/test/tester/GraphQlTesterTests.java

@@ -213,6 +213,35 @@ void entityList() {
 				.containsExactly(han, leia);
 
 		assertThat(getActualRequestDocument()).contains(document);
+
+		assertThatThrownBy(() -> entityList.singleElement())
+				.as("Should have exactly one element")
+				.hasMessage("Expecting list " +
+							"[MovieCharacter[name='Han Solo'], MovieCharacter[name='Leia Organa']] " +
+							"at path 'me.friends' to have size == 1\n" +
+							"Request: document='{me {name, friends}}'");
+	}
+
+	@Test
+	void entityListWithOneElement() {
+
+		String document = "{me {name, friends}}";
+		getGraphQlService().setDataAsJson(document,
+				"{" +
+				"  \"me\":{" +
+				"      \"name\":\"Luke Skywalker\","
+				+ "    \"friends\":[{\"name\":\"Han Solo\"}]" +
+				"  }" +
+				"}");
+
+		GraphQlTester.Response response = graphQlTester().document(document).execute();
+
+		MovieCharacter han = MovieCharacter.create("Han Solo");
+
+		response.path("me.friends")
+				.entityList(MovieCharacter.class)
+				.singleElement()
+				.isEqualTo(han);
 	}
 
 	@Test