mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-05-06 03:39:06 +08:00
122 lines
3.6 KiB
Ruby
122 lines
3.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.describe DiscourseAi::Discover::DiscoveriesController do
|
|
fab!(:user)
|
|
|
|
before do
|
|
enable_current_plugin
|
|
sign_in(user)
|
|
SiteSetting.ai_discover_enabled = true
|
|
end
|
|
|
|
describe "#reply" do
|
|
fab!(:group)
|
|
fab!(:ai_agent) { Fabricate(:ai_agent, allowed_group_ids: [group.id], default_llm_id: 1) }
|
|
|
|
before { SiteSetting.ai_discover_agent = ai_agent.id }
|
|
|
|
context "when the user doesn't have access to the agent" do
|
|
it "returns a 403" do
|
|
post "/discourse-ai/discoveries/reply", params: { query: "What is Discourse?" }
|
|
|
|
expect(response.status).to eq(403)
|
|
end
|
|
end
|
|
|
|
context "when the user is allowed to use discover" do
|
|
before do
|
|
SiteSetting.ai_discover_agent = ai_agent.id
|
|
group.add(user)
|
|
end
|
|
|
|
it "returns a 200 and queues a job to reply" do
|
|
expect {
|
|
post "/discourse-ai/discoveries/reply", params: { query: "What is Discourse?" }
|
|
}.to change(Jobs::StreamDiscoverReply.jobs, :size).by(1)
|
|
|
|
expect(response.status).to eq(200)
|
|
end
|
|
|
|
it "returns a 400 if the query is missing" do
|
|
post "/discourse-ai/discoveries/reply"
|
|
|
|
expect(response.status).to eq(400)
|
|
end
|
|
end
|
|
end
|
|
|
|
describe "#continue_convo" do
|
|
fab!(:group)
|
|
fab!(:llm_model)
|
|
fab!(:ai_agent) do
|
|
agent = Fabricate(:ai_agent, allowed_group_ids: [group.id], default_llm_id: llm_model.id)
|
|
agent.create_user!
|
|
agent
|
|
end
|
|
let(:query) { "What is Discourse?" }
|
|
let(:context) { "Discourse is an open-source discussion platform." }
|
|
|
|
context "when the user is allowed to discover" do
|
|
before do
|
|
SiteSetting.ai_discover_agent = ai_agent.id
|
|
group.add(user)
|
|
end
|
|
|
|
it "returns a 200 and creates a private message topic" do
|
|
expect {
|
|
post "/discourse-ai/discoveries/continue-convo",
|
|
params: {
|
|
query: query,
|
|
context: context,
|
|
}
|
|
}.to change(Topic, :count).by(1)
|
|
|
|
expect(response.status).to eq(200)
|
|
expect(response.parsed_body["topic_id"]).to be_present
|
|
end
|
|
|
|
it "returns invalid parameters if the context is missing" do
|
|
post "/discourse-ai/discoveries/continue-convo", params: { query: query }
|
|
|
|
expect(response.status).to eq(422)
|
|
expect(response.parsed_body["errors"]).to include("context")
|
|
end
|
|
|
|
it "does not allow suspended users to create conversations" do
|
|
user.update!(suspended_till: 1.year.from_now, suspended_at: Time.zone.now)
|
|
|
|
expect {
|
|
post "/discourse-ai/discoveries/continue-convo", params: { query:, context: }
|
|
}.not_to change { Topic.count }
|
|
|
|
expect(response.status).to eq(403)
|
|
end
|
|
|
|
it "does not allow silenced users to create conversations" do
|
|
user.update!(silenced_till: 1.year.from_now)
|
|
|
|
expect {
|
|
post "/discourse-ai/discoveries/continue-convo", params: { query:, context: }
|
|
}.not_to change { Topic.count }
|
|
|
|
expect(response.status).to eq(403)
|
|
end
|
|
|
|
describe "group-based restrictions" do
|
|
fab!(:staff_group) { Group[:staff] }
|
|
|
|
before { ai_agent.update(allowed_group_ids: [staff_group.id]) }
|
|
|
|
it "forbid users without group access from creating conversations" do
|
|
expect(user.in_any_groups?([staff_group.id])).to be_falsey
|
|
|
|
expect {
|
|
post "/discourse-ai/discoveries/continue-convo", params: { query:, context: }
|
|
}.not_to change { Topic.where(user: user).count }
|
|
|
|
expect(response.status).to eq(403)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
