// Get a list of all supported image files in a directory
func getImages(imagePath string) []string {
// Currently only supports .jpg's
imageTypes := [...]string{".jpg"}
dir, err := filetree.GetDir(imagePath)
if err != nil {
panic(err)
}
files, err := dir.GetFilePaths()
if err != nil {
panic(err)
}
return filterutils.HasSuffix(files, imageTypes[0])
// XXX Commented code is not a good practice, but this was good learning material
//allFiles := make([]string,0)
//for _, imageType := range imageTypes {
// filtered := filterutils.HasSuffix(files, imageType)
// allFiles = append(
// allFiles[:len(allFiles)],
// append(
// make([]string,0),
// filtered[:len(filtered)]...
// )...
// )
//}
}
func TestGetDirFailure(t *testing.T) {
invalidDir := "FHALKFOOLAJFLKAHJ"
_, err := filetree.GetDir(invalidDir)
if err == nil {
t.Errorf("No error was raised using an invalid directory name: " + invalidDir)
}
}
func TestGetFilePaths(t *testing.T) {
dirname, _ := os.Getwd()
d, err := filetree.GetDir(dirname)
CheckDirError(t, dirname, err)
files, err := d.GetFilePaths()
if err != nil {
t.Errorf("You were unable to get files from " + dirname)
}
if len(files) == 0 {
t.Errorf("No files were found in your current working directory")
}
}
func findFiles(imagePath, fileType string) []string {
dir, err := filetree.GetDir(imagePath)
if err != nil {
panic(err)
}
files, err := dir.GetFilePaths()
if err != nil {
panic(err)
}
return filterutils.Filter(
files,
func(i int) bool { return strings.HasSuffix(files[i], fileType) },
)
}
func TestGetDirSuccess(t *testing.T) {
dirname, _ := os.Getwd()
d, err := filetree.GetDir(dirname)
CheckDirError(t, dirname, err)
DirSuccessAsserts(dirname, d, t)
}