From d28f5e776b90a648a83246beac94518cd8183af4 Mon Sep 17 00:00:00 2001
From: Grzegorz Bizon <grzesiek.bizon@gmail.com>
Date: Mon, 5 Dec 2016 12:07:14 +0100
Subject: Implement pipeline status factory with extended status

---
 lib/gitlab/ci/status/pipeline/factory.rb | 39 ++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 lib/gitlab/ci/status/pipeline/factory.rb

(limited to 'lib')

diff --git a/lib/gitlab/ci/status/pipeline/factory.rb b/lib/gitlab/ci/status/pipeline/factory.rb
new file mode 100644
index 00000000000..71d27bf7cf5
--- /dev/null
+++ b/lib/gitlab/ci/status/pipeline/factory.rb
@@ -0,0 +1,39 @@
+module Gitlab
+  module Ci
+    module Status
+      module Pipeline
+        class Factory
+          EXTENDED_STATUSES = [Pipeline::SuccessWithWarnings]
+
+          def initialize(pipeline)
+            @pipeline = pipeline
+            @status = pipeline.status || :created
+          end
+
+          def fabricate!
+            if extended_status
+              extended_status.new(core_status)
+            else
+              core_status
+            end
+          end
+
+          private
+
+          def core_status
+            Gitlab::Ci::Status
+              .const_get(@status.capitalize)
+              .new(@pipeline)
+              .extend(Status::Pipeline::Common)
+          end
+
+          def extended_status
+            @extended ||= EXTENDED_STATUSES.find do |status|
+              status.matches?(@pipeline)
+            end
+          end
+        end
+      end
+    end
+  end
+end
-- 
cgit v1.2.1