tests: add tests for go/vt/hook (#15015)

Signed-off-by: Manik Rana <manikrana54@gmail.com>
Signed-off-by: Manik Rana <Manikrana54@gmail.com>
Co-authored-by: Florent Poinsard <35779988+frouioui@users.noreply.github.com>

Author: Maniktherana

go/vt/hook/hook_test.go

@@ -1,10 +1,29 @@
+/*
+Copyright 2024 The Vitess Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
 package hook
 
 import (
 	"context"
+	"io"
 	"os"
 	"os/exec"
 	"path"
+	"strings"
+	"sync"
 	"testing"
 	"time"
 
@@ -22,6 +41,11 @@ func TestExecuteContext(t *testing.T) {
 	require.NoError(t, err)
 
 	sleepHookPath := path.Join(vtroot, "vthook", "sleep")
+
+	if _, err := os.Lstat(sleepHookPath); err == nil {
+		require.NoError(t, os.Remove(sleepHookPath))
+	}
+
 	require.NoError(t, os.Symlink(sleep, sleepHookPath))
 	defer func() {
 		require.NoError(t, os.Remove(sleepHookPath))
@@ -38,3 +62,234 @@ func TestExecuteContext(t *testing.T) {
 	hr = h.Execute()
 	assert.Equal(t, HOOK_SUCCESS, hr.ExitStatus)
 }
+
+func TestExecuteOptional(t *testing.T) {
+	vtroot, err := vtenv.VtRoot()
+	require.NoError(t, err)
+
+	echo, err := exec.LookPath("echo")
+	require.NoError(t, err)
+
+	echoHookPath := path.Join(vtroot, "vthook", "echo")
+
+	if _, err := os.Lstat(echoHookPath); err == nil {
+		require.NoError(t, os.Remove(echoHookPath))
+	}
+
+	require.NoError(t, os.Symlink(echo, echoHookPath))
+	defer func() {
+		require.NoError(t, os.Remove(echoHookPath))
+	}()
+	tt := []struct {
+		name          string
+		hookName      string
+		parameters    []string
+		expectedError string
+	}{
+		{
+			name:       "HookSuccess",
+			hookName:   "echo",
+			parameters: []string{"test"},
+		},
+		{
+			name:       "HookDoesNotExist",
+			hookName:   "nonexistent-hook",
+			parameters: []string{"test"},
+		},
+	}
+
+	for _, tc := range tt {
+		t.Run(tc.name, func(t *testing.T) {
+			h := NewHook(tc.hookName, tc.parameters)
+			err := h.ExecuteOptional()
+			if tc.expectedError == "" {
+				assert.NoError(t, err)
+			} else {
+				assert.Error(t, err)
+				assert.ErrorContains(t, err, tc.expectedError)
+			}
+		})
+	}
+}
+
+func TestNewHook(t *testing.T) {
+	h := NewHook("test-hook", []string{"arg1", "arg2"})
+	assert.Equal(t, "test-hook", h.Name)
+	assert.Equal(t, []string{"arg1", "arg2"}, h.Parameters)
+}
+
+func TestNewSimpleHook(t *testing.T) {
+	h := NewSimpleHook("simple-hook")
+	assert.Equal(t, "simple-hook", h.Name)
+	assert.Empty(t, h.Parameters)
+}
+
+func TestNewHookWithEnv(t *testing.T) {
+	h := NewHookWithEnv("env-hook", []string{"arg1", "arg2"}, map[string]string{"KEY": "VALUE"})
+	assert.Equal(t, "env-hook", h.Name)
+	assert.Equal(t, []string{"arg1", "arg2"}, h.Parameters)
+	assert.Equal(t, map[string]string{"KEY": "VALUE"}, h.ExtraEnv)
+}
+
+func TestString(t *testing.T) {
+	tt := []struct {
+		name     string
+		input    HookResult
+		expected string
+	}{
+		{
+			name:     "HOOK_SUCCESS",
+			input:    HookResult{ExitStatus: HOOK_SUCCESS, Stdout: "output"},
+			expected: "result: HOOK_SUCCESS\nstdout:\noutput",
+		},
+		{
+			name:     "HOOK_DOES_NOT_EXIST",
+			input:    HookResult{ExitStatus: HOOK_DOES_NOT_EXIST},
+			expected: "result: HOOK_DOES_NOT_EXIST",
+		},
+		{
+			name:     "HOOK_STAT_FAILED",
+			input:    HookResult{ExitStatus: HOOK_STAT_FAILED},
+			expected: "result: HOOK_STAT_FAILED",
+		},
+		{
+			name:     "HOOK_CANNOT_GET_EXIT_STATUS",
+			input:    HookResult{ExitStatus: HOOK_CANNOT_GET_EXIT_STATUS},
+			expected: "result: HOOK_CANNOT_GET_EXIT_STATUS",
+		},
+		{
+			name:     "HOOK_INVALID_NAME",
+			input:    HookResult{ExitStatus: HOOK_INVALID_NAME},
+			expected: "result: HOOK_INVALID_NAME",
+		},
+		{
+			name:     "HOOK_VTROOT_ERROR",
+			input:    HookResult{ExitStatus: HOOK_VTROOT_ERROR},
+			expected: "result: HOOK_VTROOT_ERROR",
+		},
+		{
+			name:     "case default",
+			input:    HookResult{ExitStatus: 42},
+			expected: "result: exit(42)",
+		},
+		{
+			name:     "WithStderr",
+			input:    HookResult{ExitStatus: HOOK_SUCCESS, Stderr: "error"},
+			expected: "result: HOOK_SUCCESS\nstderr:\nerror",
+		},
+		{
+			name:     "WithStderr",
+			input:    HookResult{ExitStatus: HOOK_SUCCESS, Stderr: "error"},
+			expected: "result: HOOK_SUCCESS\nstderr:\nerror",
+		},
+		{
+			name:     "WithStdoutAndStderr",
+			input:    HookResult{ExitStatus: HOOK_SUCCESS, Stdout: "output", Stderr: "error"},
+			expected: "result: HOOK_SUCCESS\nstdout:\noutput\nstderr:\nerror",
+		},
+	}
+
+	for _, tc := range tt {
+		t.Run(tc.name, func(t *testing.T) {
+			result := tc.input.String()
+			assert.Equal(t, tc.expected, result)
+		})
+	}
+}
+
+func TestExecuteAsReadPipe(t *testing.T) {
+	vtroot, err := vtenv.VtRoot()
+	require.NoError(t, err)
+
+	cat, err := exec.LookPath("cat")
+	require.NoError(t, err)
+
+	catHookPath := path.Join(vtroot, "vthook", "cat")
+
+	if _, err := os.Lstat(catHookPath); err == nil {
+		require.NoError(t, os.Remove(catHookPath))
+	}
+
+	require.NoError(t, os.Symlink(cat, catHookPath))
+	defer func() {
+		require.NoError(t, os.Remove(catHookPath))
+	}()
+
+	h := NewHook("cat", nil)
+	reader, waitFunc, status, err := h.ExecuteAsReadPipe(strings.NewReader("Hello, World!\n"))
+	require.NoError(t, err)
+	defer reader.(io.Closer).Close()
+
+	output, err := io.ReadAll(reader)
+	require.NoError(t, err)
+	assert.Equal(t, "Hello, World!\n", string(output))
+
+	stderr, waitErr := waitFunc()
+	assert.Empty(t, stderr)
+	assert.NoError(t, waitErr)
+	assert.Equal(t, HOOK_SUCCESS, status)
+}
+
+func TestExecuteAsReadPipeErrorFindingHook(t *testing.T) {
+	h := NewHook("nonexistent-hook", nil)
+	reader, waitFunc, status, err := h.ExecuteAsReadPipe(strings.NewReader("Hello, World!\n"))
+	require.Error(t, err)
+	assert.Nil(t, reader)
+	assert.Nil(t, waitFunc)
+	assert.Equal(t, HOOK_DOES_NOT_EXIST, status)
+}
+
+func TestExecuteAsWritePipe(t *testing.T) {
+	var writer strings.Builder
+	var writerMutex sync.Mutex
+
+	vtroot, err := vtenv.VtRoot()
+	require.NoError(t, err)
+
+	echo, err := exec.LookPath("echo")
+	require.NoError(t, err)
+
+	echoHookPath := path.Join(vtroot, "vthook", "echo")
+
+	if _, err := os.Lstat(echoHookPath); err == nil {
+		require.NoError(t, os.Remove(echoHookPath))
+	}
+
+	require.NoError(t, os.Symlink(echo, echoHookPath))
+	defer func() {
+		require.NoError(t, os.Remove(echoHookPath))
+	}()
+
+	h := NewHook("echo", nil)
+
+	writerMutex.Lock()
+	var writerTemp strings.Builder
+	_, waitFunc, status, err := h.ExecuteAsWritePipe(&writerTemp)
+	writerMutex.Unlock()
+
+	require.NoError(t, err)
+	defer func() {
+		writerMutex.Lock()
+		writer.Reset()
+		writerMutex.Unlock()
+	}()
+
+	writerMutex.Lock()
+	_, err = writer.Write([]byte("Hello, World!\n"))
+	writerMutex.Unlock()
+	require.NoError(t, err)
+
+	stderr, waitErr := waitFunc()
+	assert.Empty(t, stderr)
+	assert.NoError(t, waitErr)
+	assert.Equal(t, HOOK_SUCCESS, status)
+}
+
+func TestExecuteAsWritePipeErrorFindingHook(t *testing.T) {
+	h := NewHook("nonexistent-hook", nil)
+	var writer strings.Builder
+	writerPtr := &writer
+	_, _, status, err := h.ExecuteAsWritePipe(writerPtr)
+	assert.Error(t, err)
+	assert.Equal(t, HOOK_DOES_NOT_EXIST, status)
+}