// Reimplementations of the following tests from Flecs:
// https://github.com/SanderMertens/flecs/blob/master/test/api/src/Entity.c

const std = @import("std");
const alloc = std.testing.allocator;
const expect = @import("../src/expect.zig");

const util = @import("./util.zig");
const FlecsError = @import("../src/error.zig").FlecsError;
const flecs = @import("../src/main.zig");
const c = flecs.c;

const context = flecs.Context(void);
const Entity = context.Entity;
const Path = context.Path;
const World = context.World;

test "Entity_init_id" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const e = try world.entity(.{}, .{});
    try expect.null(c.ecs_get_type(world.raw, e.raw));
}

test "Entity_init_id_name" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const e = try world.entity(.{ .name = "foo" }, .{});
    try expect.equal("foo", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("foo", "{}", .{path});
}

test "Entity_init_id_path" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const e_parts = Path.buildParts(.{ "parent", "child" });
    const e_path = Path.fromParts(false, &e_parts);
    const e = try world.entity(.{ .path = e_path }, .{});
    try expect.equal("child", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("parent.child", "{}", .{path});
}

test "Entity_init_id_add_1_comp" {
    const TagA = struct {};

    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    _ = try world.tag(TagA);

    const e = try world.entity(.{}, .{TagA});
    try expect.true(e.has(TagA));
}

test "Entity_init_id_add_2_comp" {
    const TagA = struct {};
    const TagB = struct {};

    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    _ = try world.tag(TagA);
    _ = try world.tag(TagB);

    const e = try world.entity(.{}, .{ TagA, TagB });
    try expect.true(e.has(TagA));
    try expect.true(e.has(TagB));
}

test "Entity_init_id_w_scope" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const scope = try world.entity(.{}, .{});

    _ = world.setScope(scope);
    try expect.equal(scope, world.getScope());

    const e = try world.entity(.{}, .{});

    try expect.true(e.has(.{ c.EcsChildOf, scope }));
}

test "Entity_init_id_name_w_scope" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const scope = try world.entity(.{ .name = "parent" }, .{});
    try expect.equal("parent", scope.getName());

    _ = world.setScope(scope);
    try expect.equal(scope, world.getScope());

    const e = try world.entity(.{ .name = "child" }, .{});

    try expect.true(e.has(.{ c.EcsChildOf, scope }));
    try expect.equal("child", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("parent.child", "{}", .{path});
}

test "Entity_init_id_path_w_scope" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const scope = try world.entity(.{ .name = "parent" }, .{});
    try expect.equal("parent", scope.getName());

    _ = world.setScope(scope);
    try expect.equal(scope, world.getScope());

    const e_parts = Path.buildParts(.{ "child", "grandchild" });
    const e_path = Path.fromParts(false, &e_parts);
    const e = try world.entity(.{ .path = e_path }, .{});

    try expect.equal("grandchild", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("parent.child.grandchild", "{}", .{path});
}

test "Entity_init_id_fullpath_w_scope" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const scope = try world.entity(.{ .name = "parent" }, .{});
    try expect.equal("parent", scope.getName());

    _ = world.setScope(scope);
    try expect.equal(scope, world.getScope());

    const p = try Path.fromString("::parent.child.grandchild", .{ .root_sep = "::", .sep = "." }, flecs.allocator);
    defer p.deinit();
    const e = try world.entity(.{ .path = p }, .{});

    try expect.equal("grandchild", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("parent.child.grandchild", "{}", .{path});
}

test "Entity_init_id_fullpath_w_scope_existing" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const scope = try world.entity(.{ .name = "parent" }, .{});
    try expect.equal("parent", scope.getName());

    _ = world.setScope(scope);
    try expect.equal(scope, world.getScope());

    const p = try Path.fromString("::parent.child.grandchild", .{ .root_sep = "::", .sep = "." }, flecs.allocator);
    defer p.deinit();

    const e = try world.entity(.{ .path = p }, .{});
    const r = try world.entity(.{ .path = p }, .{});
    try expect.equal(e, r);

    try expect.equal("grandchild", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("parent.child.grandchild", "{}", .{path});
}

