mirror of
https://github.com/Ukendio/jecs.git
synced 2025-04-25 09:30:03 +00:00
88 lines
2 KiB
Lua
88 lines
2 KiB
Lua
|
local testkit = require("../testkit")
|
||
|
|
local jecs = require("../lib/init")
|
||
|
|
|
||
|
|
local TEST, CASE, CHECK, FINISH, SKIP = testkit.test()
|
||
|
|
|
||
|
|
local N = 10
|
||
|
|
|
||
|
|
TEST("world:query", function()
|
||
|
|
|
||
|
|
do CASE "should query all matching entities"
|
||
|
|
|
||
|
|
local world = jecs.World.new()
|
||
|
|
local A = world:component()
|
||
|
|
local B = world:component()
|
||
|
|
|
||
|
|
local entities = {}
|
||
|
|
for i = 1, N do
|
||
|
|
local id = world:entity()
|
||
|
|
|
||
|
|
|
||
|
|
world:set(id, A, true)
|
||
|
|
if i > 5 then world:set(id, B, true) end
|
||
|
|
entities[i] = id
|
||
|
|
end
|
||
|
|
|
||
|
|
for id in world:query(A) do
|
||
|
|
table.remove(entities, CHECK(table.find(entities, id)))
|
||
|
|
end
|
||
|
|
|
||
|
|
CHECK(#entities == 0)
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
do CASE "should query all matching entities when irrelevant component is removed"
|
||
|
|
|
||
|
|
local world = jecs.World.new()
|
||
|
|
local A = world:component()
|
||
|
|
local B = world:component()
|
||
|
|
|
||
|
|
local entities = {}
|
||
|
|
for i = 1, N do
|
||
|
|
local id = world:entity()
|
||
|
|
|
||
|
|
world:set(id, A, true)
|
||
|
|
world:set(id, B, true)
|
||
|
|
if i > 5 then world:remove(id, B, true) end
|
||
|
|
entities[i] = id
|
||
|
|
end
|
||
|
|
|
||
|
|
local added = 0
|
||
|
|
for id in world:query(A) do
|
||
|
|
added += 1
|
||
|
|
table.remove(entities, CHECK(table.find(entities, id)))
|
||
|
|
end
|
||
|
|
|
||
|
|
CHECK(added == N)
|
||
|
|
end
|
||
|
|
|
||
|
|
do CASE "should query all entities without B"
|
||
|
|
|
||
|
|
local world = jecs.World.new()
|
||
|
|
local A = world:component()
|
||
|
|
local B = world:component()
|
||
|
|
|
||
|
|
local entities = {}
|
||
|
|
for i = 1, N do
|
||
|
|
local id = world:entity()
|
||
|
|
|
||
|
|
world:set(id, A, true)
|
||
|
|
if i < 5 then
|
||
|
|
entities[i] = id
|
||
|
|
else
|
||
|
|
world:set(id, B, true)
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
for id in world:query(A):without(B) do
|
||
|
|
table.remove(entities, CHECK(table.find(entities, id)))
|
||
|
|
end
|
||
|
|
|
||
|
|
CHECK(#entities == 0)
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
end)
|
||
|
|
|
||
|
|
FINISH()
|