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>
This commit is contained in:
Charlie Egan
2021-09-08 20:47:08 +01:00
parent c2191debc6
commit d3c9211f5b
2 changed files with 65 additions and 0 deletions

View File

@@ -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
}