test "Entity_init_id_name_1_comp" {
    const TagA = struct {};

    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    _ = try world.tag(TagA);

    const e = try world.entity(.{ .name = "foo" }, .{TagA});
    try expect.true(e.has(TagA));
    try expect.equal("foo", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("foo", "{}", .{path});
}

test "Entity_init_id_name_2_comp" {
    const TagA = struct {};
    const TagB = struct {};

    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    _ = try world.tag(TagA);
    _ = try world.tag(TagB);

    const e = try world.entity(.{ .name = "foo" }, .{ TagA, TagB });
    try expect.true(e.has(TagA));
    try expect.equal("foo", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("foo", "{}", .{path});
}

test "Entity_init_id_name_2_comp_w_scope" {
    const TagA = struct {};
    const TagB = struct {};

    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    _ = try world.tag(TagA);
    _ = try world.tag(TagB);

    const scope = try world.entity(.{ .name = "parent" }, .{});
    try expect.equal("parent", scope.getName());

    _ = world.setScope(scope);
    try expect.equal(scope, world.getScope());

    const e = try world.entity(.{ .name = "child" }, .{ TagA, TagB });
    try expect.true(e.has(TagA));
    try expect.equal("child", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("parent.child", "{}", .{path});
}

test "Entity_id_add_1_comp" {
    const TagA = struct {};

    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    _ = try world.tag(TagA);

    const e = try world.entity(.{}, .{});
    const r = try world.entity(.{ .id = e }, .{TagA});
    try expect.equal(e, r);
    try expect.true(e.has(TagA));
}

test "Entity_id_add_2_comp" {
    const TagA = struct {};
    const TagB = struct {};

    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    _ = try world.tag(TagA);
    _ = try world.tag(TagB);

    const e = try world.entity(.{}, .{});
    const r = try world.entity(.{ .id = e }, .{ TagA, TagB });
    try expect.equal(e, r);
    try expect.true(e.has(TagA));
    try expect.true(e.has(TagB));
}

test "Entity_init_id_path_w_sep" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const p = try Path.fromString("parent::child", .{ .root_sep = null, .sep = "::" }, flecs.allocator);
    defer p.deinit();
    const e = try world.entity(.{ .path = p }, .{});
    try expect.equal("child", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("parent.child", "{}", .{path});
}

test "Entity_find_id_name" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const e = try world.entity(.{ .name = "foo" }, .{});
    try expect.equal("foo", e.getName());

    const r = try world.entity(.{ .name = "foo" }, .{});
    try expect.equal(e, r);
}

test "Entity_find_w_existing_id_name" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const id = try world.entity(.{}, .{});

    const e = try world.entity(.{ .id = id, .name = "foo" }, .{});
    try expect.equal(e, id);
    try expect.equal("foo", e.getName());

    const r = try world.entity(.{ .name = "foo" }, .{});
    try expect.equal(e, r);
}

test "Entity_find_id_name_w_scope" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const scope = try world.entity(.{ .name = "parent" }, .{});
    try expect.equal("parent", scope.getName());

    _ = world.setScope(scope);
    try expect.equal(scope, world.getScope());

    const e = try world.entity(.{ .name = "child" }, .{});
    try expect.equal("child", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("parent.child", "{}", .{path});

    const r = try world.entity(.{ .name = "child" }, .{});
    try expect.equal(e, r);
}

test "Entity_find_id_path" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const e_parts = Path.buildParts(.{ "parent", "child" });
    const e_path = Path.fromParts(false, &e_parts);
    const e = try world.entity(.{ .path = e_path }, .{});
    try expect.equal("child", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("parent.child", "{}", .{path});

    const r = try world.entity(.{ .path = e_path }, .{});
    try expect.equal(e, r);
}

test "Entity_find_id_path_w_scope" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const scope = try world.entity(.{ .name = "parent" }, .{});
    try expect.equal("parent", scope.getName());

    _ = world.setScope(scope);
    try expect.equal(scope, world.getScope());

    const e_parts = Path.buildParts(.{ "child", "grandchild" });
    const e_path = Path.fromParts(false, &e_parts);
    const e = try world.entity(.{ .path = e_path }, .{});
    try expect.equal("grandchild", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("parent.child.grandchild", "{}", .{path});

    const r = try world.entity(.{ .path = e_path }, .{});
    try expect.equal(e, r);
}

test "Entity_find_id_name_match" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const e = try world.entity(.{ .name = "foo" }, .{});
    try expect.equal("foo", e.getName());

    const r = try world.entity(.{ .id = e, .name = "foo" }, .{});
    try expect.equal(e, r);
}

