30 lines
459 B
Go
30 lines
459 B
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"strconv"
|
|
"testing"
|
|
)
|
|
|
|
func TestListDir(t *testing.T) {
|
|
d := t.TempDir()
|
|
want := []string{}
|
|
for i := 0; i < 3; i++ {
|
|
p := path.Join(d, strconv.Itoa(i))
|
|
ioutil.WriteFile(p, []byte{}, os.ModePerm)
|
|
want = append(want, p)
|
|
}
|
|
os.Mkdir(path.Join(d, "d"), os.ModePerm)
|
|
|
|
got, err := listDir(d)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if fmt.Sprint(want) != fmt.Sprint(got) {
|
|
t.Fatal(want, got)
|
|
}
|
|
}
|