Spawn-Kill-Loot Flow

import com.frotty27.hrtk.api.annotation.*;
import com.frotty27.hrtk.api.assert_.*;
import com.frotty27.hrtk.api.context.WorldTestContext;
import com.frotty27.hrtk.api.lifecycle.IsolationStrategy;

import java.util.List;

@HytaleSuite(value = "Spawn-Kill-Loot Flow", isolation = IsolationStrategy.DEDICATED_WORLD)
@Tag("flow")
public class SpawnKillLootFlow {

    @FlowTest(timeoutTicks = 300)
    @DisplayName("Kweebec spawn -> verify alive -> kill -> verify loot")
    void fullKweebecFlow(WorldTestContext ctx) {

        // ── Step 1: Spawn ──
        ctx.log("Spawning kweebec at (0, 64, 0)...");
        Object entity = ctx.spawnEntity("Kweebec_Sapling", 0, 64, 0);
        HytaleAssert.assertNotNull("Entity ref must not be null", entity);

        // ── Step 2: Wait for initialization ──
        ctx.waitTicks(2);
        HytaleAssert.assertTrue("Entity must exist after spawn",
                                 ctx.entityExists(entity));

        // ── Step 3: Verify alive and stats ──
        Object store = ctx.getStore();
        StatsAssert.assertAlive(store, entity);
        StatsAssert.assertHealthAtMax(store, entity);
        ctx.log("Entity is alive with full health.");

        // ── Step 4: Apply lethal damage ──
        ctx.log("Applying lethal damage...");
        applyLethalDamage(ctx, entity);

        // ── Step 5: Wait for death ──
        Boolean died = ctx.awaitCondition(
            () -> isEntityDead(store, entity) ? true : null,
            100,
            "Entity should die within 100 ticks"
        );
        HytaleAssert.assertNotNull("Entity should have died", died);
        StatsAssert.assertDead(store, entity);
        ctx.log("Entity confirmed dead.");

        // ── Step 6: Verify loot ──
        List<?> drops = ctx.awaitCondition(
            () -> collectNearbyDrops(ctx, 0, 64, 0),
            60,
            "Loot should drop within 60 ticks after death"
        );
        LootAssert.assertDropsContain(drops, "kweebec_hide");
        LootAssert.assertDropCountBetween(drops, 1, 5);
        ctx.log("Loot verified: %d drop stacks found.", drops.size());
    }

    // ── Helper methods ──

    private void applyLethalDamage(WorldTestContext ctx, Object entity) {
        // Your game logic for applying damage
        // This would use your mod's damage system or direct stat manipulation
    }

    private boolean isEntityDead(Object store, Object entity) {
        try {
            StatsAssert.assertDead(store, entity);
            return true;
        } catch (Exception e) {
            return false;
        }
    }

    private List<?> collectNearbyDrops(WorldTestContext ctx, int x, int y, int z) {
        // Your logic to find dropped item entities near the given position
        // Return null if no drops found yet (awaitCondition polls until non-null)
        return null; // placeholder
    }
}

@ParameterizedTest
@ValueSource(strings = {"Kweebec_Sapling", "Trork_Warrior", "Outlander_Hunter"})
@FlowTest(timeoutTicks = 300)
void flowForCreatureType(String creatureId, WorldTestContext ctx) {
    Object entity = ctx.spawnEntity(creatureId, 0, 64, 0);
    ctx.waitTicks(2);

    StatsAssert.assertAlive(ctx.getStore(), entity);

    applyLethalDamage(ctx, entity);
    ctx.awaitCondition(
        () -> isEntityDead(ctx.getStore(), entity) ? true : null,
        100, creatureId + " should die"
    );

    ctx.log("%s flow completed successfully.", creatureId);
}