From 8e6e360fc8086f208cabe7e870608ee327e2ff64 Mon Sep 17 00:00:00 2001
From: Bertocq <bertocq@gmail.com>
Date: Sun, 4 Feb 2018 22:38:23 +0100
Subject: [PATCH] Create traits for budget group & heading with drafting budget

---
 spec/factories.rb | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/spec/factories.rb b/spec/factories.rb
index f6d121f50..73512a689 100644
--- a/spec/factories.rb
+++ b/spec/factories.rb
@@ -278,6 +278,10 @@ FactoryBot.define do
   factory :budget_group, class: 'Budget::Group' do
     budget
     sequence(:name) { |n| "Group #{n}" }
+
+    trait :drafting_budget do
+      association :budget, factory: [:budget, :drafting]
+    end
   end
 
   factory :budget_heading, class: 'Budget::Heading' do
@@ -285,6 +289,10 @@ FactoryBot.define do
     sequence(:name) { |n| "Heading #{n}" }
     price 1000000
     population 1234
+
+    trait :drafting_budget do
+      association :group, factory: [:budget_group, :drafting_budget]
+    end
   end
 
   factory :budget_investment, class: 'Budget::Investment' do