magicblock-labs-bolt/tests/intermediate-level/permissioning/world.ts

import { expect } from "chai";
import {
  AddAuthority,
  RemoveAuthority,
  ApplySystem,
  ApproveSystem,
  RemoveSystem,
} from "../../../clients/bolt-sdk/lib";

export function world(framework) {
  describe("World authority", () => {
    it("Add authority", async () => {
      const addAuthority = await AddAuthority({
        authority: framework.provider.wallet.publicKey,
        newAuthority: framework.provider.wallet.publicKey,
        world: framework.worldPda,
        connection: framework.provider.connection,
      });
      await framework.provider.sendAndConfirm(addAuthority.transaction, [], {
        skipPreflight: true,
      });
      const worldAccount = await framework.worldProgram.account.world.fetch(
        framework.worldPda,
      );
      expect(
        worldAccount.authorities.some((auth) =>
          auth.equals(framework.provider.wallet.publicKey),
        ),
      );
    });

    it("Add a second authority", async () => {
      const addAuthority = await AddAuthority({
        authority: framework.provider.wallet.publicKey,
        newAuthority: framework.secondAuthority,
        world: framework.worldPda,
        connection: framework.provider.connection,
      });
      await framework.provider.sendAndConfirm(addAuthority.transaction);
      const worldAccount = await framework.worldProgram.account.world.fetch(
        framework.worldPda,
      );
      expect(
        worldAccount.authorities.some((auth) =>
          auth.equals(framework.secondAuthority),
        ),
      );
    });

    it("Remove an authority", async () => {
      const removeAuthority = await RemoveAuthority({
        authority: framework.provider.wallet.publicKey,
        authorityToDelete: framework.secondAuthority,
        world: framework.worldPda,
        connection: framework.provider.connection,
      });
      await framework.provider.sendAndConfirm(removeAuthority.transaction);
      const worldAccount = await framework.worldProgram.account.world.fetch(
        framework.worldPda,
      );
      expect(
        !worldAccount.authorities.some((auth) =>
          auth.equals(framework.secondAuthority),
        ),
      );
    });

    it("Whitelist Fly System", async () => {
      const approveSystem = await ApproveSystem({
        authority: framework.provider.wallet.publicKey,
        systemToApprove: framework.systemFly.programId,
        world: framework.worldPda,
      });

      await framework.provider.sendAndConfirm(approveSystem.transaction, [], {
        skipPreflight: true,
      });

      // Get World and check permissionless and systems
      const worldAccount = await framework.worldProgram.account.world.fetch(
        framework.worldPda,
      );
      expect(worldAccount.permissionless).to.equal(false);
      expect(worldAccount.systems.length).to.be.greaterThan(0);
    });

    it("Whitelist Apply Velocity System system", async () => {
      const approveSystem = await ApproveSystem({
        authority: framework.provider.wallet.publicKey,
        systemToApprove: framework.systemApplyVelocity.programId,
        world: framework.worldPda,
      });

      await framework.provider.sendAndConfirm(approveSystem.transaction, [], {
        skipPreflight: true,
      });

      // Get World and check permissionless and systems
      const worldAccount = await framework.worldProgram.account.world.fetch(
        framework.worldPda,
      );
      expect(worldAccount.permissionless).to.equal(false);
      expect(worldAccount.systems.length).to.be.greaterThan(0);
    });

    it("Apply Fly System on Entity 1", async () => {
      const applySystem = await ApplySystem({
        authority: framework.provider.wallet.publicKey,
        systemId: framework.systemFly.programId,
        world: framework.worldPda,
        entities: [
          {
            entity: framework.entity1Pda,
            components: [
              { componentId: framework.exampleComponentPosition.programId },
            ],
          },
        ],
      });
      await framework.provider.sendAndConfirm(applySystem.transaction);
    });

    it("Remove Fly System", async () => {
      const removeSystem = await RemoveSystem({
        authority: framework.provider.wallet.publicKey,
        systemToRemove: framework.systemFly.programId,
        world: framework.worldPda,
      });

      await framework.provider.sendAndConfirm(removeSystem.transaction, [], {
        skipPreflight: true,
      });

      // Get World and check permissionless and systems
      const worldAccount = await framework.worldProgram.account.world.fetch(
        framework.worldPda,
      );
      expect(worldAccount.permissionless).to.equal(false);
      expect(worldAccount.systems.length).to.be.greaterThan(0);
    });

    it("Apply unauthorized Fly System on Entity 1", async () => {
      const applySystem = await ApplySystem({
        authority: framework.provider.wallet.publicKey,
        systemId: framework.systemFly.programId,
        world: framework.worldPda,
        entities: [
          {
            entity: framework.entity1Pda,
            components: [
              { componentId: framework.exampleComponentPosition.programId },
            ],
          },
        ],
      });
      let invalid = false;
      try {
        await framework.provider.sendAndConfirm(applySystem.transaction);
      } catch (error) {
        expect(error.logs.join(" ")).to.contain(
          "Error Code: SystemNotApproved",
        );
        invalid = true;
      }
      expect(invalid).to.equal(true);
    });
  });
}