Change timeout_source to enum

1 files changed, 14 insertions, 0 deletions

diff --git a/app/presenters/ci/build_presenter.rb b/app/presenters/ci/build_presenter.rb
index 255475e1fe6..be6cc2e70b1 100644
--- a/app/presenters/ci/build_presenter.rb
+++ b/app/presenters/ci/build_presenter.rb
@@ -1,5 +1,12 @@
 module Ci
   class BuildPresenter < Gitlab::View::Presenter::Delegated
+
+    TIMEOUT_SOURCES = {
+        unknown_timeout_source: nil,
+        project_timeout_source: 'project',
+        runner_timeout_source: 'runner'
+    }.freeze
+
     presents :build
 
     def erased_by_user?
@@ -18,6 +25,13 @@ module Ci
       end
     end
 
+    def timeout_source
+      return unless build.timeout_source?
+
+      TIMEOUT_SOURCES[build.timeout_source.to_sym] ||
+          build.timeout_source
+    end
+
     def trigger_variables
       return [] unless trigger_request