crowdsec/pkg/csplugin/listfiles_test.go

package csplugin

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

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/crowdsecurity/go-cs-lib/cstest"
)

func TestListFilesAtPath(t *testing.T) {
	dir, err := os.MkdirTemp("", "test-listfiles")
	require.NoError(t, err)
	t.Cleanup(func() {
		os.RemoveAll(dir)
	})
	_, err = os.Create(filepath.Join(dir, "notification-gitter"))
	require.NoError(t, err)
	_, err = os.Create(filepath.Join(dir, "slack"))
	require.NoError(t, err)
	err = os.Mkdir(filepath.Join(dir, "somedir"), 0755)
	require.NoError(t, err)
	_, err = os.Create(filepath.Join(dir, "somedir", "inner"))
	require.NoError(t, err)

	tests := []struct {
		name        string
		path        string
		want        []string
		expectedErr string
	}{
		{
			name: "valid directory",
			path: dir,
			want: []string{
				filepath.Join(dir, "notification-gitter"),
				filepath.Join(dir, "slack"),
			},
		},
		{
			name:        "invalid directory",
			path:        "./foo/bar/",
			expectedErr: "open ./foo/bar/: " + cstest.PathNotFoundMessage,
		},
	}
	for _, tc := range tests {
		tc := tc
		t.Run(tc.name, func(t *testing.T) {
			got, err := listFilesAtPath(tc.path)
			cstest.RequireErrorContains(t, err, tc.expectedErr)
			assert.ElementsMatch(t, tc.want, got)
		})
	}
}