From d3c9211f5b217e95fdea91d4dccad20bc419e912 Mon Sep 17 00:00:00 2001
From: Charlie Egan <charlieegan3@users.noreply.github.com>
Date: Wed, 8 Sep 2021 20:47:08 +0100
Subject: [PATCH] Allow setting the baseURL of clients

This allows clients to be pointed at local test servers in tests to mock
responses to Airtable API calls.

Signed-off-by: Charlie Egan <charlieegan3@users.noreply.github.com>
---
 client.go      | 19 +++++++++++++++++++
 client_test.go | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 65 insertions(+)

diff --git a/client.go b/client.go
index e77616a..d136811 100644
--- a/client.go
+++ b/client.go
@@ -48,6 +48,25 @@ func (at *Client) SetRateLimit(customRateLimit int) {
 	at.rateLimiter = time.Tick(time.Second / time.Duration(customRateLimit))
 }
 
+func (at *Client) SetBaseURL(baseURL string) error {
+	url, err := url.Parse(baseURL)
+	if err != nil {
+		return fmt.Errorf("failed to parse baseURL: %s", err)
+	}
+
+	if url.Scheme == "" {
+		return fmt.Errorf("scheme of http or https must be specified")
+	}
+
+	if url.Scheme != "https" && url.Scheme != "http" {
+		return fmt.Errorf("http or https baseURL must be used")
+	}
+
+	at.baseURL = url.String()
+
+	return nil
+}
+
 func (at *Client) rateLimit() {
 	<-at.rateLimiter
 }
diff --git a/client_test.go b/client_test.go
index 1635d73..bc267d3 100644
--- a/client_test.go
+++ b/client_test.go
@@ -9,6 +9,7 @@ import (
 	"errors"
 	"log"
 	"net/http"
+	"strings"
 	"testing"
 )
 
@@ -35,3 +36,48 @@ func TestClient_do(t *testing.T) {
 		t.Errorf("there should be an error, but was nil")
 	}
 }
+
+func TestClient_SetBaseURL(t *testing.T) {
+	testCases := []struct {
+		description   string
+		url           string
+		expectedError string
+	}{
+		{
+			description: "accepts a valid URL",
+			url:         "http://localhost:3000",
+		},
+		{
+			description: "accepts a valid HTTPS URL",
+			url:         "https://example.com",
+		},
+		{
+			description:   "rejects non http/https scheme url",
+			url:           "ftp://example.com",
+			expectedError: "http or https baseURL must be used",
+		},
+		{
+			description:   "rejects url without scheme",
+			url:           "example.com",
+			expectedError: "scheme of http or https must be specified",
+		},
+	}
+
+	c := testClient(t)
+
+	for _, testCase := range testCases {
+		t.Run(testCase.description, func(t *testing.T) {
+			err := c.SetBaseURL(testCase.url)
+
+			if err != nil {
+				if testCase.expectedError != "" {
+					if !strings.Contains(err.Error(), testCase.expectedError) {
+						t.Fatalf("unexpected error: %s", err)
+					}
+				} else {
+					t.Fatalf("unexpected error: %s", err)
+				}
+			}
+		})
+	}
+}