Ensure provided directory has at least one yaml manifest file

Signed-off-by: Vu Dinh <[email protected]>

Author: dinhxuanvu

pkg/lib/bundle/generate.go

@@ -104,13 +104,13 @@ func GetMediaType(directory string) (string, error) {
 	for _, item := range items {
 		if item.IsDir() {
 			continue
-		} else {
+		} else if filepath.Ext(item.Name()) == ".yaml" {
 			files = append(files, item.Name())
 		}
 	}
 
 	if len(files) == 0 {
-		return "", fmt.Errorf("The directory %s contains no files", directory)
+		return "", fmt.Errorf("The directory %s contains no yaml files", directory)
 	}
 
 	// Validate the file names to determine media type

pkg/lib/bundle/generate_test.go

@@ -37,7 +37,7 @@ func TestGetMediaType(t *testing.T) {
 		{
 			testDir,
 			"",
-			fmt.Sprintf("The directory %s contains no files", testDir),
+			fmt.Sprintf("The directory %s contains no yaml files", testDir),
 		},
 	}