discourse-ai/spec/lib/agents/sql_helper_spec.rb

# frozen_string_literal: true

RSpec.describe DiscourseAi::Agents::SqlHelper do
  let :sql_helper do
    subject
  end

  it "renders schema" do
    prompt = sql_helper.system_prompt
    expect(prompt).to include("posts(")
    expect(prompt).to include("topics(")
    expect(prompt).not_to include("translation_key") # not a priority table
    expect(prompt).to include("user_api_keys") # not a priority table

    expect(sql_helper.tools).to eq([DiscourseAi::Agents::Tools::DbSchema])
  end
end