save token expiration

api_client.go

@@ -6,13 +6,16 @@ import (
 	"fmt"
 	"net/http"
 	"net/url"
+	"time"
 )
 
 type ApiClient struct {
-	tenantId   int
-	apiKey     string
-	httpClient *http.Client
-	baseUrl    string
+	tenantId    int
+	apiKey      string
+	httpClient  *http.Client
+	baseUrl     string
+	apiToken    string
+	apiTokenExp time.Time
 }
 
 type ApiClientOption func(*ApiClient)
@@ -94,5 +97,12 @@ func (client *ApiClient) GenerateToken() (string, error) {
 		return "", err
 	}
 
+	client.apiToken = tokenResponse.Token
+	client.apiTokenExp = time.Now().Add(time.Minute * 45)
+
 	return tokenResponse.Token, nil
 }
+
+func (client *ApiClient) isApiTokenExpired() bool {
+	return client.apiTokenExp.Before(time.Now())
+}

api_client_http_test.go

@@ -45,7 +45,12 @@ func TestGenerateToken(t *testing.T) {
 	)
 	defer clientTest.Close()
 
+	assert.Equal(t, "", clientTest.apiClient.apiToken, "The initial api token should be empty")
+	assert.True(t, clientTest.apiClient.isApiTokenExpired(), "The initial api token exp should be before now")
+
 	token, err := clientTest.apiClient.GenerateToken()
 	assert.NoError(t, err, "Generating a token")
 	assert.Equal(t, "test-token-hello", token)
+	assert.Equal(t, "test-token-hello", clientTest.apiClient.apiToken)
+	assert.False(t, clientTest.apiClient.isApiTokenExpired(), "The api token should be unexpired")
 }