From e815dcfb1aa1230c714d306cf246e5ce20d2e48c Mon Sep 17 00:00:00 2001
From: jwijenbergh <jeroenwijenbergh@protonmail.com>
Date: Wed, 17 Jul 2024 11:33:04 +0200
Subject: Client: Add test for getting ServerList in the main state

---
 client/client_test.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/client/client_test.go b/client/client_test.go
index a84d87e..b12919e 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -77,11 +77,19 @@ func TestServer(t *testing.T) {
 	ck := cookie.NewWithContext(context.Background())
 	defer ck.Cancel() //nolint:errcheck
 	dir := t.TempDir()
+	var state *Client
 	state, err := New(
 		"org.letsconnect-vpn.app.linux",
 		"0.1.0-test",
 		dir,
 		func(old FSMStateID, new FSMStateID, data interface{}) bool {
+			// test if main server server list succeeds
+			if new == StateMain {
+				_, listErr := state.ServerList()
+				if listErr != nil {
+					t.Fatalf("Got server list error: %v", listErr)
+				}
+			}
 			go stateCallback(ck, old, new, data)
 			return true
 		},
-- 
cgit v1.2.3