lib/meibo/factory_bot/enrollment.rb in meibo-0.8.0 vs lib/meibo/factory_bot/enrollment.rb in meibo-0.9.0

@@ -31,7 +31,19 @@
     end
 
     trait :guardian do
       role { Meibo::Enrollment::ROLES[:guardian] }
     end
+
+    trait :jp do
+      initialize_with { Meibo::JapanProfile::Classroom.new(**attributes) }
+    end
+
+    trait :public do
+      public_flg { true }
+    end
+
+    trait :private do
+      public_flg { false }
+    end
   end
 end