Merge pull request #21 from codeclimate/pb-asana

Add Asana

Author: pbrisbin

lib/cc/services/asana.rb

@@ -0,0 +1,68 @@
+class CC::Service::Asana < CC::Service
+  class Config < CC::Service::Config
+    attribute :api_key, String, label: "API key"
+
+    attribute :workspace_id, String, label: "Workspace ID"
+
+    attribute :project_id, String, label: "Project ID",
+      description: "(optional)"
+
+    attribute :assignee, String, label: "Assignee",
+      description: "Assignee email address (optional)"
+
+    validates :api_key, presence: true
+    validates :workspace_id, presence: true
+  end
+
+  self.title = "Asana"
+  self.description = "Create tasks in Asana"
+  self.issue_tracker = true
+
+  def receive_test
+    create_task("Test task from Code Climate")
+  end
+
+  def receive_quality
+    title = "Refactor #{constant_name} from #{rating} on Code Climate"
+
+    create_task("#{title} - #{details_url}")
+  end
+
+  def receive_vulnerability
+    formatter = CC::Formatters::TicketFormatter.new(self)
+    title     = formatter.format_vulnerability_title
+
+    create_task("#{title} - #{details_url}")
+  end
+
+private
+
+  def create_task(name)
+    params = {
+      data: { workspace: config.workspace_id, name: name }
+    }
+
+    if config.project_id.present?
+      # Note this is undocumented, found via trial & error
+      params[:data][:projects] = [config.project_id]
+    end
+
+    if config.assignee.present?
+      params[:data][:assignee] = config.assignee
+    end
+
+    http.headers["Content-Type"] = "application/json"
+    http.basic_auth(config.api_key, "")
+
+    url = "https://app.asana.com/api/1.0/tasks"
+    res = http_post(url, params.to_json)
+
+    body = JSON.parse(res.body)
+
+    id = body['data']['id']
+    url = "https://app.asana.com/0/#{config.workspace_id}/#{id}"
+
+    { id: id, url: url }
+  end
+
+end

service_test.rb

@@ -4,17 +4,17 @@
 #
 # Usage:
 #
-#   bundle exec ruby service_test.rb
+#   $ <SERVICE>_<CONFIG_ATTR_1>="..." \
+#     <SERVICE>_<CONFIG_ATTR_2>="..." \
+#     ... ... bundle exec ruby service_test.rb
 #
-# Environment variables used:
+# Example:
 #
-#   REPO_NAME           Defaults to "App"
-#   SLACK_WEBHOOK_URL   Slack is not tested unless set
-#   FLOWDOCK_API_TOKEN  Flowdock is not tested unless set
+#   $ SLACK_WEBHOOK_URL="http://..." bundle exec ruby service_test.rb
 #
-# Example:
+# Other Environment variables used:
 #
-#   SLACK_WEBHOOK_URL="http://..." bundle exec ruby service_test.rb
+#   REPO_NAME  Defaults to "App"
 #
 ###
 require 'cc/services'
@@ -30,33 +30,52 @@ def call
   end
 end
 
-def test_service(klass, config)
-  repo_name = ENV["REPO_NAME"] || "App"
+class ServiceTest
+  def initialize(klass, *params)
+    @klass = klass
+    @params = params
+  end
+
+  def test
+    config = {}
 
-  service = klass.new(config, name: :test, repo_name: repo_name)
+    puts "-"*80
+    puts @klass
 
-  CC::Service::Invocation.new(service) do |i|
-    i.wrap(WithResponseLogging)
+    @params.each do |param|
+      if var = ENV[to_env_var(param)]
+        config[param] = var
+      else
+        puts "  -> skipping"
+        return false
+      end
+    end
+
+    puts "  -> testing"
+    puts "  -> #{config.inspect}"
+    print "  => "
+
+    test_service(@klass, config)
   end
-end
 
-if webhook_url = ENV["SLACK_WEBHOOK_URL"]
-  puts "Testing Slack..."
-  test_service(CC::Service::Slack, webhook_url: webhook_url)
-end
+private
 
-if api_token = ENV["FLOWDOCK_API_TOKEN"]
-  puts "Testing Flowdock..."
-  test_service(CC::Service::Flowdock, api_token: api_token)
-end
+  def to_env_var(param)
+    "#{@klass.to_s.split("::").last}_#{param}".upcase
+  end
+
+  def test_service(klass, config)
+    repo_name = ENV["REPO_NAME"] || "App"
 
-if (jira_username = ENV["JIRA_USERNAME"]) &&
-   (jira_password = ENV["JIRA_PASSWORD"]) &&
-   (jira_domain   = ENV["JIRA_DOMAIN"])   &&
-   (jira_project  = ENV["JIRA_PROJECT"])
-  puts "Testing JIRA"
-  test_service(CC::Service::Jira, { username:   jira_username,
-                                    password:   jira_password,
-                                    domain:     jira_domain,
-                                    project_id: jira_project })
+    service = klass.new(config, name: :test, repo_name: repo_name)
+
+    CC::Service::Invocation.new(service) do |i|
+      i.wrap(WithResponseLogging)
+    end
+  end
 end
+
+ServiceTest.new(CC::Service::Slack, :webhook_url).test
+ServiceTest.new(CC::Service::Flowdock, :api_token).test
+ServiceTest.new(CC::Service::Jira, :username, :password, :domain, :project_id).test
+ServiceTest.new(CC::Service::Asana, :api_key, :workspace_id, :project_id).test

test/asana_test.rb

@@ -0,0 +1,42 @@
+require File.expand_path('../helper', __FILE__)
+
+class TestAsana < CC::Service::TestCase
+  def test_quality
+    assert_asana_receives(
+      event(:quality, to: "D", from: "C"),
+      "Refactor User from a D on Code Climate - https://codeclimate.com/repos/1/feed"
+    )
+  end
+
+  def test_vulnerability
+    assert_asana_receives(
+      event(:vulnerability, vulnerabilities: [{
+        "warning_type" => "critical",
+        "location" => "app/user.rb line 120"
+      }]),
+      "New critical issue found in app/user.rb line 120 - https://codeclimate.com/repos/1/feed"
+    )
+  end
+
+  private
+
+  def assert_asana_receives(event_data, name)
+    @stubs.post '/api/1.0/tasks' do |env|
+      body = JSON.parse(env[:body])
+      data = body["data"]
+
+      assert_equal "1",             data["workspace"]
+      assert_equal "2",             data["projects"].first
+      assert_equal "[email protected]", data["assignee"]
+      assert_equal name,            data["name"]
+
+      [200, {}, '{"data":{"id":{}}}']
+    end
+
+    receive(
+      CC::Service::Asana,
+      { api_key: "abc123", workspace_id: "1", project_id: "2", assignee: "[email protected]" },
+      event_data
+    )
+  end
+end