1 files changed, 75 insertions, 11 deletions

diff --git a/src/server_test.go b/src/server_test.go
index 3492e01..618c3b6 100644
--- a/src/server_test.go
+++ b/src/server_test.go
@@ -1,13 +1,13 @@
 package eduvpn
 
 import (
+	"crypto/tls"
 	"errors"
 	"fmt"
-	"testing"
 	"net/http"
-	"crypto/tls"
 	"os/exec"
 	"strings"
+	"testing"
 )
 
 func runCommand(t *testing.T, errBuffer *strings.Builder, name string, args ...string) error {
@@ -22,12 +22,11 @@ func runCommand(t *testing.T, errBuffer *strings.Builder, name string, args ...s
 	return cmd.Wait()
 }
 
-func LoginOAuthSelenium(t* testing.T, url string) {
+func LoginOAuthSelenium(t *testing.T, url string) {
 	// We could use the go selenium library
 	// But it does not support the latest selenium v4 just yet
 	var errBuffer strings.Builder
 	err := runCommand(t, &errBuffer, "python3", "../selenium_eduvpn.py", url)
-
 	if err != nil {
 		t.Errorf("Login OAuth with selenium script failed with error %v and stderr %s", err, errBuffer.String())
 	}
@@ -45,9 +44,10 @@ func Test_server(t *testing.T) {
 	// Do not verify because during testing, the cert is self-signed
 	http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
 
-	state.Register("org.eduvpn.app.linux", "configs", func(old string, new string, data string) {
+	state.Register("org.eduvpn.app.linux", "configstest", func(old string, new string, data string) {
 		StateCallback(t, old, new, data)
 	})
+
 	_, configErr := state.Connect("https://eduvpnserver")
 
 	if configErr != nil {
@@ -55,7 +55,7 @@ func Test_server(t *testing.T) {
 	}
 }
 
-func test_connect_oauth_parameter(t* testing.T, parameters URLParameters, expectedErr interface{}) {
+func test_connect_oauth_parameter(t *testing.T, parameters URLParameters, expectedErr interface{}) {
 	state := &VPNState{}
 
 	// Do not verify because during testing, the cert is self-signed
@@ -78,23 +78,87 @@ func test_connect_oauth_parameter(t* testing.T, parameters URLParameters, expect
 	}
 }
 
-func Test_connect_oauth_parameters(t* testing.T) {
-
+func Test_connect_oauth_parameters(t *testing.T) {
 	var (
-		failedCallbackParameterError *OAuthFailedCallbackParameterError
+		failedCallbackParameterError  *OAuthFailedCallbackParameterError
 		failedCallbackStateMatchError *OAuthFailedCallbackStateMatchError
 	)
 
 	tests := []struct {
 		expectedErr interface{}
-		parameters URLParameters
+		parameters  URLParameters
 	}{
 		{&failedCallbackParameterError, URLParameters{}},
 		{&failedCallbackParameterError, URLParameters{"code": "42"}},
-		{&failedCallbackStateMatchError, URLParameters{"code": "42", "state": "21",}},
+		{&failedCallbackStateMatchError, URLParameters{"code": "42", "state": "21"}},
 	}
 
 	for _, test := range tests {
 		test_connect_oauth_parameter(t, test.parameters, test.expectedErr)
 	}
 }
+
+func Test_token_refresh(t *testing.T) {
+	state := GetVPNState()
+
+	// Do not verify because during testing, the cert is self-signed
+	http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
+
+	state.Register("org.eduvpn.app.linux", "configsrefresh", func(old string, new string, data string) {
+		StateCallback(t, old, new, data)
+	})
+
+	// Fake expiry
+	state.Server.OAuth.Token.ExpiredTimestamp = GenerateTimeSeconds()
+	accessToken := state.Server.OAuth.Token.Access
+	refreshToken := state.Server.OAuth.Token.Refresh
+
+	_, configErr := state.Connect("https://eduvpnserver")
+
+	if configErr != nil {
+		t.Errorf("Connect error: %v", configErr)
+	}
+
+	// Check if tokens have changed
+	accessTokenAfter := state.Server.OAuth.Token.Access
+	refreshTokenAfter := state.Server.OAuth.Token.Refresh
+
+	if accessToken == accessTokenAfter {
+		t.Errorf("Access token is the same after refresh")
+	}
+
+	if refreshToken == refreshTokenAfter {
+		t.Errorf("Refresh token is the same after refresh")
+	}
+}
+
+func Test_token_invalid(t *testing.T) {
+	state := GetVPNState()
+
+	// Do not verify because during testing, the cert is self-signed
+	http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
+
+	state.Register("org.eduvpn.app.linux", "configsinvalid", func(old string, new string, data string) {
+		StateCallback(t, old, new, data)
+	})
+
+	dummy_value := "37"
+
+	// Override tokens with invalid values
+	state.Server.OAuth.Token.Access = dummy_value
+	state.Server.OAuth.Token.Refresh = dummy_value
+
+	_, configErr := state.Connect("https://eduvpnserver")
+
+	if configErr != nil {
+		t.Errorf("Connect error: %v", configErr)
+	}
+
+	if state.Server.OAuth.Token.Access == dummy_value {
+		t.Errorf("Access token is equal to dummy value: %s", dummy_value)
+	}
+
+	if state.Server.OAuth.Token.Refresh == dummy_value {
+		t.Errorf("Refresh token is equal to dummy value: %s", dummy_value)
+	}
+}