d5a49b1579
Introduced an “Inference Tokens” page to manage tokens for inference endpoints. Users can create and delete tokens directly from this interface. The tokens are required to send requests to into inference endpoints. Requests are billed to the project the token belongs to.
72 lines
2.2 KiB
Ruby
72 lines
2.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "spec_helper"
|
|
|
|
RSpec.describe Clover, "inference-token" do
|
|
let(:user) { create_account }
|
|
|
|
let(:project) { user.create_project_with_default_policy("project-1") }
|
|
|
|
describe "feature enabled" do
|
|
before do
|
|
project.set_ff_inference_ui(true)
|
|
login(user.email)
|
|
visit "#{project.path}/inference-token"
|
|
expect(ApiKey.all).to be_empty
|
|
click_button "Create Token"
|
|
@api_key = ApiKey.first
|
|
end
|
|
|
|
it "inference token page allows creating inference tokens" do
|
|
expect(find_by_id("flash-notice").text).to include("Created inference token with id ")
|
|
|
|
expect(ApiKey.count).to eq(1)
|
|
expect(@api_key.owner_id).to eq(project.id)
|
|
expect(@api_key.owner_table).to eq("project")
|
|
expect(@api_key.projects).to eq([project])
|
|
expect(@api_key.used_for).to eq("inference_endpoint")
|
|
expect(@api_key.is_valid).to be(true)
|
|
end
|
|
|
|
it "inference token page allows removing inference tokens" do
|
|
access_tag_ds = DB[:access_tag].where(hyper_tag_id: @api_key.id)
|
|
expect(access_tag_ds.all).not_to be_empty
|
|
|
|
btn = find(".delete-btn")
|
|
data_url = btn["data-url"]
|
|
_csrf = btn["data-csrf"]
|
|
page.driver.delete data_url, {_csrf:}
|
|
expect(page.status_code).to eq(204)
|
|
expect(ApiKey.all).to be_empty
|
|
expect(access_tag_ds.all).to be_empty
|
|
visit "#{project.path}/user/token"
|
|
expect(find_by_id("flash-notice").text).to eq("Inference token deleted successfully")
|
|
|
|
page.driver.delete data_url, {_csrf:}
|
|
expect(page.status_code).to eq(204)
|
|
visit "#{project.path}/user/token"
|
|
expect(page.html).not_to include("Inference token deleted successfully")
|
|
end
|
|
end
|
|
|
|
describe "feature disabled" do
|
|
before do
|
|
project.set_ff_inference_ui(false)
|
|
login(user.email)
|
|
visit "#{project.path}/inference-token"
|
|
end
|
|
|
|
it "inference token page is not accessible" do
|
|
expect(page.status_code).to eq(404)
|
|
end
|
|
end
|
|
|
|
describe "unauthenticated" do
|
|
it "inference token page is not accessible" do
|
|
visit "/inference-token"
|
|
|
|
expect(page.title).to eq("Ubicloud - Login")
|
|
end
|
|
end
|
|
end
|