Пример #1
0
func TestReadEmptyDirectory(t *testing.T) {
	var dir string
	var err error

	if dir, err = ioutil.TempDir(os.TempDir(), "TestReadEmptyDirectory"); err != nil {
		t.Fail()
		return
	}

	defer os.Remove(dir)

	assert.Len(t, fs.ReadDir(dir), 0)
}
Пример #2
0
func TestReadDir(t *testing.T) {
	var dir string
	var err error

	if dir, err = ioutil.TempDir(os.TempDir(), "TestReadDir"); err != nil {
		t.Fail()
		return
	}

	defer os.RemoveAll(dir)

	ioutil.TempFile(dir, "1.mkv")
	ioutil.TempFile(dir, "2.mp4")
	ioutil.TempFile(dir, "3.avi")

	assert.Len(t, fs.ReadDir(dir), 3)
}
Пример #3
0
func ExtractFilesFromArgs(args []string) (files []string) {
	for _, arg := range args {
		fi, err := os.Stat(arg)

		if err != nil {
			continue
		}

		abs, err := filepath.Abs(arg)

		if err != nil {
			continue
		}

		if fi.IsDir() {
			files = append(files, fs.ReadDir(abs)...)
			continue
		}

		files = append(files, abs)
	}

	return files
}