tests

Author: xfalcox

spec/lib/inference/cloudflare_workers_ai_spec.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+require "rails_helper"
+require "webmock/rspec"
+
+RSpec.describe DiscourseAi::Inference::CloudflareWorkersAi do
+  let(:account_id) { "test_account_id" }
+  let(:api_token) { "test_api_token" }
+  let(:model) { "test_model" }
+  let(:content) { "test content" }
+  let(:endpoint) do
+    "https://api.cloudflare.com/client/v4/accounts/#{account_id}/ai/run/@cf/#{model}"
+  end
+  let(:headers) do
+    {
+      "Referer" => Discourse.base_url,
+      "Content-Type" => "application/json",
+      "Authorization" => "Bearer #{api_token}",
+    }
+  end
+  let(:payload) { { text: [content] }.to_json }
+
+  subject { described_class.new(account_id, api_token, model) }
+
+  before do
+    stub_request(:post, endpoint).with(body: payload, headers: headers).to_return(
+      status: response_status,
+      body: response_body,
+    )
+  end
+
+  describe "#perform!" do
+    context "when the response status is 200" do
+      let(:response_status) { 200 }
+      let(:response_body) { { result: { data: ["embedding_result"] } }.to_json }
+
+      it "returns the embedding result" do
+        result = subject.perform!(content)
+        expect(result).to eq("embedding_result")
+      end
+    end
+
+    context "when the response status is 429" do
+      let(:response_status) { 429 }
+      let(:response_body) { "" }
+
+      it "doesn't raises a Net::HTTPBadResponse error" do
+        expect { subject.perform!(content) }.not_to raise_error
+      end
+    end
+
+    context "when the response status is not 200 or 429" do
+      let(:response_status) { 500 }
+      let(:response_body) { "Internal Server Error" }
+
+      it "raises a Net::HTTPBadResponse error" do
+        expect(Rails.logger).to receive(:warn).with(
+          "Cloudflare Workers AI Embeddings failed with status: #{response_status} body: #{response_body}",
+        )
+        expect { subject.perform!(content) }.to raise_error(Net::HTTPBadResponse)
+      end
+    end
+  end
+end