feat: zoxide tests

Author: lazysegtree

src/internal/model_test.go

@@ -8,6 +8,7 @@ import (
 	"runtime"
 	"testing"
 
+	tea "github.com/charmbracelet/bubbletea"
 	zoxidelib "github.com/lazysegtree/go-zoxide"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
@@ -248,6 +249,7 @@ func TestChooserFile(t *testing.T) {
 	}
 }
 
+//nolint:gocognit,maintidx // Integration test with multiple subtests for comprehensive zoxide functionality
 func TestZoxide(t *testing.T) {
 	zoxideDataDir := t.TempDir()
 	zClient, err := zoxidelib.New(zoxidelib.WithDataDir(zoxideDataDir))
@@ -354,4 +356,156 @@ func TestZoxide(t *testing.T) {
 		results := m.zoxideModal.GetResults()
 		assert.Empty(t, results, "Zoxide modal should show no results when ZoxideSupport is disabled")
 	})
+
+	t.Run("Zoxide modal size on window resize", func(t *testing.T) {
+		common.Config.ZoxideSupport = true
+		m := defaultTestModelWithZClient(zClient, dir1)
+		p := NewTestTeaProgWithEventLoop(t, m)
+
+		p.SendKey(common.Hotkeys.OpenZoxide[0])
+		assert.Eventually(t, func() bool {
+			return p.getModel().zoxideModal.IsOpen()
+		}, DefaultTestTimeout, DefaultTestTick, "Zoxide modal should open")
+
+		initialWidth := p.getModel().zoxideModal.GetWidth()
+		initialMaxHeight := p.getModel().zoxideModal.GetMaxHeight()
+
+		newWidth := 4 * common.MinimumWidth
+		newHeight := 4 * common.MinimumHeight
+		p.Send(tea.WindowSizeMsg{Width: newWidth, Height: newHeight})
+
+		assert.Eventually(t, func() bool {
+			updatedWidth := p.getModel().zoxideModal.GetWidth()
+			updatedMaxHeight := p.getModel().zoxideModal.GetMaxHeight()
+			return updatedWidth != initialWidth && updatedMaxHeight != initialMaxHeight
+		}, DefaultTestTimeout, DefaultTestTick, "Zoxide modal dimensions should update on window resize")
+
+		updatedWidth := p.getModel().zoxideModal.GetWidth()
+		updatedMaxHeight := p.getModel().zoxideModal.GetMaxHeight()
+		assert.Greater(t, updatedWidth, initialWidth, "Width should increase with larger window")
+		assert.Greater(t, updatedMaxHeight, initialMaxHeight, "MaxHeight should increase with larger window")
+	})
+
+	t.Run("Zoxide 'z' key suppression on open", func(t *testing.T) {
+		common.Config.ZoxideSupport = true
+		m := defaultTestModelWithZClient(zClient, dir1)
+		p := NewTestTeaProgWithEventLoop(t, m)
+
+		p.SendKey(common.Hotkeys.OpenZoxide[0])
+		assert.Eventually(t, func() bool {
+			return p.getModel().zoxideModal.IsOpen()
+		}, DefaultTestTimeout, DefaultTestTick, "Zoxide modal should open")
+
+		assert.Eventually(t, func() bool {
+			return p.getModel().zoxideModal.GetTextInputValue() == ""
+		}, DefaultTestTimeout, DefaultTestTick, "The 'z' key should not be added to textInput")
+
+		p.SendKey("a")
+		p.SendKey("b")
+		p.SendKey("c")
+
+		assert.Eventually(t, func() bool {
+			return p.getModel().zoxideModal.GetTextInputValue() == "abc"
+		}, DefaultTestTimeout, DefaultTestTick, "Subsequent keys should be added to textInput")
+	})
+
+	t.Run("Multi-space directory name navigation", func(t *testing.T) {
+		common.Config.ZoxideSupport = true
+		multiSpaceDir := filepath.Join(curTestDir, "test  dir")
+		utils.SetupDirectories(t, multiSpaceDir)
+		defer os.RemoveAll(multiSpaceDir)
+
+		m := defaultTestModelWithZClient(zClient, dir1)
+		p := NewTestTeaProgWithEventLoop(t, m)
+
+		err := p.getModel().updateCurrentFilePanelDir(multiSpaceDir)
+		require.NoError(t, err, "Failed to navigate to multi-space directory")
+
+		err = p.getModel().updateCurrentFilePanelDir(dir1)
+		require.NoError(t, err, "Failed to navigate back to dir1")
+
+		p.SendKey(common.Hotkeys.OpenZoxide[0])
+		assert.Eventually(t, func() bool {
+			return p.getModel().zoxideModal.IsOpen()
+		}, DefaultTestTimeout, DefaultTestTick, "Zoxide modal should open")
+
+		for _, char := range "test  dir" {
+			p.SendKey(string(char))
+		}
+
+		assert.Eventually(t, func() bool {
+			results := p.getModel().zoxideModal.GetResults()
+			for _, result := range results {
+				if result.Path == multiSpaceDir {
+					return true
+				}
+			}
+			return false
+		}, DefaultTestTimeout, DefaultTestTick, "Multi-space directory should be found by zoxide")
+
+		results := p.getModel().zoxideModal.GetResults()
+		multiSpaceDirIndex := -1
+		for i, result := range results {
+			if result.Path == multiSpaceDir {
+				multiSpaceDirIndex = i
+				break
+			}
+		}
+		require.NotEqual(t, -1, multiSpaceDirIndex, "Multi-space directory should be in results")
+
+		for range multiSpaceDirIndex {
+			p.SendKey(common.Hotkeys.ListDown[0])
+		}
+
+		p.SendKey(common.Hotkeys.ConfirmTyping[0])
+		assert.Eventually(t, func() bool {
+			return p.getModel().getFocusedFilePanel().location == multiSpaceDir
+		}, DefaultTestTimeout, DefaultTestTick, "Should navigate to multi-space directory")
+	})
+
+	t.Run("Zoxide escape key closes modal", func(t *testing.T) {
+		common.Config.ZoxideSupport = true
+		m := defaultTestModelWithZClient(zClient, dir1)
+		p := NewTestTeaProgWithEventLoop(t, m)
+
+		p.SendKey(common.Hotkeys.OpenZoxide[0])
+		assert.Eventually(t, func() bool {
+			return p.getModel().zoxideModal.IsOpen()
+		}, DefaultTestTimeout, DefaultTestTick, "Zoxide modal should open")
+
+		p.SendKey(common.Hotkeys.CancelTyping[0])
+		assert.Eventually(t, func() bool {
+			return !p.getModel().zoxideModal.IsOpen()
+		}, DefaultTestTimeout, DefaultTestTick, "Zoxide modal should close on escape key")
+	})
+
+	t.Run("Zoxide with invalid/non-existent directory", func(t *testing.T) {
+		common.Config.ZoxideSupport = true
+		m := defaultTestModelWithZClient(zClient, dir1)
+		p := NewTestTeaProgWithEventLoop(t, m)
+
+		currentLocation := p.getModel().getFocusedFilePanel().location
+
+		p.SendKey(common.Hotkeys.OpenZoxide[0])
+		assert.Eventually(t, func() bool {
+			return p.getModel().zoxideModal.IsOpen()
+		}, DefaultTestTimeout, DefaultTestTick, "Zoxide modal should open")
+
+		for _, char := range "uniquexyzabc123" {
+			p.SendKey(string(char))
+		}
+
+		assert.Eventually(t, func() bool {
+			results := p.getModel().zoxideModal.GetResults()
+			return len(results) == 0
+		}, DefaultTestTimeout, DefaultTestTick, "Should have no results for unique search")
+
+		p.SendKey(common.Hotkeys.ConfirmTyping[0])
+		assert.Eventually(t, func() bool {
+			return !p.getModel().zoxideModal.IsOpen()
+		}, DefaultTestTimeout, DefaultTestTick, "Zoxide modal should close")
+
+		assert.Equal(t, currentLocation, p.getModel().getFocusedFilePanel().location,
+			"Should stay in current location when confirming with no results")
+	})
 }

src/internal/ui/zoxide/utils.go

@@ -64,6 +64,10 @@ func (m *Model) GetResults() []zoxidelib.Result {
 	return out
 }
 
+func (m *Model) GetTextInputValue() string {
+	return m.textInput.Value()
+}
+
 func isKeyAlphaNum(msg tea.KeyMsg) bool {
 	r := []rune(msg.String())
 	if len(r) != 1 {