Merge pull request #38 from DripEmail/allow_alternative_url_prefix

Allow alternative url prefix

Author: will-in-wi

lib/drip/client.rb

@@ -36,12 +36,13 @@ class Client
     include Workflows
     include WorkflowTriggers
 
-    attr_accessor :access_token, :api_key, :account_id
+    attr_accessor :access_token, :api_key, :account_id, :url_prefix
 
     def initialize(options = {})
       @account_id = options[:account_id]
       @access_token = options[:access_token]
       @api_key = options[:api_key]
+      @url_prefix = options[:url_prefix] || "https://api.getdrip.com/v2/"
       yield(self) if block_given?
     end
 
@@ -90,7 +91,7 @@ def build_response(&block)
 
     def connection
       @connection ||= Faraday.new do |f|
-        f.url_prefix = "https://api.getdrip.com/v2/"
+        f.url_prefix = url_prefix
         f.headers['User-Agent'] = "Drip Ruby v#{Drip::VERSION}"
         f.headers['Content-Type'] = content_type
         f.headers['Accept'] = "*/*"

test/drip/client_test.rb

@@ -11,6 +11,19 @@ class Drip::ClientTest < Drip::TestCase
       assert_equal "aaaa", client.api_key
     end
 
+    should "accept url prefix" do
+      client = Drip::Client.new do |config|
+        config.url_prefix = "aaaa"
+      end
+
+      assert_equal "aaaa", client.url_prefix
+    end
+
+    should "have default url prefix" do
+      client = Drip::Client.new
+      assert_equal "https://api.getdrip.com/v2/", client.url_prefix
+    end
+
     should "accept access token" do
       client = Drip::Client.new do |config|
         config.access_token = "aaaa"
@@ -66,6 +79,21 @@ class Drip::ClientTest < Drip::TestCase
     end
   end
 
+  context "given a different url prefix" do
+    setup do
+      @key = "aaaa"
+      @url_prefix = "https://api.example.com/v9001/"
+      @client = Drip::Client.new do |config|
+        config.api_key = @key
+        config.url_prefix = @url_prefix
+      end
+    end
+
+    should "connect to alternate prefix" do
+      assert_equal @url_prefix, @client.connection.url_prefix.to_s
+    end
+  end
+
   context "given a OAuth access token" do
     setup do
       @key = "aaaa"