Add SetTokens() as well

Author: loafoe

console/login.go

@@ -46,10 +46,21 @@ func (c *Client) SetToken(token string) *Client {
 	defer c.Unlock()
 
 	c.token = token
-	c.expiresAt = time.Now().Add(24 * time.Hour)
+	c.expiresAt = time.Now().Add(600 * time.Second)
 	return c
 }
 
+// SetTokens sets the tokens
+func (c *Client) SetTokens(accessToken, refreshToken, idToken string, expiresAt int64) {
+	c.Lock()
+	defer c.Unlock()
+	c.token = accessToken
+	c.refreshToken = refreshToken
+	c.idToken = idToken
+	c.expiresAt = time.Unix(expiresAt, 0)
+	c.tokenType = oAuthToken
+}
+
 func (c *Client) doTokenRequest(req *http.Request) error {
 	var tokenResponse tokenResponse
 

console/login_test.go

@@ -4,6 +4,7 @@ import (
 	"github.com/philips-software/go-hsdp-api/console"
 	"github.com/stretchr/testify/assert"
 	"testing"
+	"time"
 )
 
 func TestSetToken(t *testing.T) {
@@ -18,5 +19,8 @@ func TestSetToken(t *testing.T) {
 		return
 	}
 	token := "MyToken"
+	token2 := "TokenMy"
 	assert.Equal(t, token, client.SetToken(token).Token())
+	client.SetTokens(token, token2, token, time.Now().Add(10*time.Minute).Unix())
+	assert.Equal(t, token2, client.RefreshToken())
 }