test "Entity_find_id_name_match_w_scope" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const scope = try world.entity(.{ .name = "parent" }, .{});
    try expect.equal("parent", scope.getName());

    _ = world.setScope(scope);
    try expect.equal(scope, world.getScope());

    const e = try world.entity(.{ .name = "child" }, .{});
    try expect.equal("child", e.getName());

    const path = try e.getPath(flecs.allocator);
    defer path.deinit();
    try expect.fmt("parent.child", "{}", .{path});

    const r1 = try world.entity(.{ .id = e, .name = "child" }, .{});
    try expect.equal(e, r1);

    _ = world.setScope(null);

    const r2_parts = Path.buildParts(.{ "parent", "child" });
    const r2_path = Path.fromParts(false, &r2_parts);
    const r2 = try world.entity(.{ .id = e, .path = r2_path }, .{});
    try expect.equal(e, r2);
}

test "Entity_find_id_path_match" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const e_parts = Path.buildParts(.{ "parent", "child" });
    const e_path = Path.fromParts(false, &e_parts);
    const e = try world.entity(.{ .path = e_path }, .{});
    try expect.equal("child", e.getName());

    const r = try world.entity(.{ .id = e, .path = e_path }, .{});
    try expect.equal(e, r);
}

test "Entity_find_id_path_match_w_scope" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const scope = try world.entity(.{ .name = "parent" }, .{});
    try expect.equal("parent", scope.getName());

    _ = world.setScope(scope);
    try expect.equal(scope, world.getScope());

    const e_parts = Path.buildParts(.{ "child", "grandchild" });
    const e_path = Path.fromParts(false, &e_parts);
    const e = try world.entity(.{ .path = e_path }, .{});
    try expect.equal("grandchild", e.getName());

    const r1 = try world.entity(.{ .id = e, .path = e_path }, .{});
    try expect.equal(e, r1);

    _ = world.setScope(null);

    const r2_parts = Path.buildParts(.{ "parent", "child", "grandchild" });
    const r2_path = Path.fromParts(false, &r2_parts);
    const r2 = try world.entity(.{ .id = e, .path = r2_path }, .{});
    try expect.equal(e, r2);
}

test "Entity_find_id_name_mismatch" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const e = try world.entity(.{ .name = "foo" }, .{});
    try expect.equal("foo", e.getName());

    _ = try world.entity(.{ .name = "bar" }, .{});

    const prev_log_level = c.ecs_log_set_level(-4);
    defer _ = c.ecs_log_set_level(prev_log_level);

    const r = world.entity(.{ .id = e, .name = "bar" }, .{});
    try expect.err(FlecsError.Unknown, r);
}

test "Entity_find_id_name_mismatch_w_scope" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const scope = try world.entity(.{ .name = "parent" }, .{});
    try expect.equal("parent", scope.getName());

    _ = world.setScope(scope);
    try expect.equal(scope, world.getScope());

    const e = try world.entity(.{ .name = "child" }, .{});
    try expect.equal("child", e.getName());

    const prev_log_level = c.ecs_log_set_level(-4);
    defer _ = c.ecs_log_set_level(prev_log_level);

    const r = world.entity(.{ .id = e, .name = "parent" }, .{});
    try expect.err(FlecsError.Unknown, r);
}

test "Entity_find_id_path_mismatch" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const e_parts = Path.buildParts(.{ "parent", "child" });
    const e_path = Path.fromParts(false, &e_parts);
    const e = try world.entity(.{ .path = e_path }, .{});
    try expect.equal("child", e.getName());

    const prev_log_level = c.ecs_log_set_level(-4);
    defer _ = c.ecs_log_set_level(prev_log_level);

    const r_parts = Path.buildParts(.{ "parent", "foo" });
    const r_path = Path.fromParts(false, &r_parts);
    const r = world.entity(.{ .id = e, .path = r_path }, .{});
    try expect.err(FlecsError.Unknown, r);
}

test "Entity_find_id_path_mismatch_w_scope" {
    flecs.init(alloc);
    var world = try World.initMinimal();
    defer world.deinit();

    const scope = try world.entity(.{ .name = "parent" }, .{});
    try expect.equal("parent", scope.getName());

    _ = world.setScope(scope);
    try expect.equal(scope, world.getScope());

    const e_parts = Path.buildParts(.{ "child", "grandchild" });
    const e_path = Path.fromParts(false, &e_parts);
    const e = try world.entity(.{ .path = e_path }, .{});
    try expect.equal("grandchild", e.getName());

    const unnamed_parts = Path.buildParts(.{ "child", "foo" });
    const unnamed_path = Path.fromParts(false, &unnamed_parts);
    _ = try world.entity(.{ .path = unnamed_path }, .{});

    const prev_log_level = c.ecs_log_set_level(-4);
    defer _ = c.ecs_log_set_level(prev_log_level);

    const r = world.entity(.{ .id = e, .path = unnamed_path }, .{});
    try expect.err(FlecsError.Unknown, r);
}