dagu/internal/dagstore/grep/grep_test.go

package grep

import (
	"os"
	"path/filepath"
	"testing"

	"github.com/dagu-org/dagu/internal/dagstore"
	"github.com/stretchr/testify/require"
)

func TestGrep(t *testing.T) {
	t.Parallel()
	wd, _ := os.Getwd()
	dir := filepath.Join(wd, "/testdata")
	for _, tc := range []struct {
		Name    string
		File    string
		Pattern string
		Opts    GrepOptions
		Want    []*dagstore.Match
		IsErr   bool
	}{
		{
			Name:    "simple",
			File:    filepath.Join(dir, "test.txt"),
			Pattern: "b",
			Want: []*dagstore.Match{
				{
					LineNumber: 2,
					StartLine:  2,
					Line:       "bb",
				}},
		},
		{
			Name:    "regexp",
			File:    filepath.Join(dir, "test.txt"),
			Pattern: "^b.",
			Opts: GrepOptions{
				IsRegexp: true,
			},
			Want: []*dagstore.Match{
				{
					LineNumber: 2,
					StartLine:  2,
					Line:       "bb",
				}},
		},
		{
			Name:    "before",
			File:    filepath.Join(dir, "test.txt"),
			Pattern: "b",
			Opts: GrepOptions{
				Before: 1,
			},
			Want: []*dagstore.Match{
				{
					LineNumber: 2,
					StartLine:  1,
					Line:       "aa\nbb",
				}},
		},
		{
			Name:    "before+after",
			File:    filepath.Join(dir, "test.txt"),
			Pattern: "cc",
			Opts: GrepOptions{
				Before: 2,
				After:  2,
			},
			Want: []*dagstore.Match{
				{
					LineNumber: 3,
					StartLine:  1,
					Line:       "aa\nbb\ncc\ndd\nee",
				}},
		},
		{
			Name:    "before+after,firstline",
			File:    filepath.Join(dir, "test.txt"),
			Pattern: "aa",
			Opts: GrepOptions{
				Before: 1,
				After:  1,
			},
			Want: []*dagstore.Match{
				{
					LineNumber: 1,
					StartLine:  1,
					Line:       "aa\nbb",
				}},
		},
		{
			Name:    "before+after,lastline",
			File:    filepath.Join(dir, "test.txt"),
			Pattern: "ee",
			Opts: GrepOptions{
				Before: 1,
				After:  1,
			},
			Want: []*dagstore.Match{
				{
					LineNumber: 5,
					StartLine:  4,
					Line:       "dd\nee",
				}},
		},
		{
			Name:    "no match",
			File:    filepath.Join(dir, "test.txt"),
			Pattern: "no match text",
			IsErr:   true,
		},
		{
			Name:    "no file",
			File:    filepath.Join(dir, "dummy.txt"),
			Pattern: "aa",
			IsErr:   true,
		},
		{
			Name:    "no pattern",
			File:    filepath.Join(dir, "test.txt"),
			Pattern: "",
			IsErr:   true,
		},
		{
			Name:    "invalid regexp",
			File:    filepath.Join(dir, "test.txt"),
			Pattern: "(aa",
			Opts: GrepOptions{
				IsRegexp: true,
			},
			IsErr: true,
		},
	} {
		t.Run(tc.Name, func(t *testing.T) {
			dat, _ := os.ReadFile(tc.File)
			ret, err := Grep(dat, tc.Pattern, tc.Opts)
			if tc.IsErr {
				require.Empty(t, ret)
				require.Error(t, err)
				return
			}
			require.NoError(t, err)
			require.Equal(t, tc.Want, ret)
		})
	}
}