Add test for Reconnect

Fix tests

cmd/run_test.go

@@ -165,7 +165,7 @@ func Test_runCmdWithCachePlugin(t *testing.T) {
 	var waitGroup sync.WaitGroup
 	waitGroup.Add(1)
 	go func(waitGroup *sync.WaitGroup) {
-		time.Sleep(500 * time.Millisecond)
+		time.Sleep(time.Second)
 
 		StopGracefully(
 			runCmd.Context(),

network/client_test.go

@@ -49,7 +49,7 @@ func TestNewClient(t *testing.T) {
 	assert.Equal(t, "tcp", client.Network)
 	assert.Equal(t, "127.0.0.1:5432", client.Address)
 	assert.NotEmpty(t, client.ID)
-	assert.NotNil(t, client.Conn)
+	assert.NotNil(t, client.conn)
 }
 
 // TestSend tests the Send function.
@@ -96,7 +96,7 @@ func TestClose(t *testing.T) {
 	assert.Equal(t, "", client.ID)
 	assert.Equal(t, "", client.Network)
 	assert.Equal(t, "", client.Address)
-	assert.Nil(t, client.Conn)
+	assert.Nil(t, client.conn)
 }
 
 // TestIsConnected tests the IsConnected function.
@@ -108,6 +108,24 @@ func TestIsConnected(t *testing.T) {
 	assert.False(t, client.IsConnected())
 }
 
+func TestReconnect(t *testing.T) {
+	client := CreateNewClient(t)
+	defer client.Close()
+
+	assert.NotNil(t, client)
+	assert.NotNil(t, client.conn)
+	assert.NotEmpty(t, client.ID)
+	localAddr := client.LocalAddr()
+	assert.NotEmpty(t, localAddr)
+
+	assert.NoError(t, client.Reconnect())
+	assert.NotNil(t, client)
+	assert.NotNil(t, client.conn)
+	assert.NotEmpty(t, client.ID)
+	assert.NotEmpty(t, client.LocalAddr())
+	assert.NotEqual(t, localAddr, client.LocalAddr()) // This is a new connection.
+}
+
 func BenchmarkNewClient(b *testing.B) {
 	cfg := logging.LoggerConfig{
 		Output:            []config.LogOutput{config.Console},