From 59b44e4f736344bbc0ed2d4f54236bcbc1707385 Mon Sep 17 00:00:00 2001
From: Michael Hunger <github@jexp.de>
Date: Mon, 11 Nov 2013 01:13:27 +0100
Subject: [PATCH] fix DataGraphNamespaceHandlerTests

---
 .../data/neo4j/config/DataGraphNamespaceHandlerTests.java   | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/spring-data-neo4j/src/test/java/org/springframework/data/neo4j/config/DataGraphNamespaceHandlerTests.java b/spring-data-neo4j/src/test/java/org/springframework/data/neo4j/config/DataGraphNamespaceHandlerTests.java
index 62b9cf1e1..a46a00d54 100644
--- a/spring-data-neo4j/src/test/java/org/springframework/data/neo4j/config/DataGraphNamespaceHandlerTests.java
+++ b/spring-data-neo4j/src/test/java/org/springframework/data/neo4j/config/DataGraphNamespaceHandlerTests.java
@@ -104,7 +104,11 @@ public class DataGraphNamespaceHandlerTests {
         Config config = assertInjected("-entities");
         Collection<Neo4jPersistentEntityImpl<?>> entities = config.mappingContext.getPersistentEntities();
         assertTrue(entities.size() > 0);
-        assertEquals(TestEntity.class, entities.iterator().next().getType());
+        boolean found = false;
+        for (Neo4jPersistentEntityImpl<?> entity : entities) {
+            found |= TestEntity.class.equals(entity.getType());
+        }
+        assertEquals(true, found);
     }
     @Test
     public void injectionForConversionService() {