builtins.test.ts 3.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. import { describe, expect } from "bun:test"
  2. import { Effect, Layer } from "effect"
  3. import * as TestClock from "effect/testing/TestClock"
  4. import { AppNodeBuilder } from "@opencode-ai/core/effect/app-node-builder"
  5. import { Location } from "@opencode-ai/core/location"
  6. import { FSUtil } from "@opencode-ai/util/fs-util"
  7. import { Global } from "@opencode-ai/util/global"
  8. import { AbsolutePath } from "@opencode-ai/core/schema"
  9. import { InstructionBuiltIns } from "@opencode-ai/core/instructions/builtins"
  10. import { SessionSchema } from "@opencode-ai/core/session/schema"
  11. import { location } from "../fixture/location"
  12. import { testEffect } from "../lib/effect"
  13. import { readInitial, readUpdate } from "../lib/instructions"
  14. const directory = AbsolutePath.make(FSUtil.resolve("/repo/packages/core"))
  15. const projectDirectory = AbsolutePath.make(FSUtil.resolve("/repo"))
  16. const timestamp = Date.parse("2026-06-03T12:00:00.000Z")
  17. const sessionID = SessionSchema.ID.make("ses_builtin_test")
  18. const localDate = (time: number) => new Date(time).toDateString()
  19. const locationLayer = Layer.succeed(
  20. Location.Service,
  21. Location.Service.of(
  22. location(
  23. { directory },
  24. { projectDirectory, vcs: { type: "git", store: AbsolutePath.make(FSUtil.resolve("/repo/.git")) } },
  25. ),
  26. ),
  27. )
  28. const it = testEffect(
  29. AppNodeBuilder.build(InstructionBuiltIns.node, [
  30. [Location.node, locationLayer],
  31. [Global.node, Global.layerWith({ config: "/global" })],
  32. ]),
  33. )
  34. describe("InstructionBuiltIns", () => {
  35. it.effect("loads location-scoped environment and host-local date instructions", () =>
  36. Effect.gen(function* () {
  37. yield* TestClock.setTime(timestamp)
  38. const context = yield* InstructionBuiltIns.Service
  39. const initialized = yield* readInitial(yield* context.load(sessionID))
  40. expect(initialized.text).toBe(
  41. [
  42. "Here is some useful information about the environment you are running in:",
  43. "<env>",
  44. ` Session ID: ${sessionID}`,
  45. ` Working directory: ${directory}`,
  46. ` Workspace root folder: ${projectDirectory}`,
  47. " Is directory a git repo: yes",
  48. ` Platform: ${process.platform}`,
  49. "</env>",
  50. "",
  51. `Today's date: ${localDate(timestamp)}`,
  52. ].join("\n"),
  53. )
  54. }),
  55. )
  56. it.effect("updates the date without repeating unchanged environment instructions", () =>
  57. Effect.gen(function* () {
  58. yield* TestClock.setTime(timestamp)
  59. const context = yield* InstructionBuiltIns.Service
  60. const initialized = yield* readInitial(yield* context.load(sessionID))
  61. yield* TestClock.setTime(timestamp + 24 * 60 * 60 * 1000)
  62. const refreshed = yield* readUpdate(yield* context.load(sessionID), initialized)
  63. expect(refreshed.text).toBe(`Today's date is now: ${localDate(timestamp + 24 * 60 * 60 * 1000)}`)
  64. }),
  65. )
  66. it.effect("does not update again within the same local calendar day", () =>
  67. Effect.gen(function* () {
  68. yield* TestClock.setTime(timestamp)
  69. const context = yield* InstructionBuiltIns.Service
  70. const initialized = yield* readInitial(yield* context.load(sessionID))
  71. yield* TestClock.setTime(timestamp + 60 * 60 * 1000)
  72. expect((yield* readUpdate(yield* context.load(sessionID), initialized)).changed).toBe(false)
  73. }),
  74. )
  75. })