github.com/jenkins-x/jx/v2@v2.1.155/pkg/tests/files.go

package tests

import (
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"testing"

	"github.com/pkg/errors"

	"github.com/stretchr/testify/require"

	"github.com/jenkins-x/jx/v2/pkg/util"
	"github.com/stretchr/testify/assert"
)

// AssertFileContains asserts that a given file exists and contains the given text
func AssertFileContains(t *testing.T, fileName string, containsText string) {
	if AssertFileExists(t, fileName) {
		data, err := ioutil.ReadFile(fileName)
		assert.NoError(t, err, "Failed to read file %s", fileName)
		if err == nil {
			text := string(data)
			assert.True(t, strings.Index(text, containsText) >= 0, "The file %s does not contain text: %s", fileName, containsText)
		}
	}
}

// AssertFileDoesNotContain asserts that a given file exists and does not contain the given text
func AssertFileDoesNotContain(t *testing.T, fileName string, containsText string) {
	if AssertFileExists(t, fileName) {
		data, err := ioutil.ReadFile(fileName)
		assert.NoError(t, err, "Failed to read file %s", fileName)
		if err == nil {
			text := string(data)
			idx := strings.Index(text, containsText)
			line := ""
			if idx > 0 {
				lines := strings.Split(text, "\n")
				for i, l := range lines {
					if strings.Index(l, containsText) >= 0 {
						line = "line " + strconv.Itoa(i+1) + " = " + l
						break
					}
				}
			}
			assert.True(t, idx < 0, "The file %s should not contain text: %s as %s", fileName, containsText, line)
		}
	}
}

// AssertFileExists asserts that the given file exists
func AssertFileExists(t *testing.T, fileName string) bool {
	exists, err := util.FileExists(fileName)
	assert.NoError(t, err, "Failed checking if file exists %s", fileName)
	assert.True(t, exists, "File %s should exist", fileName)
	if exists {
		Debugf("File %s exists\n", fileName)
	}
	return exists
}

// AssertFileDoesNotExist asserts that the given file does not exist
func AssertFileDoesNotExist(t *testing.T, fileName string) bool {
	exists, err := util.FileExists(fileName)
	assert.NoError(t, err, "Failed checking if file exists %s", fileName)
	assert.False(t, exists, "File %s should not exist", fileName)
	if exists {
		Debugf("File %s does not exist\n", fileName)
	}
	return exists
}

// AssertFilesExist asserts that the list of file paths either exist or don't exist
func AssertFilesExist(t *testing.T, expected bool, paths ...string) {
	for _, path := range paths {
		if expected {
			AssertFileExists(t, path)
		} else {
			AssertFileDoesNotExist(t, path)
		}
	}
}

// AssertDirsExist asserts that the list of directory paths either exist or don't exist
func AssertDirsExist(t *testing.T, expected bool, paths ...string) {
	for _, path := range paths {
		if expected {
			AssertDirExists(t, path)
		} else {
			AssertDirDoesNotExist(t, path)
		}
	}
}

// AssertDirExists asserts that the given directory exists
func AssertDirExists(t *testing.T, dir string) bool {
	exists, err := util.DirExists(dir)
	assert.NoError(t, err, "Failed checking if directory exists %s", dir)
	assert.True(t, exists, "directory %s should exist", dir)
	return exists
}

// AssertDirDoesNotExist asserts that the given directory does not exist
func AssertDirDoesNotExist(t *testing.T, dir string) bool {
	exists, err := util.DirExists(dir)
	assert.NoError(t, err, "failed checking if directory exists %s", dir)
	assert.False(t, exists, "directory %s should not exist", dir)
	return exists
}

func AssertEqualFileText(t *testing.T, expectedFile string, actualFile string) error {
	expectedText, err := AssertLoadFileText(t, expectedFile)
	if err != nil {
		return err
	}
	actualText, err := AssertLoadFileText(t, actualFile)
	if err != nil {
		return err
	}
	assert.Equal(t, expectedText, actualText, "comparing text content of files %s and %s", expectedFile, actualFile)
	return nil
}

// AssertLoadFileText asserts that the given file name can be loaded and returns the string contents
func AssertLoadFileText(t *testing.T, fileName string) (string, error) {
	if !AssertFileExists(t, fileName) {
		return "", fmt.Errorf("File %s does not exist", fileName)
	}
	data, err := ioutil.ReadFile(fileName)
	assert.NoError(t, err, "Failed loading data for file: %s", fileName)
	if err != nil {
		return "", err
	}
	return string(data), nil
}

// AssertTextFileContentsEqual asserts that both the expected and actual files can be loaded as text
// and that their contents are identical
func AssertTextFileContentsEqual(t *testing.T, expectedFile string, actualFile string) {
	assert.NotEqual(t, expectedFile, actualFile, "should be given different file names")

	expected, err := AssertLoadFileText(t, expectedFile)
	require.NoError(t, err)
	actual, err := AssertLoadFileText(t, actualFile)
	require.NoError(t, err)

	assert.Equal(t, expected, actual, "contents of expected file %s and actual file %s", expectedFile, actualFile)
}

// AssertDirContentsEqual walks two directory structures and validates that the same files exist (by name) and that they have the same content
func AssertDirContentsEqual(t *testing.T, expectedDir string, actualDir string) {
	actualFiles := make(map[string]string, 0)
	expectedFiles := make(map[string]string, 0)
	err := filepath.Walk(actualDir, func(path string, info os.FileInfo, err error) error {
		relativePath, err := filepath.Rel(actualDir, path)
		if err != nil {
			return errors.WithStack(err)
		}
		actualFiles[relativePath] = path
		return nil
	})
	assert.NoError(t, err)
	err = filepath.Walk(expectedDir, func(path string, info os.FileInfo, err error) error {
		relativePath, err := filepath.Rel(expectedDir, path)
		if err != nil {
			return errors.WithStack(err)
		}
		expectedFiles[relativePath] = path
		return nil
	})
	assert.Len(t, actualFiles, len(expectedFiles))
	for relativePath, path := range expectedFiles {
		actualFile, ok := actualFiles[relativePath]
		assert.True(t, ok, "%s not present", relativePath)
		info, err := os.Stat(actualFile)
		assert.NoError(t, err)
		if !info.IsDir() {
			AssertTextFileContentsEqual(t, path, actualFile)
		}
	}
}