FactoryBot.define do
  factory :orcamento_fiscal, class: 'Base::TipoDeOrcamento' do
		numero {1}
    codigo {"F"}
    descricao {"Orçamento Fiscal"}
    trait :orcamento_2016 do
			modulo_id { Orcamento.find_or_create_by!( FactoryBot.attributes_for( :orcamento_2016 ) ).id }
			modulo_type {"Orcamento"}
    end

    trait :orcamento_2017 do
			modulo_id { Orcamento.find_or_create_by!( FactoryBot.attributes_for( :orcamento_2017 ) ).id }
			modulo_type {"Orcamento"}
    end
  end

  factory :seguridade_social, class: 'Base::TipoDeOrcamento' do
		numero {2}
    codigo {"S"}
    descricao {"Orçamento da Seguridade Social"}

    trait :orcamento_2016 do
			modulo_id { Orcamento.find_or_create_by!( FactoryBot.attributes_for( :orcamento_2016 ) ).id }
			modulo_type {"Orcamento"}
    end

    trait :orcamento_2017 do
			modulo_id { Orcamento.find_or_create_by!( FactoryBot.attributes_for( :orcamento_2017 ) ).id }
			modulo_type {"Orcamento"}
    end
  end
end
