enable throttling for anrok

app/graphql/types/invoices/object.rb

@@ -19,6 +19,7 @@ class Object < Types::BaseObject
       field :payment_dispute_lost_at, GraphQL::Types::ISO8601DateTime
       field :payment_status, Types::Invoices::PaymentStatusTypeEnum, null: false
       field :status, Types::Invoices::StatusTypeEnum, null: false
+      field :tax_status, Types::Invoices::TaxStatusTypeEnum, null: true
       field :voidable, Boolean, null: false, method: :voidable?
 
       field :currency, Types::CurrencyEnum

app/graphql/types/invoices/tax_status_type_enum.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Types
+  module Invoices
+    class TaxStatusTypeEnum < Types::BaseEnum
+      graphql_name 'InvoiceTaxStatusTypeEnum'
+
+      Invoice::TAX_STATUSES.keys.each do |type|
+        value type
+      end
+    end
+  end
+end

app/jobs/fees/create_pay_in_advance_job.rb

@@ -2,8 +2,12 @@
 
 module Fees
   class CreatePayInAdvanceJob < ApplicationJob
+    include ConcurrencyThrottlable
+
     queue_as :default
 
+    retry_on BaseService::ThrottlingError, wait: :polynomially_longer, attempts: 25
+
     unique :until_executed, on_conflict: :log
 
     def perform(charge:, event:, billing_at: nil)

app/jobs/invoices/create_pay_in_advance_charge_job.rb

@@ -2,6 +2,8 @@
 
 module Invoices
   class CreatePayInAdvanceChargeJob < ApplicationJob
+    include ConcurrencyThrottlable
+
     queue_as do
       if ActiveModel::Type::Boolean.new.cast(ENV['SIDEKIQ_BILLING'])
         :billing
@@ -11,6 +13,7 @@ class CreatePayInAdvanceChargeJob < ApplicationJob
     end
 
     retry_on Sequenced::SequenceError
+    retry_on BaseService::ThrottlingError, wait: :polynomially_longer, attempts: 25
 
     unique :until_executed, on_conflict: :log
 

app/jobs/invoices/provider_taxes/pull_taxes_and_apply_job.rb

@@ -3,8 +3,12 @@
 module Invoices
   module ProviderTaxes
     class PullTaxesAndApplyJob < ApplicationJob
+      include ConcurrencyThrottlable
+
       queue_as 'integrations'
 
+      retry_on BaseService::ThrottlingError, wait: :polynomially_longer, attempts: 25
+
       def perform(invoice:)
         Invoices::ProviderTaxes::PullTaxesAndApplyService.call(invoice:)
       end

app/services/integrations/aggregator/taxes/invoices/create_draft_service.rb

@@ -13,6 +13,8 @@ def call
             return result unless integration
             return result unless integration.type == 'Integrations::AnrokIntegration'
 
+            throttle!(:anrok)
+
             response = http_client.post_with_response(payload, headers)
             body = JSON.parse(response.body)
 

app/services/integrations/aggregator/taxes/invoices/create_service.rb

@@ -13,6 +13,8 @@ def call
             return result unless integration
             return result unless integration.type == 'Integrations::AnrokIntegration'
 
+            throttle!(:anrok)
+
             response = http_client.post_with_response(payload, headers)
             body = JSON.parse(response.body)
 

spec/graphql/mutations/invoices/finalize_spec.rb

@@ -15,6 +15,7 @@
         finalizeInvoice(input: $input) {
           id
           status
+          taxStatus
         }
       }
     GQL
@@ -44,4 +45,35 @@
       end
     end
   end
+
+  context 'with tax provider' do
+    let(:integration) { create(:anrok_integration, organization:) }
+    let(:integration_customer) { create(:anrok_customer, integration:, customer:) }
+
+    before do
+      integration_customer
+    end
+
+    it 'returns pending invoice' do
+      freeze_time do
+        result = execute_graphql(
+          current_user: membership.user,
+          current_organization: organization,
+          permissions: required_permission,
+          query: mutation,
+          variables: {
+            input: {id: invoice.id}
+          }
+        )
+
+        result_data = result['data']['finalizeInvoice']
+
+        aggregate_failures do
+          expect(result_data['id']).to be_present
+          expect(result_data['status']).to eq('pending')
+          expect(result_data['taxStatus']).to eq('pending')
+        end
+      end
+    end
+  end
 end