From 97a86bc273ddf974e2bf2b9d95d25cd5a8d837ea Mon Sep 17 00:00:00 2001
From: KMY <tt@kmycode.net>
Date: Tue, 25 Feb 2025 08:41:36 +0900
Subject: [PATCH] Fix test

---
 .../admin/export_domain_blocks_controller_spec.rb  | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/spec/controllers/admin/export_domain_blocks_controller_spec.rb b/spec/controllers/admin/export_domain_blocks_controller_spec.rb
index b8e0696e55..1e748e34b6 100644
--- a/spec/controllers/admin/export_domain_blocks_controller_spec.rb
+++ b/spec/controllers/admin/export_domain_blocks_controller_spec.rb
@@ -44,11 +44,11 @@ RSpec.describe Admin::ExportDomainBlocksController do
       end
 
       it 'renders page with extended domain blocks' do
-        expect(assigns(:domain_blocks).map { |block| [block.domain, block.reject_favourite, block.reject_friend] }).to contain_exactly(
-          ['bad.domain', false, false],
-          ['worse.domain', false, false],
-          ['reject.media', false, false],
-          ['little.spam', true, false]
+        expect(mapped_batch_table_rows_with_expanded_params).to contain_exactly(
+          ['bad.domain', false],
+          ['worse.domain', false],
+          ['reject.media', false],
+          ['little.spam', true]
         )
       end
 
@@ -73,6 +73,10 @@ RSpec.describe Admin::ExportDomainBlocksController do
       batch_table_rows.map { |row| [row.at_css('[id$=_domain]')['value'], row.at_css('[id$=_severity]')['value'].to_sym] }
     end
 
+    def mapped_batch_table_rows_with_expanded_params
+      batch_table_rows.map { |row| [row.at_css('[id$=_domain]')['value'], row.at_css('[id$=_reject_favourite]')['value'] == 'true'] }
+    end
+
     def batch_table_rows
       response.parsed_body.css('body div.batch-table__row')
     end