From 0aeee949d309eef699b4a65c8f00eb447592c29f Mon Sep 17 00:00:00 2001
From: Christian Treffs <ctreffs@gmail.com>
Date: Thu, 15 Oct 2020 22:11:21 +0200
Subject: [PATCH] Add EntityIterator test

---
 Tests/FirebladeECSTests/NexusTests.swift      | 11 +++++++++++
 Tests/FirebladeECSTests/XCTestManifests.swift |  3 ++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/Tests/FirebladeECSTests/NexusTests.swift b/Tests/FirebladeECSTests/NexusTests.swift
index 714db97..3468e40 100644
--- a/Tests/FirebladeECSTests/NexusTests.swift
+++ b/Tests/FirebladeECSTests/NexusTests.swift
@@ -155,4 +155,15 @@ class NexusTests: XCTestCase {
         XCTAssert(pB.x != pA.x)
         XCTAssert(pB.y != pA.y)
     }
+
+    func testEntityIteration() {
+        nexus.createEntities(count: 1000) { ctx in Position(x: ctx.index, y: ctx.index) }
+
+        let entityArray = [Entity](nexus.makeEntitiesIterator()).lazy
+
+        XCTAssertEqual(entityArray.count, 1000)
+
+        XCTAssertTrue(entityArray.contains(where: { $0.identifier.index == 0 }))
+        XCTAssertTrue(entityArray.contains(where: { $0.identifier.index == 999 }))
+    }
 }
diff --git a/Tests/FirebladeECSTests/XCTestManifests.swift b/Tests/FirebladeECSTests/XCTestManifests.swift
index c9b2465..97a839f 100644
--- a/Tests/FirebladeECSTests/XCTestManifests.swift
+++ b/Tests/FirebladeECSTests/XCTestManifests.swift
@@ -342,7 +342,8 @@ extension NexusTests {
         ("testComponentRetrieval", testComponentRetrieval),
         ("testComponentUniqueness", testComponentUniqueness),
         ("testEntityCreate", testEntityCreate),
-        ("testEntityDestroy", testEntityDestroy)
+        ("testEntityDestroy", testEntityDestroy),
+        ("testEntityIteration", testEntityIteration)
     ]
 }