From 59555e1ed298cf8d42dbed6627a825e9abb8dae2 Mon Sep 17 00:00:00 2001
From: Bharath Gajjala <120367134+bgajjala8@users.noreply.github.com>
Date: Mon, 30 Jun 2025 23:36:49 -0500
Subject: [PATCH] adding error checks

---
 .../e2e/tests/base_plus/target_tcp_connect_mysql_test.go    | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/testing/internal/e2e/tests/base_plus/target_tcp_connect_mysql_test.go b/testing/internal/e2e/tests/base_plus/target_tcp_connect_mysql_test.go
index be846bc55d..3b69c7f313 100644
--- a/testing/internal/e2e/tests/base_plus/target_tcp_connect_mysql_test.go
+++ b/testing/internal/e2e/tests/base_plus/target_tcp_connect_mysql_test.go
@@ -34,13 +34,15 @@ func TestCliTcpTargetConnectMysql(t *testing.T) {
 	network, err := pool.CreateNetwork("e2e-mysql-test")
 	require.NoError(t, err)
 	t.Cleanup(func() {
-		pool.RemoveNetwork(network)
+		err := pool.RemoveNetwork(network)
+		require.NoError(t, err, "Failed to remove network")
 	})
 
 	// Start MySQL container (using official MySQL image)
 	mysqlContainer := infra.StartMysql(t, pool, network, "mysql", "8.0")
 	t.Cleanup(func() {
-		pool.Purge(mysqlContainer.Resource)
+		err := pool.Purge(mysqlContainer.Resource)
+		require.NoError(t, err, "Failed to purge MySQL container")
 	})
 
 	// Wait for MySQL to be ready