diff options
Diffstat (limited to 'platform/testFramework/src/com/intellij/testFramework/UsefulTestCase.java')
-rw-r--r-- | platform/testFramework/src/com/intellij/testFramework/UsefulTestCase.java | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/platform/testFramework/src/com/intellij/testFramework/UsefulTestCase.java b/platform/testFramework/src/com/intellij/testFramework/UsefulTestCase.java index fa752b4643b0..6deda3d4b70a 100644 --- a/platform/testFramework/src/com/intellij/testFramework/UsefulTestCase.java +++ b/platform/testFramework/src/com/intellij/testFramework/UsefulTestCase.java @@ -570,8 +570,12 @@ public abstract class UsefulTestCase extends TestCase { public static <T> T assertOneElement(Collection<T> collection) { Assert.assertNotNull(collection); - Assert.assertEquals(toString(collection), 1, collection.size()); - return collection.iterator().next(); + Iterator<T> iterator = collection.iterator(); + String toString = toString(collection); + Assert.assertTrue(toString, iterator.hasNext()); + T t = iterator.next(); + Assert.assertFalse(toString, iterator.hasNext()); + return t; } public static <T> T assertOneElement(T[] ts) { |