Add some validations to persisted stage model

1 files changed, 4 insertions, 8 deletions

diff --git a/app/models/concerns/has_status.rb b/app/models/concerns/has_status.rb
index 235196cae13..8ea5a007f76 100644
--- a/app/models/concerns/has_status.rb
+++ b/app/models/concerns/has_status.rb
@@ -12,6 +12,10 @@ module HasStatus
                     failed: 4, canceled: 5, skipped: 6, manual: 7 }
 
   class_methods do
+    def enumerated_status!
+      enum status: HasStatus::STATUSES_ENUM
+    end
+
     def status_sql
       scope_relevant = respond_to?(:exclude_ignored) ? exclude_ignored : all
       scope_warnings = respond_to?(:failed_but_allowed) ? failed_but_allowed : none
@@ -56,14 +60,6 @@ module HasStatus
     def all_state_names
       state_machines.values.flat_map(&:states).flat_map { |s| s.map(&:name) }
     end
-
-    private
-
-    def enumerated_status!
-      @status_strategy = :enumerator
-
-      enum status: HasStatus::STATUSES_ENUM
-    end
   end
 
   included do