From 7d5d6b578cf90b63b74b9576e382b8e47dd442b8 Mon Sep 17 00:00:00 2001
From: Konstantin Haase <konstantin.mailinglists@googlemail.com>
Date: Wed, 14 Jan 2015 17:00:16 +0100
Subject: [PATCH] instrument access token

---
 lib/travis/api/app.rb              | 1 +
 lib/travis/api/app/access_token.rb | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/lib/travis/api/app.rb b/lib/travis/api/app.rb
index 07682455..8c0c86f3 100644
--- a/lib/travis/api/app.rb
+++ b/lib/travis/api/app.rb
@@ -1,3 +1,4 @@
+require 'skylight'
 require 'travis'
 require 'travis/model'
 require 'travis/support/amqp'
diff --git a/lib/travis/api/app/access_token.rb b/lib/travis/api/app/access_token.rb
index 99a70ad5..13219b3b 100644
--- a/lib/travis/api/app/access_token.rb
+++ b/lib/travis/api/app/access_token.rb
@@ -3,6 +3,8 @@ require 'securerandom'
 
 class Travis::Api::App
   class AccessToken
+    include Skylight::Helpers
+
     DEFAULT_SCOPES = [:public, :private]
     attr_reader :token, :scopes, :user_id, :app_id, :expires_in, :extra
 
@@ -22,6 +24,7 @@ class Travis::Api::App
       new(token: token, scopes: scopes, user_id: user_id, app_id: app_id, extra: extra) if user_id
     end
 
+    instrument_method
     def initialize(options = {})
       raise ArgumentError, 'must supply either user_id or user' unless options.key?(:user) ^ options.key?(:user_id)
       raise ArgumentError, 'must supply app_id' unless options.key?(:app_id)
@@ -40,6 +43,7 @@ class Travis::Api::App
       @extra    = options[:extra]
     end
 
+    instrument_method
     def save
       key = key(token)
       redis.del(key)
@@ -90,6 +94,7 @@ class Travis::Api::App
 
     private
 
+      instrument_method
       def reuse_token
         redis.get(reuse_key) unless expires_in
       end