Mixins.spec.ts 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. import { describe, it, expect } from "vitest";
  2. import { Mixins } from "./Mixins.js";
  3. describe("Mixins", () => {
  4. it("Base resolution", () => {
  5. const mixins = new Mixins<{
  6. foo: (v: number, ctx: { a: number }) => number;
  7. }>({ autoIncrementPriority: true });
  8. mixins.add("foo", (v) => v ^ 0b0001); // 1 (prio 0)
  9. mixins.add("foo", (v) => v ^ 0b1000); // 2 (prio 1)
  10. mixins.add("foo", (v, c) => v ^ c.a); // 3 (prio 2)
  11. // input: 0b1100
  12. // 1: 0b1100 ^ 0b0001 = 0b1101
  13. // 2: 0b1101 ^ 0b1000 = 0b0101
  14. // 3: 0b0101 ^ 0b0100 = 0b0001
  15. // result: 0b0001 = 1
  16. expect(mixins.resolve("foo", 0b1100, { a: 0b0100 })).toBe(1);
  17. });
  18. it("Priority resolution", () => {
  19. const mixins = new Mixins<{
  20. foo: (v: number) => number;
  21. }>();
  22. mixins.add("foo", (v) => v / 2, 1); // 2 (prio 1)
  23. mixins.add("foo", (v) => Math.round(Math.log(v) * 10), -1); // 4 (prio -1)
  24. mixins.add("foo", (v) => Math.pow(v, 2)); // 3 (prio 0)
  25. mixins.add("foo", (v) => Math.sqrt(v), Number.MAX_SAFE_INTEGER); // 1 (prio max)
  26. // input: 100
  27. // 1: sqrt(100) = 10
  28. // 2: 10 / 2 = 5
  29. // 3: 5 ^ 2 = 25
  30. // 4: round(log(25) * 10) = round(32.188758248682006) = 32
  31. // result: 3
  32. expect(mixins.resolve("foo", 100)).toBe(32);
  33. });
  34. it("Sync/async resolution & cleanup", async () => {
  35. const acAll = new AbortController();
  36. const mixins = new Mixins<{
  37. foo: (v: number) => Promise<number>;
  38. }>({
  39. defaultSignal: acAll.signal,
  40. });
  41. const ac1 = new AbortController();
  42. mixins.add("foo", (v) => Math.sqrt(v), { signal: ac1.signal }); // 1 (prio 0, index 0)
  43. mixins.add("foo", (v) => Math.pow(v, 4)); // 2 (prio 0, index 1)
  44. const rem3 = mixins.add("foo", async (v) => { // 3 (prio 0, index 2)
  45. await new Promise((r) => setTimeout(r, 50));
  46. return v + 2;
  47. });
  48. const res1 = mixins.resolve("foo", 100);
  49. expect(res1).toBeInstanceOf(Promise);
  50. expect(await res1).toBe(10002);
  51. rem3();
  52. const res2 = mixins.resolve("foo", 100);
  53. expect(res2).not.toBeInstanceOf(Promise);
  54. expect(res2).toBe(10000);
  55. ac1.abort();
  56. const res3 = mixins.resolve("foo", 100);
  57. expect(res3).not.toBeInstanceOf(Promise);
  58. expect(res3).toBe(100000000);
  59. acAll.abort();
  60. const res4 = mixins.resolve("foo", 100);
  61. expect(res4).not.toBeInstanceOf(Promise);
  62. expect(res4).toBe(100);
  63. });
  64. });