1
0
Fork 0
mirror of https://github.com/maybe-finance/maybe.git synced 2025-07-19 21:29:38 +02:00
Maybe/test/controllers/api/v1/chats_controller_test.rb

134 lines
3.5 KiB
Ruby
Raw Permalink Normal View History

# frozen_string_literal: true
require "test_helper"
class Api::V1::ChatsControllerTest < ActionDispatch::IntegrationTest
setup do
@user = users(:family_admin)
@user.update!(ai_enabled: true)
@oauth_app = Doorkeeper::Application.create!(
name: "Test API App",
redirect_uri: "https://example.com/callback",
scopes: "read write read_write"
)
@read_token = Doorkeeper::AccessToken.create!(
application: @oauth_app,
resource_owner_id: @user.id,
scopes: "read"
)
@write_token = Doorkeeper::AccessToken.create!(
application: @oauth_app,
resource_owner_id: @user.id,
scopes: "read_write"
)
@chat = chats(:one)
end
test "should require authentication" do
get "/api/v1/chats"
assert_response :unauthorized
end
test "should require AI to be enabled" do
@user.update!(ai_enabled: false)
get "/api/v1/chats", headers: bearer_auth_header(@read_token)
assert_response :forbidden
response_body = JSON.parse(response.body)
assert_equal "feature_disabled", response_body["error"]
end
test "should list chats with read scope" do
get "/api/v1/chats", headers: bearer_auth_header(@read_token)
assert_response :success
response_body = JSON.parse(response.body)
assert response_body["chats"].is_a?(Array)
assert response_body["pagination"].present?
end
test "should show chat with messages" do
get "/api/v1/chats/#{@chat.id}", headers: bearer_auth_header(@read_token)
assert_response :success
response_body = JSON.parse(response.body)
assert_equal @chat.id, response_body["id"]
assert response_body["messages"].is_a?(Array)
end
test "should create chat with write scope" do
assert_difference "Chat.count" do
post "/api/v1/chats",
params: { title: "New chat", message: "Hello AI" },
headers: bearer_auth_header(@write_token)
end
assert_response :created
response_body = JSON.parse(response.body)
assert_equal "New chat", response_body["title"]
end
test "should not create chat with read scope" do
post "/api/v1/chats",
params: { title: "New chat" },
headers: bearer_auth_header(@read_token)
assert_response :forbidden
end
test "should update chat" do
patch "/api/v1/chats/#{@chat.id}",
params: { title: "Updated title" },
headers: bearer_auth_header(@write_token)
assert_response :success
response_body = JSON.parse(response.body)
assert_equal "Updated title", response_body["title"]
end
test "should delete chat" do
assert_difference "Chat.count", -1 do
delete "/api/v1/chats/#{@chat.id}", headers: bearer_auth_header(@write_token)
end
assert_response :no_content
end
test "should not access other user's chat" do
other_user = users(:family_member)
other_user.update!(family: families(:empty))
other_chat = chats(:two)
other_chat.update!(user: other_user)
get "/api/v1/chats/#{other_chat.id}", headers: bearer_auth_header(@read_token)
assert_response :not_found
end
test "should support API key authentication" do
# Remove any existing API keys for this user
@user.api_keys.destroy_all
plain_key = ApiKey.generate_secure_key
api_key = @user.api_keys.build(
name: "Test API Key",
scopes: [ "read_write" ]
)
api_key.key = plain_key
api_key.save!
get "/api/v1/chats", headers: { "X-Api-Key" => plain_key }
assert_response :success
end
private
def bearer_auth_header(token)
{ "Authorization" => "Bearer #{token.token}" }
end
end