2017-07-24 19:39:39 +08:00
|
|
|
package tree
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
2020-11-05 23:18:51 +08:00
|
|
|
"context"
|
2017-07-24 19:39:39 +08:00
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/a8m/tree"
|
2019-07-29 01:47:38 +08:00
|
|
|
_ "github.com/rclone/rclone/backend/local"
|
|
|
|
"github.com/rclone/rclone/fs"
|
|
|
|
"github.com/rclone/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
|
|
|
|
2020-11-05 23:18:51 +08:00
|
|
|
f, err := fs.NewFs(context.Background(), "testfiles")
|
2017-07-24 19:39:39 +08:00
|
|
|
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())
|
|
|
|
}
|