opencode/packages/core/test/plugin/provider-databricks.test.ts

import { describe, expect } from "bun:test"
import { Effect } from "effect"
import { Plugin } from "@opencode/core/plugin"
import { PluginHost } from "@opencode/core/plugin/host"
import { DatabricksPlugin } from "@opencode/core/plugin/provider/databricks"
import { Provider } from "@opencode/core/provider"
import { testEffect } from "../lib/effect"
import { PluginTestLayer } from "./fixture"

const it = testEffect(PluginTestLayer)
const providerID = Provider.ID.make("databricks")

const baseURLFor = Effect.fn(function* (host: string) {
  const previous = process.env.DATABRICKS_HOST
  yield* Effect.addFinalizer(() =>
    Effect.sync(() => {
      if (previous === undefined) delete process.env.DATABRICKS_HOST
      if (previous !== undefined) process.env.DATABRICKS_HOST = previous
    }),
  )
  process.env.DATABRICKS_HOST = host
  const catalog = yield* Provider.Service
  yield* catalog.transform((catalog) => {
    catalog.update(providerID, (item) => {
      item.settings = { baseURL: "https://${DATABRICKS_HOST}/ai-gateway/mlflow/v1" }
    })
  })
  const plugin = yield* Plugin.Service
  yield* DatabricksPlugin.effect(yield* PluginHost.make(plugin))
  return (yield* catalog.get(providerID))?.settings?.baseURL
})

describe("DatabricksPlugin", () => {
  it.effect("fills the base URL from a bare host", () =>
    Effect.gen(function* () {
      expect(yield* baseURLFor("adb-123.azuredatabricks.net")).toBe(
        "https://adb-123.azuredatabricks.net/ai-gateway/mlflow/v1",
      )
    }),
  )

  it.effect("fills the base URL from a host that includes the scheme", () =>
    Effect.gen(function* () {
      expect(yield* baseURLFor("https://adb-123.azuredatabricks.net/")).toBe(
        "https://adb-123.azuredatabricks.net/ai-gateway/mlflow/v1",
      )
    }),
  )
})