Implement the refresh tokens flow (#29)

* Implemen refresh tokens flow

* Update e2e/keycloak/keycloak_test.go

Co-authored-by: Ignasi Barrera <[email protected]>

---------

Co-authored-by: Ignasi Barrera <[email protected]>

e2e/keycloak/docker-compose.yaml

@@ -73,7 +73,7 @@ services:
       test: /opt/keycloak/bin/kcadm.sh get realms/master --server http://localhost:8080 --realm master --user admin --password admin
       interval: 5s
       timeout: 2s
-      retries: 10
+      retries: 30
       start_period: 5s
     extra_hosts: # Required when running on Linux
       - "host.docker.internal:host-gateway"

e2e/keycloak/keycloak_test.go

@@ -20,6 +20,7 @@ import (
 	"net"
 	"net/http"
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/require"
 
@@ -74,3 +75,57 @@ func TestOIDC(t *testing.T) {
 	require.Equal(t, http.StatusOK, res.StatusCode)
 	require.Contains(t, string(body), "Access allowed")
 }
+
+func TestOIDCRefreshTokens(t *testing.T) {
+	skipIfDockerHostNonResolvable(t)
+
+	// Initialize the test OIDC client that will keep track of the state of the OIDC login process
+	client, err := common.NewOIDCTestClient(
+		common.WithCustomCA(testCAFile),
+		common.WithLoggingOptions(t.Log, true),
+	)
+	require.NoError(t, err)
+
+	// Send a request to the test server. It will be redirected to the IdP login page
+	res, err := client.Get(testURL)
+	require.NoError(t, err)
+
+	// Parse the response body to get the URL where the login page would post the user-entered credentials
+	require.NoError(t, client.ParseLoginForm(res.Body, keyCloakLoginFormID))
+
+	// Submit the login form to the IdP. This will authenticate and redirect back to the application
+	res, err = client.Login(map[string]string{"username": username, "password": password, "credentialId": ""})
+	require.NoError(t, err)
+
+	// Verify that we get the expected response from the application
+	body, err := io.ReadAll(res.Body)
+	require.NoError(t, err)
+	require.Equal(t, http.StatusOK, res.StatusCode)
+	require.Contains(t, string(body), "Access allowed")
+
+	// Access tokens should expire in 10 seconds (tried with 5, but keycloak setup fails)
+	// Let's perform a request now and after 10 seconds to verify that the access token is refreshed
+
+	t.Run("request with same tokens", func(t *testing.T) {
+		res, err = client.Get(testURL)
+		require.NoError(t, err)
+
+		body, err = io.ReadAll(res.Body)
+		require.NoError(t, err)
+		require.Equal(t, http.StatusOK, res.StatusCode)
+		require.Contains(t, string(body), "Access allowed")
+	})
+
+	t.Log("waiting for access token to expire...")
+	time.Sleep(10 * time.Second)
+
+	t.Run("request with expired tokens", func(t *testing.T) {
+		res, err = client.Get(testURL)
+		require.NoError(t, err)
+
+		body, err = io.ReadAll(res.Body)
+		require.NoError(t, err)
+		require.Equal(t, http.StatusOK, res.StatusCode)
+		require.Contains(t, string(body), "Access allowed")
+	})
+}

e2e/keycloak/setup-keycloak.sh

@@ -24,6 +24,13 @@ REDIRECT_URL=https://host.docker.internal:8443/callback
 
 set -ex
 
+/opt/keycloak/bin/kcadm.sh update realms/${REALM} \
+    -s accessTokenLifespan=10 \
+    --realm "${REALM}" \
+    --server "${KEYCLOAK_SERVER}" \
+    --user "${KEYCLOAK_ADMIN}" \
+    --password "${KEYCLOAK_ADMIN_PASSWORD}"
+
 /opt/keycloak/bin/kcadm.sh create users \
     -s username="${USERNAME}" \
     -s enabled=true \

e2e/suite.mk

@@ -18,7 +18,8 @@
 
 ROOT := $(shell git rev-parse --show-toplevel)
 
-include $(ROOT)/env.mk
+include $(ROOT)/env.mk # Load common variables
+-include $(ROOT)/.makerc  # Pick up any local overrides.
 
 # Force run of the e2e tests by default
 E2E_TEST_OPTS ?= -count=1