lib/github-status/support/source.rb in concourse-github-status-0.2.2 vs lib/github-status/support/source.rb in concourse-github-status-0.2.4

@@ -19,8 +19,13 @@
         @repo ||= source.fetch 'repo'
       rescue KeyError
         STDERR.puts 'Source is missing repo'
         abort
       end
+
+      Contract None => String
+      def branch
+        @branch ||= source.fetch('branch') { 'master' }
+      end
     end
   end
 end