From 43604f7ffcbbf5b06ae481d2af7e66f6423f183f Mon Sep 17 00:00:00 2001
From: jwijenbergh <jeroenwijenbergh@protonmail.com>
Date: Tue, 3 May 2022 14:10:40 +0200
Subject: Refactor: Secure internet into a different type but with interface

---
 state_test.go | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

(limited to 'state_test.go')

diff --git a/state_test.go b/state_test.go
index c6e33e0..c174c72 100644
--- a/state_test.go
+++ b/state_test.go
@@ -161,29 +161,31 @@ func Test_token_expired(t *testing.T) {
 	_, configErr := state.ConnectInstituteAccess(serverURI)
 
 	if configErr != nil {
-		t.Errorf("Connect error before expired: %v", configErr)
+		t.Fatalf("Connect error before expired: %v", configErr)
 	}
 
 	server, serverErr := state.Servers.GetCurrentServer()
 	if serverErr != nil {
-		t.Errorf("No server found")
+		t.Fatalf("No server found")
 	}
 
-	accessToken := server.OAuth.Token.Access
-	refreshToken := server.OAuth.Token.Refresh
+	oauth := server.GetOAuth()
+
+	accessToken := oauth.Token.Access
+	refreshToken := oauth.Token.Refresh
 
 	// Wait for TTL so that the tokens expire
 	time.Sleep(time.Duration(expiredInt) * time.Second)
 
-	infoErr := server.APIInfo()
+	infoErr := internal.APIInfo(server)
 
 	if infoErr != nil {
 		t.Errorf("Info error after expired: %v", infoErr)
 	}
 
 	// Check if tokens have changed
-	accessTokenAfter := server.OAuth.Token.Access
-	refreshTokenAfter := server.OAuth.Token.Refresh
+	accessTokenAfter := oauth.Token.Access
+	refreshTokenAfter := oauth.Token.Refresh
 
 	if accessToken == accessTokenAfter {
 		t.Errorf("Access token is the same after refresh")
@@ -221,21 +223,23 @@ func Test_token_invalid(t *testing.T) {
 		return
 	}
 
+	oauth := server.GetOAuth()
+
 	// Override tokens with invalid values
-	server.OAuth.Token.Access = dummy_value
-	server.OAuth.Token.Refresh = dummy_value
+	oauth.Token.Access = dummy_value
+	oauth.Token.Refresh = dummy_value
 
-	infoErr := server.APIInfo()
+	infoErr := internal.APIInfo(server)
 
 	if infoErr != nil {
 		t.Errorf("Info error after invalid: %v", infoErr)
 	}
 
-	if server.OAuth.Token.Access == dummy_value {
+	if oauth.Token.Access == dummy_value {
 		t.Errorf("Access token is equal to dummy value: %s", dummy_value)
 	}
 
-	if server.OAuth.Token.Refresh == dummy_value {
+	if oauth.Token.Refresh == dummy_value {
 		t.Errorf("Refresh token is equal to dummy value: %s", dummy_value)
 	}
 }
-- 
cgit v1.2.3