100 lines
2.6 KiB
Go
100 lines
2.6 KiB
Go
package filecount
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/influxdata/telegraf/internal"
|
|
"github.com/influxdata/telegraf/testutil"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestNoFilters(t *testing.T) {
|
|
fc := getNoFilterFileCount()
|
|
matches := []string{"foo", "bar", "baz", "qux",
|
|
"subdir/", "subdir/quux", "subdir/quuz"}
|
|
require.True(t, fileCountEquals(fc, len(matches)))
|
|
}
|
|
|
|
func TestNameFilter(t *testing.T) {
|
|
fc := getNoFilterFileCount()
|
|
fc.Name = "ba*"
|
|
matches := []string{"bar", "baz"}
|
|
require.True(t, fileCountEquals(fc, len(matches)))
|
|
}
|
|
|
|
func TestNonRecursive(t *testing.T) {
|
|
fc := getNoFilterFileCount()
|
|
fc.Recursive = false
|
|
matches := []string{"foo", "bar", "baz", "qux", "subdir"}
|
|
require.True(t, fileCountEquals(fc, len(matches)))
|
|
}
|
|
|
|
func TestRegularOnlyFilter(t *testing.T) {
|
|
fc := getNoFilterFileCount()
|
|
fc.RegularOnly = true
|
|
matches := []string{
|
|
"foo", "bar", "baz", "qux", "subdir/quux", "subdir/quuz",
|
|
}
|
|
require.True(t, fileCountEquals(fc, len(matches)))
|
|
}
|
|
|
|
func TestSizeFilter(t *testing.T) {
|
|
fc := getNoFilterFileCount()
|
|
fc.Size = -100
|
|
matches := []string{"foo", "bar", "baz",
|
|
"subdir/quux", "subdir/quuz"}
|
|
require.True(t, fileCountEquals(fc, len(matches)))
|
|
|
|
fc.Size = 100
|
|
matches = []string{"qux"}
|
|
require.True(t, fileCountEquals(fc, len(matches)))
|
|
}
|
|
|
|
func TestMTimeFilter(t *testing.T) {
|
|
oldFile := filepath.Join(getTestdataDir(), "baz")
|
|
mtime := time.Date(1979, time.December, 14, 18, 25, 5, 0, time.UTC)
|
|
if err := os.Chtimes(oldFile, mtime, mtime); err != nil {
|
|
t.Skip("skipping mtime filter test.")
|
|
}
|
|
fileAge := time.Since(mtime) - (60 * time.Second)
|
|
|
|
fc := getNoFilterFileCount()
|
|
fc.MTime = internal.Duration{Duration: -fileAge}
|
|
matches := []string{"foo", "bar", "qux",
|
|
"subdir/", "subdir/quux", "subdir/quuz"}
|
|
require.True(t, fileCountEquals(fc, len(matches)))
|
|
|
|
fc.MTime = internal.Duration{Duration: fileAge}
|
|
matches = []string{"baz"}
|
|
require.True(t, fileCountEquals(fc, len(matches)))
|
|
}
|
|
|
|
func getNoFilterFileCount() FileCount {
|
|
return FileCount{
|
|
Directory: getTestdataDir(),
|
|
Name: "*",
|
|
Recursive: true,
|
|
RegularOnly: false,
|
|
Size: 0,
|
|
MTime: internal.Duration{Duration: 0},
|
|
fileFilters: nil,
|
|
}
|
|
}
|
|
|
|
func getTestdataDir() string {
|
|
_, filename, _, _ := runtime.Caller(1)
|
|
return strings.Replace(filename, "filecount_test.go", "testdata/", 1)
|
|
}
|
|
|
|
func fileCountEquals(fc FileCount, expectedCount int) bool {
|
|
tags := map[string]string{"directory": getTestdataDir()}
|
|
acc := testutil.Accumulator{}
|
|
acc.GatherError(fc.Gather)
|
|
return acc.HasPoint("filecount", tags, "count", int64(expectedCount))
|
|
}
|