external-directory.test.ts 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126
  1. import { describe, expect, test } from "bun:test"
  2. import path from "path"
  3. import type { Tool } from "../../src/tool/tool"
  4. import { Instance } from "../../src/project/instance"
  5. import { assertExternalDirectory } from "../../src/tool/external-directory"
  6. import type { PermissionNext } from "../../src/permission/next"
  7. const baseCtx: Omit<Tool.Context, "ask"> = {
  8. sessionID: "test",
  9. messageID: "",
  10. callID: "",
  11. agent: "build",
  12. abort: AbortSignal.any([]),
  13. metadata: () => {},
  14. }
  15. describe("tool.assertExternalDirectory", () => {
  16. test("no-ops for empty target", async () => {
  17. const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
  18. const ctx: Tool.Context = {
  19. ...baseCtx,
  20. ask: async (req) => {
  21. requests.push(req)
  22. },
  23. }
  24. await Instance.provide({
  25. directory: "/tmp",
  26. fn: async () => {
  27. await assertExternalDirectory(ctx)
  28. },
  29. })
  30. expect(requests.length).toBe(0)
  31. })
  32. test("no-ops for paths inside Instance.directory", async () => {
  33. const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
  34. const ctx: Tool.Context = {
  35. ...baseCtx,
  36. ask: async (req) => {
  37. requests.push(req)
  38. },
  39. }
  40. await Instance.provide({
  41. directory: "/tmp/project",
  42. fn: async () => {
  43. await assertExternalDirectory(ctx, path.join("/tmp/project", "file.txt"))
  44. },
  45. })
  46. expect(requests.length).toBe(0)
  47. })
  48. test("asks with a single canonical glob", async () => {
  49. const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
  50. const ctx: Tool.Context = {
  51. ...baseCtx,
  52. ask: async (req) => {
  53. requests.push(req)
  54. },
  55. }
  56. const directory = "/tmp/project"
  57. const target = "/tmp/outside/file.txt"
  58. const expected = path.join(path.dirname(target), "*")
  59. await Instance.provide({
  60. directory,
  61. fn: async () => {
  62. await assertExternalDirectory(ctx, target)
  63. },
  64. })
  65. const req = requests.find((r) => r.permission === "external_directory")
  66. expect(req).toBeDefined()
  67. expect(req!.patterns).toEqual([expected])
  68. expect(req!.always).toEqual([expected])
  69. })
  70. test("uses target directory when kind=directory", async () => {
  71. const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
  72. const ctx: Tool.Context = {
  73. ...baseCtx,
  74. ask: async (req) => {
  75. requests.push(req)
  76. },
  77. }
  78. const directory = "/tmp/project"
  79. const target = "/tmp/outside"
  80. const expected = path.join(target, "*")
  81. await Instance.provide({
  82. directory,
  83. fn: async () => {
  84. await assertExternalDirectory(ctx, target, { kind: "directory" })
  85. },
  86. })
  87. const req = requests.find((r) => r.permission === "external_directory")
  88. expect(req).toBeDefined()
  89. expect(req!.patterns).toEqual([expected])
  90. expect(req!.always).toEqual([expected])
  91. })
  92. test("skips prompting when bypass=true", async () => {
  93. const requests: Array<Omit<PermissionNext.Request, "id" | "sessionID" | "tool">> = []
  94. const ctx: Tool.Context = {
  95. ...baseCtx,
  96. ask: async (req) => {
  97. requests.push(req)
  98. },
  99. }
  100. await Instance.provide({
  101. directory: "/tmp/project",
  102. fn: async () => {
  103. await assertExternalDirectory(ctx, "/tmp/outside/file.txt", { bypass: true })
  104. },
  105. })
  106. expect(requests.length).toBe(0)
  107. })
  108. })