Avoid enum issue (#39)

djezzzl · Jun 15, 2023 · 7ba2052 · 7ba2052
1 parent 1dcd9f2
commit 7ba2052
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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
@@ -2,6 +2,8 @@
 
 FactoryBot.define do
   factory :user do
+    traits_for_enum :visibility, [:public, :private]
+
     trait :with_phone do
       phone { "phone" }
     end