2017-07-24 19:39:39 +08:00
|
|
|
package tree
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/a8m/tree"
|
2018-10-13 21:41:15 +08:00
|
|
|
_ "github.com/ncw/rclone/backend/local"
|
2017-07-24 19:39:39 +08:00
|
|
|
"github.com/ncw/rclone/fs"
|
2017-08-01 21:46:21 +08:00
|
|
|
"github.com/ncw/rclone/fstest"
|
2017-07-24 19:39:39 +08:00
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestTree(t *testing.T) {
|
2017-08-01 21:46:21 +08:00
|
|
|
fstest.Initialise()
|
|
|
|
|
2017-07-24 19:39:39 +08:00
|
|
|
buf := new(bytes.Buffer)
|
2017-08-01 21:46:21 +08:00
|
|
|
|
2017-07-24 19:39:39 +08:00
|
|
|
f, err := fs.NewFs("testfiles")
|
|
|
|
require.NoError(t, err)
|
|
|
|
err = Tree(f, buf, new(tree.Options))
|
|
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, `/
|
|
|
|
├── file1
|
|
|
|
├── file2
|
|
|
|
├── file3
|
|
|
|
└── subdir
|
|
|
|
├── file4
|
|
|
|
└── file5
|
|
|
|
|
|
|
|
1 directories, 5 files
|
|
|
|
`, buf.String())
|
|
|
|
}
|