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

@@ -18,9 +18,25 @@
     start_date { Date.new(school_year, 4, 1).iso8601 }
     end_date { Date.new(school_year + 1, 3, 31).iso8601 }
     parent_sourced_id { parent&.sourced_id }
     school_year { today.year }
 
+    trait :grading_period do
+      type { Meibo::AcademicSession::TYPES[:grading_period] }
+    end
+
+    trait :semester do
+      type { Meibo::AcademicSession::TYPES[:semester] }
+    end
+
+    trait :school_year do
+      type { Meibo::AcademicSession::TYPES[:school_year] }
+    end
+
+    trait :term do
+      type { Meibo::AcademicSession::TYPES[:term] }
+    end
+
     trait :jp do
       initialize_with { Meibo::JapanProfile::AcademicSession.new(**attributes) }
     end
   end
 end