systems can be queries themselves

develop/testing/system_as_query_tests.lua

@@ -0,0 +1,128 @@
+local evo = require 'evolved'
+
+do
+    local f1, f2, f3 = evo.id(3)
+
+    local q1e3 = evo.builder():include(f1):exclude(f3):spawn()
+    local q2e3 = evo.builder():include(f2):exclude(f3):spawn()
+
+    local e1 = evo.builder():set(f1, 1):spawn()
+    local e12 = evo.builder():set(f1, 11):set(f2, 12):spawn()
+    local e2 = evo.builder():set(f2, 2):spawn()
+    local e23 = evo.builder():set(f2, 23):set(f3, 3):spawn()
+
+    local c1 = evo.chunk(f1)
+    local c12 = evo.chunk(f1, f2)
+    local c2 = evo.chunk(f2)
+
+    do
+        local _, entity_list, entity_count = evo.chunk(f2, f3)
+        assert(entity_count == 1 and entity_list[1] == e23)
+    end
+
+    do
+        local entity_sum = 0
+
+        local s = evo.builder()
+            :query(q1e3)
+            :execute(function(chunk, entity_list, entity_count)
+                for i = 1, entity_count do
+                    entity_sum = entity_sum + entity_list[i]
+                end
+
+                if chunk == c1 then
+                    assert(entity_count == 1)
+                    assert(entity_list[1] == e1)
+                elseif chunk == c12 then
+                    assert(entity_count == 1)
+                    assert(entity_list[1] == e12)
+                else
+                    assert(false, "Unexpected chunk: " .. tostring(chunk))
+                end
+            end):spawn()
+
+        evo.process(s)
+
+        assert(entity_sum == e1 + e12)
+    end
+
+    do
+        local entity_sum = 0
+
+        local s = evo.builder()
+            :query(q2e3)
+            :execute(function(chunk, entity_list, entity_count)
+                for i = 1, entity_count do
+                    entity_sum = entity_sum + entity_list[i]
+                end
+
+                if chunk == c12 then
+                    assert(entity_count == 1)
+                    assert(entity_list[1] == e12)
+                elseif chunk == c2 then
+                    assert(entity_count == 1)
+                    assert(entity_list[1] == e2)
+                else
+                    assert(false, "Unexpected chunk: " .. tostring(chunk))
+                end
+            end):spawn()
+
+        evo.process(s)
+
+        assert(entity_sum == e12 + e2)
+    end
+
+    do
+        local entity_sum = 0
+
+        local s = evo.builder()
+            :include(f1)
+            :exclude(f3)
+            :execute(function(chunk, entity_list, entity_count)
+                for i = 1, entity_count do
+                    entity_sum = entity_sum + entity_list[i]
+                end
+
+                if chunk == c1 then
+                    assert(entity_count == 1)
+                    assert(entity_list[1] == e1)
+                elseif chunk == c12 then
+                    assert(entity_count == 1)
+                    assert(entity_list[1] == e12)
+                else
+                    assert(false, "Unexpected chunk: " .. tostring(chunk))
+                end
+            end):spawn()
+
+        evo.process(s)
+
+        assert(entity_sum == e1 + e12)
+    end
+
+    do
+        local entity_sum = 0
+
+        local s = evo.builder()
+            :include(f2)
+            :exclude(f3)
+            :execute(function(chunk, entity_list, entity_count)
+                for i = 1, entity_count do
+                    entity_sum = entity_sum + entity_list[i]
+                end
+
+                if chunk == c12 then
+                    assert(entity_count == 1)
+                    assert(entity_list[1] == e12)
+                elseif chunk == c2 then
+                    assert(entity_count == 1)
+                    assert(entity_list[1] == e2)
+                else
+                    assert(false, "Unexpected chunk: " .. tostring(chunk))
+                end
+            end):spawn()
+
+        evo.process(s)
+
+        assert(entity_sum == e12 + e2)
+    end
+end