From 80488fa7c86ae4999f36fd3c26fa8d65a6833159 Mon Sep 17 00:00:00 2001
From: Erik Brakkee <erik@brakkee.org>
Date: Wed, 21 Aug 2024 23:52:06 +0200
Subject: [PATCH] test for initialization failure when connection is closed.

---
 pkg/comms/agentserver_test.go | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/pkg/comms/agentserver_test.go b/pkg/comms/agentserver_test.go
index 4214f2a..52d13b9 100644
--- a/pkg/comms/agentserver_test.go
+++ b/pkg/comms/agentserver_test.go
@@ -188,8 +188,36 @@ func (s *AgentServerTestSuite) Test_InitializationProtocolVersionMismatch() {
 		})
 }
 
+func (s *AgentServerTestSuite) Test_InitializationAgentConnectionClosed() {
+	s.agentConnection.Close()
+	s.checkInitializationFailure()
+}
+
+func (s *AgentServerTestSuite) Test_InitializationServerConnectionClosed() {
+	s.serverConnection.Close()
+	s.checkInitializationFailure()
+}
+
+func (s *AgentServerTestSuite) checkInitializationFailure() []any {
+	return testsupport.RunAndWait(
+		&s.Suite,
+		func() any {
+			serverInfo, err := AgentInitialization(s.agentConnection,
+				NewEnvironmentInfo("myshell"))
+			s.NotNil(err)
+			s.Equal(ServerInfo{}, serverInfo)
+			return nil
+		},
+		func() any {
+			serverInfo := ServerInfo{TestId: 1000}
+			environmentInfo, err := ServerInitialization(s.serverConnection, serverInfo)
+			s.NotNil(err)
+			s.Equal(EnvironmentInfo{}, environmentInfo)
+			return nil
+		})
+}
+
 // TODO:
-// Test for protocolversion mismatch: verify error is returned. Use sentinel error
 // Tests when connection is close from agent and from server: verify error is returned
 
 func (s *AgentServerTestSuite) Test_ListenForAgentEvents() {