From f76e87fa5ec30c416a562c4b29689f6a3e29458c Mon Sep 17 00:00:00 2001
From: jwijenbergh <jeroenwijenbergh@protonmail.com>
Date: Fri, 22 Apr 2022 10:55:37 +0200
Subject: Servers: Add initial support for multiple using a map

---
 src/server_test.go | 30 ++++++++++++++++++++----------
 1 file changed, 20 insertions(+), 10 deletions(-)

(limited to 'src/server_test.go')

diff --git a/src/server_test.go b/src/server_test.go
index 7081bde..ccf58f6 100644
--- a/src/server_test.go
+++ b/src/server_test.go
@@ -134,21 +134,26 @@ func Test_token_expired(t *testing.T) {
 		t.Errorf("Connect error before expired: %v", configErr)
 	}
 
-	accessToken := state.Server.OAuth.Token.Access
-	refreshToken := state.Server.OAuth.Token.Refresh
+	server, serverErr := state.Servers.GetCurrentServer()
+	if serverErr != nil {
+		t.Errorf("No server found")
+	}
+
+	accessToken := server.OAuth.Token.Access
+	refreshToken := server.OAuth.Token.Refresh
 
 	// Wait for TTL so that the tokens expire
 	time.Sleep(time.Duration(expiredInt) * time.Second)
 
-	infoErr := state.Server.APIInfo()
+	infoErr := server.APIInfo()
 
 	if infoErr != nil {
 		t.Errorf("Info error after expired: %v", infoErr)
 	}
 
 	// Check if tokens have changed
-	accessTokenAfter := state.Server.OAuth.Token.Access
-	refreshTokenAfter := state.Server.OAuth.Token.Refresh
+	accessTokenAfter := server.OAuth.Token.Access
+	refreshTokenAfter := server.OAuth.Token.Refresh
 
 	if accessToken == accessTokenAfter {
 		t.Errorf("Access token is the same after refresh")
@@ -184,21 +189,26 @@ func Test_token_invalid(t *testing.T) {
 
 	dummy_value := "37"
 
+	server, serverErr := state.Servers.GetCurrentServer()
+	if serverErr != nil {
+		t.Errorf("No server found")
+	}
+
 	// Override tokens with invalid values
-	state.Server.OAuth.Token.Access = dummy_value
-	state.Server.OAuth.Token.Refresh = dummy_value
+	server.OAuth.Token.Access = dummy_value
+	server.OAuth.Token.Refresh = dummy_value
 
-	infoErr := state.Server.APIInfo()
+	infoErr := server.APIInfo()
 
 	if infoErr != nil {
 		t.Errorf("Info error after invalid: %v", infoErr)
 	}
 
-	if state.Server.OAuth.Token.Access == dummy_value {
+	if 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 {
+	if server.OAuth.Token.Refresh == dummy_value {
 		t.Errorf("Refresh token is equal to dummy value: %s", dummy_value)
 	}
 }
-- 
cgit v1.2.3