Merge pull request pardot#14 from drippdropp/prospect-account-support

Support for Prospect Accounts

Author: znbailey

lib/pardot/client.rb

@@ -14,6 +14,7 @@ class Client
     include Objects::ListMemberships
     include Objects::Opportunities
     include Objects::Prospects
+    include Objects::ProspectAccounts
     include Objects::Users
     include Objects::Visitors
     include Objects::Visits

lib/pardot/objects/prospect_accounts.rb

@@ -0,0 +1,55 @@
+module Pardot
+  module Objects
+    module ProspectAccounts
+      def prospect_accounts
+        @prospect_accounts ||= ProspectAccounts.new self
+      end
+
+      class ProspectAccounts
+
+        def initialize(client)
+          @client = client
+        end
+
+        def query(search_criteria)
+          result = get '/do/query', search_criteria, 'result'
+          result['total_results'] = result['total_results'].to_i if
+            result['total_results']
+          result
+        end
+
+        def describe(params={})
+          post('/do/describe', params)
+        end
+
+        def create(params={})
+          post('/do/create', params)
+        end
+
+        # read_by_id
+        # update_by_id
+        [:read, :update].each do |verb|
+          define_method(verb) do |id, params={}|
+            post(api_url(verb, 'id', id), params)
+          end
+        end
+
+        private
+
+        def api_url(verb, direct_to, value)
+          "/do/#{verb}/#{direct_to}/#{value}"
+        end
+
+        def get(path, params={}, result='prospectAccount')
+          response = @client.get('prospectAccount', path, params)
+          result ? response[result] : response
+        end
+
+        def post(path, params={}, result='prospectAccount')
+          response = @client.post('prospectAccount', path, params)
+          result ? response[result] : response
+        end
+      end
+    end
+  end
+end

lib/ruby-pardot.rb

@@ -11,6 +11,7 @@
 require 'pardot/objects/list_memberships'
 require 'pardot/objects/opportunities'
 require 'pardot/objects/prospects'
+require 'pardot/objects/prospect_accounts'
 require 'pardot/objects/users'
 require 'pardot/objects/visitors'
 require 'pardot/objects/visitor_activities'

spec/pardot/objects/prospect_accounts_spec.rb

@@ -0,0 +1,78 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../spec_helper')
+
+describe Pardot::Objects::ProspectAccounts do
+
+  before do
+    @client = create_client
+  end
+
+  describe "query" do
+
+    def sample_results
+      %(<?xml version="1.0" encoding="UTF-8"?>
+      <rsp stat="ok" version="1.0">
+        <result>
+          <total_results>2</total_results>
+          <prospectAccount>
+            <name>Spaceships R Us</name>
+          </prospectAccount>
+          <prospectAccount>
+            <name>Monsters Inc</name>
+          </prospectAccount>
+        </result>
+      </rsp>)
+    end
+
+    it "should take in some arguments and respond with valid items" do
+      fake_get "/api/prospectAccount/version/3/do/query?assigned=true&format=simple&user_key=bar&api_key=my_api_key", sample_results
+
+      @client.prospect_accounts.query(:assigned => true).should == {'total_results' => 2,
+        'prospectAccount'=>[
+          {'name'=>'Spaceships R Us'},
+          {'name'=>'Monsters Inc'}
+        ]}
+    end
+
+  end
+
+  describe 'read' do
+    def sample_results
+      %(<?xml version="1.0" encoding="UTF-8"?>
+      <rsp stat="ok" version="1.0">
+        <prospectAccount>
+          <id>1234</id>
+          <name>SupaDupaPanda</name>
+        </prospectAccount>
+        </rsp>)
+    end
+
+    it 'should return a valid account' do
+      fake_post '/api/prospectAccount/version/3/do/read/id/1234?assigned=true&format=simple&user_key=bar&api_key=my_api_key', sample_results
+
+      @client.prospect_accounts.read('1234', :assigned => true).should == {'id' => '1234', 'name' => 'SupaDupaPanda' }
+    end
+
+  end
+
+
+  describe 'create' do
+
+    def sample_results
+      %(<?xml version="1.0" encoding="UTF-8"?>
+      <rsp stat="ok" version="1.0">
+        <prospectAccount>
+          <name>SuperPanda</name>
+        </prospectAccount>
+      </rsp>)
+    end
+
+    it 'should return the prospect account' do
+      fake_post '/api/prospectAccount/version/3/do/create?api_key=my_api_key&user_key=bar&format=simple&name=SuperPanda', sample_results
+
+      @client.prospect_accounts.create(:name => 'SuperPanda').should == {"name"=>"SuperPanda"}
+
+    end
+
+  end
+
+end