mirror of
https://gitee.com/johng/gf.git
synced 2024-12-02 20:28:17 +08:00
68 lines
2.0 KiB
Go
68 lines
2.0 KiB
Go
// Copyright 2017-2018 gf Author(https://github.com/gogf/gf). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package gfile_test
|
|
|
|
import (
|
|
"github.com/gogf/gf/debug/gdebug"
|
|
"testing"
|
|
|
|
"github.com/gogf/gf/os/gfile"
|
|
|
|
"github.com/gogf/gf/test/gtest"
|
|
)
|
|
|
|
func Test_ScanDir(t *testing.T) {
|
|
teatPath := gdebug.TestDataPath()
|
|
gtest.C(t, func(t *gtest.T) {
|
|
files, err := gfile.ScanDir(teatPath, "*", false)
|
|
t.Assert(err, nil)
|
|
t.AssertIN(teatPath+gfile.Separator+"dir1", files)
|
|
t.AssertIN(teatPath+gfile.Separator+"dir2", files)
|
|
t.AssertNE(teatPath+gfile.Separator+"dir1"+gfile.Separator+"file1", files)
|
|
})
|
|
gtest.C(t, func(t *gtest.T) {
|
|
files, err := gfile.ScanDir(teatPath, "*", true)
|
|
t.Assert(err, nil)
|
|
t.AssertIN(teatPath+gfile.Separator+"dir1", files)
|
|
t.AssertIN(teatPath+gfile.Separator+"dir2", files)
|
|
t.AssertIN(teatPath+gfile.Separator+"dir1"+gfile.Separator+"file1", files)
|
|
t.AssertIN(teatPath+gfile.Separator+"dir2"+gfile.Separator+"file2", files)
|
|
})
|
|
}
|
|
|
|
func Test_ScanDirFunc(t *testing.T) {
|
|
teatPath := gdebug.TestDataPath()
|
|
gtest.C(t, func(t *gtest.T) {
|
|
files, err := gfile.ScanDirFunc(teatPath, "*", true, func(path string) string {
|
|
if gfile.Name(path) != "file1" {
|
|
return ""
|
|
}
|
|
return path
|
|
})
|
|
t.Assert(err, nil)
|
|
t.Assert(len(files), 1)
|
|
t.Assert(gfile.Name(files[0]), "file1")
|
|
})
|
|
}
|
|
|
|
func Test_ScanDirFile(t *testing.T) {
|
|
teatPath := gdebug.TestDataPath()
|
|
gtest.C(t, func(t *gtest.T) {
|
|
files, err := gfile.ScanDirFile(teatPath, "*", false)
|
|
t.Assert(err, nil)
|
|
t.Assert(len(files), 0)
|
|
})
|
|
gtest.C(t, func(t *gtest.T) {
|
|
files, err := gfile.ScanDirFile(teatPath, "*", true)
|
|
t.Assert(err, nil)
|
|
t.AssertNI(teatPath+gfile.Separator+"dir1", files)
|
|
t.AssertNI(teatPath+gfile.Separator+"dir2", files)
|
|
t.AssertIN(teatPath+gfile.Separator+"dir1"+gfile.Separator+"file1", files)
|
|
t.AssertIN(teatPath+gfile.Separator+"dir2"+gfile.Separator+"file2", files)
|
|
})
|
|
}
|