72 lines
1.3 KiB
Go
72 lines
1.3 KiB
Go
package commands
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/csv"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func captureStdout(f func() (*cobra.Command, error)) (string, error) {
|
|
old := os.Stdout
|
|
r, w, _ := os.Pipe()
|
|
os.Stdout = w
|
|
|
|
_, err := f()
|
|
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
w.Close()
|
|
os.Stdout = old
|
|
|
|
var buf bytes.Buffer
|
|
io.Copy(&buf, r)
|
|
return buf.String(), nil
|
|
}
|
|
|
|
func TestListAll(t *testing.T) {
|
|
assert := require.New(t)
|
|
dir, err := createSimpleTestSite(t, testSiteConfig{})
|
|
|
|
assert.NoError(err)
|
|
|
|
hugoCmd := newCommandsBuilder().addAll().build()
|
|
cmd := hugoCmd.getCommand()
|
|
|
|
defer func() {
|
|
os.RemoveAll(dir)
|
|
}()
|
|
|
|
cmd.SetArgs([]string{"-s=" + dir, "list", "all"})
|
|
|
|
out, err := captureStdout(cmd.ExecuteC)
|
|
assert.NoError(err)
|
|
|
|
r := csv.NewReader(strings.NewReader(out))
|
|
|
|
header, err := r.Read()
|
|
assert.NoError(err)
|
|
|
|
assert.Equal([]string{
|
|
"path", "slug", "title",
|
|
"date", "expiryDate", "publishDate",
|
|
"draft", "permalink",
|
|
}, header)
|
|
|
|
record, err := r.Read()
|
|
assert.NoError(err)
|
|
assert.Equal([]string{
|
|
filepath.Join("content", "p1.md"), "", "P1",
|
|
"0001-01-01T00:00:00Z", "0001-01-01T00:00:00Z", "0001-01-01T00:00:00Z",
|
|
"false", "https://example.org/p1/",
|
|
}, record)
|
|
}
|