Implement the main class of test coverage badge

1 files changed, 30 insertions, 1 deletions

diff --git a/lib/gitlab/badge/coverage/report.rb b/lib/gitlab/badge/coverage/report.rb
index e6de15e085f..f06142003e3 100644
--- a/lib/gitlab/badge/coverage/report.rb
+++ b/lib/gitlab/badge/coverage/report.rb
@@ -5,13 +5,42 @@ module Gitlab
       # Test coverage report badge
       #
       class Report < Badge::Base
+        attr_reader :project, :ref, :job
+
         def initialize(project, ref, job = nil)
           @project = project
           @ref = ref
           @job = job
+
+          @pipeline = @project.pipelines
+            .where(ref: @ref)
+            .where(sha: @project.commit(@ref).try(:sha))
+            .first
         end
 
-        def coverage
+        def entity
+          'coverage'
+        end
+
+        def status
+          @coverage ||= raw_coverage
+          return unless @coverage
+
+          @coverage.to_i
+        end
+
+        private
+
+        def raw_coverage
+          return unless @pipeline
+
+          if @job.blank?
+            @pipeline.coverage
+          else
+            @pipeline.builds
+              .find_by(name: @job)
+              .try(:coverage)
+          end
         end
       end
     end