package commands

import (
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/require"
)

func absPaths(files []string) []string {
	var abs []string
	for _, f := range files {
		absPath, _ := filepath.Abs("../testdata/list/" + f)
		abs = append(abs, absPath)
	}
	return abs
}

func TestListTestFiles(t *testing.T) {
	require.Equal(t,
		absPaths([]string{"sum/private_test.go", "sum/public_test.go"}),
		listTestFiles("../testdata/list"))
}

func TestProtectedFiles(t *testing.T) {
	require.Equal(t,
		absPaths([]string{"sum/dontchange.go"}),
		listProtectedFiles("../testdata/list"))
}

func TestPrivateFiles(t *testing.T) {
	require.Equal(t,
		absPaths([]string{"sum/private_test.go", "sum/solution.go"}),
		listPrivateFiles("../testdata/list"))
}