import { describe, expect, test } from "bun:test"
import {
  getAgentToolRestrictions,
  hasAgentToolRestrictions,
} from "./agent-tool-restrictions"

describe("agent-tool-restrictions", () => {
  test("athena restrictions include call_omo_agent", () => {
    //#given
    //#when
    const restrictions = getAgentToolRestrictions("athena")
    //#then
    expect(restrictions.write).toBe(false)
    expect(restrictions.edit).toBe(false)
    expect(restrictions.call_omo_agent).toBe(false)
  })

  test("council-member restrictions include call_omo_agent", () => {
    //#given
    //#when
    const restrictions = getAgentToolRestrictions("council-member")
    //#then
    expect(restrictions.call_omo_agent).toBe(false)
  })

  test("hasAgentToolRestrictions returns true for athena", () => {
    //#given
    //#when
    const result = hasAgentToolRestrictions("athena")
    //#then
    expect(result).toBe(true)
  })
})