From ec0bfcfcb0824e54f05fb06f0b9fe03c82c5d669 Mon Sep 17 00:00:00 2001 From: Shivam Mishra Date: Tue, 21 Jan 2025 14:29:50 +0530 Subject: [PATCH] test: specs for overriding captain limits --- spec/enterprise/models/account_spec.rb | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/spec/enterprise/models/account_spec.rb b/spec/enterprise/models/account_spec.rb index 3667c7095..b063084c1 100644 --- a/spec/enterprise/models/account_spec.rb +++ b/spec/enterprise/models/account_spec.rb @@ -120,7 +120,7 @@ RSpec.describe Account, type: :model do end end - describe 'captain limits are not configured' do + describe 'when captain limits are not configured' do it 'returns default values' do account.custom_attributes = { 'plan_name': 'unknown' } expect(account.captain_monthly_limit).to eq( @@ -129,6 +129,19 @@ RSpec.describe Account, type: :model do end end + describe 'when limits are configured for an account' do + before do + create(:installation_config, name: 'CAPTAIN_CLOUD_PLAN_LIMITS', value: captain_limits.to_json) + account.update(custom_attributes: { captain_document_limit: 5555, captain_response_limit: 9999 }) + end + + it 'returns limits based on custom attributes' do + usage_limits = account.usage_limits + expect(usage_limits[:captain][:documents][:total_count]).to eq(5555) + expect(usage_limits[:captain][:generated_responses][:total_count]).to eq(9999) + end + end + describe 'audit logs' do it 'returns audit logs' do # checking whether associated_audits method is present