From 7eea4fdd1c5408524c77285455f4d478869e59cd Mon Sep 17 00:00:00 2001
From: Maikel Linke <maikel@email.org.au>
Date: Mon, 8 Aug 2022 17:01:48 +1000
Subject: [PATCH] Adds zone_with_state_member to zone factory

And make zone factories more flexible.
---
 spec/factories/zone_factory.rb | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/spec/factories/zone_factory.rb b/spec/factories/zone_factory.rb
index 9215373e64..4d4ed09542 100644
--- a/spec/factories/zone_factory.rb
+++ b/spec/factories/zone_factory.rb
@@ -7,10 +7,15 @@ FactoryBot.define do
   end
 
   factory :zone_with_member, parent: :zone do
-    default_tax { true }
-
-    after(:create) do |zone|
-      Spree::ZoneMember.create!(zone: zone, zoneable: Spree::Country.find_by(name: 'Australia'))
+    transient do
+      member { Spree::Country.find_by(name: "Australia") }
     end
+
+    default_tax { true }
+    zone_members { [Spree::ZoneMember.new(zoneable: member)] }
+  end
+
+  factory :zone_with_state_member, parent: :zone_with_member do
+    member { Spree::State.find_by(name: "Victoria") }
   end
 end