From 02885845ff4db30cfc3e1705d88eb94e720adb30 Mon Sep 17 00:00:00 2001
From: Slava Semushin <slava.semushin@gmail.com>
Date: Mon, 6 Jan 2020 15:21:54 +0000
Subject: [PATCH] kubetestgen: improve errors handling

- don't proceed when a file creation has been failed
- handle a possible error from WriteString()
---
 test/conformance/kubetestgen/kubetestgen.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/test/conformance/kubetestgen/kubetestgen.go b/test/conformance/kubetestgen/kubetestgen.go
index 87d9bbdb25f..d96a19305a0 100644
--- a/test/conformance/kubetestgen/kubetestgen.go
+++ b/test/conformance/kubetestgen/kubetestgen.go
@@ -131,6 +131,7 @@ func printYAML(fileName string, areaO Area) {
 	f, err := os.Create(fileName + ".yaml")
 	if err != nil {
 		fmt.Printf("ERROR: %s\n", err.Error())
+		os.Exit(1)
 	}
 	defer f.Close()
 	y, err := yaml.Marshal(areaO)
@@ -139,7 +140,11 @@ func printYAML(fileName string, areaO Area) {
 		os.Exit(1)
 	}
 
-	f.WriteString(string(y))
+	_, err = f.WriteString(string(y))
+	if err != nil {
+		fmt.Printf("ERROR: %s\n", err.Error())
+		os.Exit(1)
+	}
 }
 
 func countFields(suites []Suite) {