From 30325babe923cfd4212e80ba3b1dfa8705c21dec Mon Sep 17 00:00:00 2001
From: Anna Velentsevich <annvelents@gmail.com>
Date: Mon, 30 Dec 2024 15:11:23 +0100
Subject: [PATCH] fix tests

---
 spec/services/fees/one_off_service_spec.rb                    | 4 ----
 spec/services/invoices/add_on_service_spec.rb                 | 2 +-
 spec/services/invoices/create_one_off_service_spec.rb         | 2 +-
 .../invoices/refresh_draft_and_finalize_service_spec.rb       | 2 +-
 4 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/spec/services/fees/one_off_service_spec.rb b/spec/services/fees/one_off_service_spec.rb
index d0e7fca9417e..7131607206a9 100644
--- a/spec/services/fees/one_off_service_spec.rb
+++ b/spec/services/fees/one_off_service_spec.rb
@@ -342,9 +342,5 @@
         end
       end
     end
-
-    it_behaves_like "applies invoice_custom_sections" do
-      let(:service_call) { one_off_service.call }
-    end
   end
 end
diff --git a/spec/services/invoices/add_on_service_spec.rb b/spec/services/invoices/add_on_service_spec.rb
index 4f1361b70340..4430340863df 100644
--- a/spec/services/invoices/add_on_service_spec.rb
+++ b/spec/services/invoices/add_on_service_spec.rb
@@ -108,7 +108,7 @@
     end
 
     it_behaves_like "applies invoice_custom_sections" do
-      let(:service_call) { invoice_service.call }
+      let(:service_call) { invoice_service.create }
     end
 
     context 'with customer timezone' do
diff --git a/spec/services/invoices/create_one_off_service_spec.rb b/spec/services/invoices/create_one_off_service_spec.rb
index 21b9cb5b8617..af26b5f3f9cb 100644
--- a/spec/services/invoices/create_one_off_service_spec.rb
+++ b/spec/services/invoices/create_one_off_service_spec.rb
@@ -68,7 +68,7 @@
     end
 
     it_behaves_like "applies invoice_custom_sections" do
-      let(:service_call) { invoice_service.call }
+      let(:service_call) { create_service.call }
     end
 
     it 'calls SegmentTrackJob' do
diff --git a/spec/services/invoices/refresh_draft_and_finalize_service_spec.rb b/spec/services/invoices/refresh_draft_and_finalize_service_spec.rb
index 7c93708790eb..dfc977abbfdb 100644
--- a/spec/services/invoices/refresh_draft_and_finalize_service_spec.rb
+++ b/spec/services/invoices/refresh_draft_and_finalize_service_spec.rb
@@ -80,7 +80,7 @@
     end
 
     it_behaves_like "applies invoice_custom_sections" do
-      let(:service_call) { invoice_service.call }
+      let(:service_call) { finalize_service.call }
     end
 
     it 'enqueues a SendWebhookJob' do