add rails_brakeman service

flyerhzm · flyerhzm · commit a9d7042837c8 · 2012-05-19T09:43:27.000+08:00
diff --git a/services/rails_brakeman.rb b/services/rails_brakeman.rb
@@ -0,0 +1,20 @@
+class Service::RailsBrakeman < Service
+  string :rails_brakeman_url, :token
+  white_list :rails_brakeman_url
+
+  def receive_push
+    http_post rails_brakeman_url, :token => token, :payload => payload.to_json
+  end
+
+  def rails_brakeman_url
+    if !(url = data["rails_brakeman_url"].to_s).empty?
+      url.strip
+    else
+      "https://rails-brakeman.com"
+    end
+  end
+
+  def token
+    data['token'].strip
+  end
+end
diff --git a/test/rails_brakeman_test.rb b/test/rails_brakeman_test.rb
@@ -0,0 +1,87 @@
+require File.expand_path('../helper', __FILE__)
+
+class RailsBrakemanTest < Service::TestCase
+  def setup
+    @stubs = Faraday::Adapter::Test::Stubs.new
+    @svc = service(data, payload)
+  end
+
+  def test_reads_token_from_data
+    assert_equal "xAAQZtJhYHGagsed1kYR", @svc.token
+  end
+
+  def test_reads_default_rails_brakeman_url_from_data
+    assert_equal "https://rails-brakeman.com", @svc.rails_brakeman_url
+  end
+
+  def test_reads_custom_rails_brakeman_url_from_data
+    data = { "token" => "xAAQZtJhYHGagsed1kYR", "rails_brakeman_url" => "http://rails-brakeman.heroku.com" }
+    svc = service(data, payload)
+    assert_equal "http://rails-brakeman.heroku.com", svc.rails_brakeman_url
+  end
+
+  def test_strips_whitespace_from_form_values
+    data = { "token" => "  xAAQZtJhYHGagsed1kYR  ", "rails_brakeman_url" => "  http://rails-brakeman.heroku.com  " }
+    svc = service(data, payload)
+    assert_equal "xAAQZtJhYHGagsed1kYR", svc.token
+    assert_equal "http://rails-brakeman.heroku.com", svc.rails_brakeman_url
+  end
+
+  def test_posts_payload
+    @stubs.post "/" do |env|
+      assert_equal payload, JSON.parse(Rack::Utils.parse_query(env[:body])['payload'])
+    end
+    @svc.receive_push
+  end
+
+  def service(*args)
+    super Service::RailsBrakeman, *args
+  end
+
+  def data
+    { "token" => "xAAQZtJhYHGagsed1kYR", 'rails_brakeman_url' => '' }
+  end
+
+  def payload
+    {
+      "before" => "a6ab010bc21151e238c73d5229c36892d51c2d4f",
+      "repository" => {
+        "url" => "https =>//github.com/flyerhzm/rails-brakeman.com",
+        "name" => "rails-brakeman.com",
+        "description" => "rails-brakeman.com",
+        "watchers" => 1,
+        "forks" => 1,
+        "private" => 0,
+        "owner" => {
+          "email" => "flyerhzm@gmail.com",
+          "name" => "Richard Huang"
+        }
+      },
+      "commits" => [
+        {
+          "id" => "af9718a9bee64b9bbbefc4c9cf54c4cc102333a8",
+          "url" => "https =>//github.com/flyerhzm/rails-brakeman.com/commit/af9718a9bee64b9bbbefc4c9cf54c4cc102333a8",
+          "author" => {
+            "email" => "flyerhzm@gmail.com",
+            "name" => "Richard Huang"
+          },
+          "message" => "fix typo in .travis.yml",
+          "timestamp" => "2011-12-25T18 =>57 =>17+08 =>00",
+          "modified" => [".travis.yml"]
+        },
+        {
+          "id" => "473d12b3ca40a38f12620e31725922a9d88b5386",
+          "url" => "https =>//github.com/flyerhzm/rails-brakeman.com/commit/473d12b3ca40a38f12620e31725922a9d88b5386",
+          "author" => {
+            "email" => "flyerhzm@gmail.com",
+            "name" => "Richard Huang"
+          },
+          "message" => "copy config yaml files for travis",
+          "timestamp" => "2011-12-25T20 =>36 =>34+08 =>00"
+        }
+      ],
+      "after" => "473d12b3ca40a38f12620e31725922a9d88b5386",
+      "ref" => "refs/heads/master"
+    }
+  end
+end
