From 0a1b2837cf3da02411371379fa79d3b916a83b62 Mon Sep 17 00:00:00 2001
From: Matt Robinson <matt@mattrobinson.net>
Date: Tue, 13 Mar 2018 13:43:00 -0700
Subject: [PATCH] Add test for oauth based login

---
 test/forcex/client_test.exs | 39 ++++++++++++++++++++++++++++++++++++-
 1 file changed, 38 insertions(+), 1 deletion(-)

diff --git a/test/forcex/client_test.exs b/test/forcex/client_test.exs
index 384be0b..105fb15 100644
--- a/test/forcex/client_test.exs
+++ b/test/forcex/client_test.exs
@@ -8,9 +8,10 @@ defmodule Forcex.ClientTest do
     test "sets the auth header and endpoint when successful" do
       session_id = "forcex_session_id"
       server_url = "https://forcex.my.salesforce.com/services/Soap/u/41.0/00Dd0000000cQ8L"
+      org_id = "org_id"
 
       response = """
-      <?xml version=\"1.0\" encoding=\"UTF-8\"?><soapenv:Envelope xmlns:soapenv=\"http://schemas.xmlsoap.org/soap/envelope/\" xmlns=\"urn:partner.soap.sforce.com\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"><soapenv:Body><loginResponse><result><metadataServerUrl>#{server_url}</metadataServerUrl><passwordExpired>false</passwordExpired><sandbox>false</sandbox><serverUrl>#{server_url}</serverUrl><sessionId>#{session_id}</sessionId><userId>005d0000001Jb9tAAC</userId><userInfo><accessibilityMode>false</accessibilityMode><chatterExternal>false</chatterExternal><currencySymbol>$</currencySymbol><orgAttachmentFileSizeLimit>5242880</orgAttachmentFileSizeLimit><orgDefaultCurrencyIsoCode>USD</orgDefaultCurrencyIsoCode><orgDefaultCurrencyLocale>en_US</orgDefaultCurrencyLocale><orgDisallowHtmlAttachments>false</orgDisallowHtmlAttachments><orgHasPersonAccounts>true</orgHasPersonAccounts><organizationId>00Dd0000000cQ8LEAU</organizationId><organizationMultiCurrency>false</organizationMultiCurrency><organizationName>MY-ORG</organizationName><profileId>00ed0000000Ods2AAC</profileId><roleId>00Ed0000000II8UEAW</roleId><sessionSecondsValid>7200</sessionSecondsValid><userDefaultCurrencyIsoCode xsi:nil=\"true\"/><userEmail>forcex@example.com</userEmail><userFullName>John Doe</userFullName><userId>005d0000001Jb9tAAC</userId><userLanguage>en_US</userLanguage><userLocale>en_US</userLocale><userName>forcex@example.com</userName><userTimeZone>America/New_York</userTimeZone><userType>Standard</userType><userUiSkin>Theme3</userUiSkin></userInfo></result></loginResponse></soapenv:Body></soapenv:Envelope>
+      <?xml version=\"1.0\" encoding=\"UTF-8\"?><soapenv:Envelope xmlns:soapenv=\"http://schemas.xmlsoap.org/soap/envelope/\" xmlns=\"urn:partner.soap.sforce.com\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"><soapenv:Body><loginResponse><result><metadataServerUrl>#{server_url}</metadataServerUrl><passwordExpired>false</passwordExpired><sandbox>false</sandbox><serverUrl>#{server_url}</serverUrl><sessionId>#{session_id}</sessionId><userId>005d0000001Jb9tAAC</userId><userInfo><accessibilityMode>false</accessibilityMode><chatterExternal>false</chatterExternal><currencySymbol>$</currencySymbol><orgAttachmentFileSizeLimit>5242880</orgAttachmentFileSizeLimit><orgDefaultCurrencyIsoCode>USD</orgDefaultCurrencyIsoCode><orgDefaultCurrencyLocale>en_US</orgDefaultCurrencyLocale><orgDisallowHtmlAttachments>false</orgDisallowHtmlAttachments><orgHasPersonAccounts>true</orgHasPersonAccounts><organizationId>#{org_id}</organizationId><organizationMultiCurrency>false</organizationMultiCurrency><organizationName>MY-ORG</organizationName><profileId>00ed0000000Ods2AAC</profileId><roleId>00Ed0000000II8UEAW</roleId><sessionSecondsValid>7200</sessionSecondsValid><userDefaultCurrencyIsoCode xsi:nil=\"true\"/><userEmail>forcex@example.com</userEmail><userFullName>John Doe</userFullName><userId>005d0000001Jb9tAAC</userId><userLanguage>en_US</userLanguage><userLocale>en_US</userLocale><userName>forcex@example.com</userName><userTimeZone>America/New_York</userTimeZone><userType>Standard</userType><userUiSkin>Theme3</userUiSkin></userInfo></result></loginResponse></soapenv:Body></soapenv:Envelope>
 """
 
       Forcex.Api.MockHttp
@@ -32,4 +33,40 @@ defmodule Forcex.ClientTest do
       assert client.endpoint == "https://forcex.my.salesforce.com/"
     end
   end
+
+  describe "oauth based login" do
+    test "sets the auth header and endpoint when successful" do
+      org_id = "org_id"
+      access_token = "access_token"
+
+      response = %{
+        access_token: access_token,
+        id: "https://login.salesforce.com/id/#{org_id}/005d0000001Jb9tAAC",
+        instance_url: "https://forcex.my.salesforce.com",
+        issued_at: "1520973086810",
+        signature: "oo7i3klbG6OjXlMFQSBzFaNYCP9pnWZ98f6Kdu/Th2Q=",
+        token_type: "Bearer"
+      }
+
+      Forcex.Api.MockHttp
+      |> expect(:raw_request, fn :post, _, _, _, _ -> response end)
+
+      config = %{
+        password: "password",
+        security_token: "security_token",
+        username: "forcex@example.com",
+        client_id: "big_ol_id",
+        client_secret: "sssshhhhhhhh"
+      }
+
+      client = Forcex.Client.login(config)
+
+      assert client.authorization_header == [{
+        "Authorization",
+        "Bearer #{access_token}"
+      }]
+
+      assert client.endpoint == "https://forcex.my.salesforce.com"
+    end
+  end
 end