Merge pull request #427 from bruin-data/mysql-conn-test

Mysql conn test

pkg/mysql/db.go

@@ -157,3 +157,16 @@ func (c *Client) SelectWithSchema(ctx context.Context, queryObj *query.Query) (*
 
 	return result, nil
 }
+
+func (c *Client) Ping(ctx context.Context) error {
+	q := query.Query{
+		Query: "SELECT 1",
+	}
+
+	err := c.RunQueryWithoutResult(ctx, &q)
+	if err != nil {
+		return errors.Wrap(err, "failed to run test query on MySQL connection")
+	}
+
+	return nil
+}

pkg/mysql/db_test.go

@@ -174,3 +174,56 @@ func TestClient_SelectWithSchema(t *testing.T) {
 		})
 	}
 }
+
+func TestClient_Ping(t *testing.T) {
+	t.Parallel()
+
+	tests := []struct {
+		name           string
+		mockConnection func(mock sqlmock.Sqlmock)
+		wantErr        bool
+		errorMessage   string
+	}{
+		{
+			name: "successful ping",
+			mockConnection: func(mock sqlmock.Sqlmock) {
+				mock.ExpectExec(`SELECT 1`).
+					WillReturnResult(sqlmock.NewResult(0, 0))
+			},
+			wantErr: false,
+		},
+		{
+			name: "failed ping",
+			mockConnection: func(mock sqlmock.Sqlmock) {
+				mock.ExpectExec(`SELECT 1`).
+					WillReturnError(errors.New("connection refused"))
+			},
+			wantErr:      true,
+			errorMessage: "failed to run test query on MySQL connection: failed to execute query: connection refused",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			t.Parallel()
+
+			mockDB, mock, err := sqlmock.New(sqlmock.QueryMatcherOption(sqlmock.QueryMatcherEqual))
+			require.NoError(t, err)
+			defer mockDB.Close()
+			sqlxDB := sqlx.NewDb(mockDB, "sqlmock")
+
+			tt.mockConnection(mock)
+			client := Client{conn: sqlxDB}
+
+			err = client.Ping(context.Background())
+			if tt.wantErr {
+				require.Error(t, err)
+				assert.Equal(t, tt.errorMessage, err.Error())
+			} else {
+				require.NoError(t, err)
+			}
+
+			require.NoError(t, mock.ExpectationsWereMet())
+		})
+	}
+}