From 7ba2052305542f0dfe58163c305e75d8ea0d5941 Mon Sep 17 00:00:00 2001
From: Evgeniy Demin <lawliet.djez@gmail.com>
Date: Thu, 15 Jun 2023 21:19:14 +0200
Subject: [PATCH] Avoid enum issue (#39)

---
 lib/factory_trace/monkey_patches/factory.rb | 2 +-
 lib/factory_trace/monkey_patches/trait.rb   | 2 +-
 rails7/app/models/user.rb                   | 1 +
 rails7/spec/factories.rb                    | 2 ++
 4 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/lib/factory_trace/monkey_patches/factory.rb b/lib/factory_trace/monkey_patches/factory.rb
index 86e4e15..9f2c75f 100644
--- a/lib/factory_trace/monkey_patches/factory.rb
+++ b/lib/factory_trace/monkey_patches/factory.rb
@@ -5,7 +5,7 @@ module MonkeyPatches
     module Factory
       attr_reader :definition_path
 
-      def initialize(name, definition_path, options = {})
+      def initialize(name, definition_path = "", options = {})
         @definition_path = definition_path
         super(name, options)
       end
diff --git a/lib/factory_trace/monkey_patches/trait.rb b/lib/factory_trace/monkey_patches/trait.rb
index 72fba99..4f2a809 100644
--- a/lib/factory_trace/monkey_patches/trait.rb
+++ b/lib/factory_trace/monkey_patches/trait.rb
@@ -5,7 +5,7 @@ module MonkeyPatches
     module Trait
       attr_reader :definition_path
 
-      def initialize(name, definition_path, &block)
+      def initialize(name, definition_path = "", &block)
         @definition_path = definition_path
         super(name, &block)
       end
diff --git a/rails7/app/models/user.rb b/rails7/app/models/user.rb
index 32949b9..2a0e66d 100644
--- a/rails7/app/models/user.rb
+++ b/rails7/app/models/user.rb
@@ -1,4 +1,5 @@
 # frozen_string_literal: true
 
 class User < ApplicationRecord
+  attr_accessor :visibility
 end
diff --git a/rails7/spec/factories.rb b/rails7/spec/factories.rb
index 9d4d934..c27a237 100644
--- a/rails7/spec/factories.rb
+++ b/rails7/spec/factories.rb
@@ -2,6 +2,8 @@
 
 FactoryBot.define do
   factory :user do
+    traits_for_enum :visibility, [:public, :private]
+
     trait :with_phone do
       phone { "phone" }
     end