1 files changed, 8 insertions, 0 deletions
diff --git a/spec/factories/environments.rb b/spec/factories/environments.rb
index 9286f49bc59..44aa4527e12 100644
--- a/spec/factories/environments.rb
+++ b/spec/factories/environments.rb
@@ -44,5 +44,13 @@ FactoryBot.define do
       status { 'created' }
       self.when { 'manual' }
     end
+
+    trait :auto_stopped do
+      auto_stop_at { 1.day.ago }
+    end
+
+    trait :will_auto_stop do
+      auto_stop_at { 1.day.from_now }
+    end
   end
 end