lib/meibo/factory_bot/classroom.rb in meibo-0.9.0 vs lib/meibo/factory_bot/classroom.rb in meibo-0.10.0

@@ -18,9 +18,17 @@
     course_sourced_id { course&.sourced_id }
     class_type { Meibo::Classroom::TYPES[:homeroom] }
     school_sourced_id { school&.sourced_id }
     term_sourced_ids { terms.map(&:sourced_id) }
 
+    trait :homeroom do
+      class_type { Meibo::Classroom::TYPES[:homeroom] }
+    end
+
+    trait :scheduled do
+      class_type { Meibo::Classroom::TYPES[:scheduled] }
+    end
+
     trait :jp do
       initialize_with { Meibo::JapanProfile::Classroom.new(**attributes) }
     end
 
     trait :special_needs do