@@ -18,11 +18,9 @@ package upgrade
18
18
19
19
import (
20
20
"bytes"
21
- "fmt"
22
21
"io/ioutil"
23
22
"os"
24
23
"testing"
25
- "time"
26
24
27
25
kubeadmapi "k8s.io/kubernetes/cmd/kubeadm/app/apis/kubeadm"
28
26
)
@@ -77,17 +75,25 @@ func TestGetK8sVersionFromUserInput(t *testing.T) {
77
75
name : "Version is optional" ,
78
76
},
79
77
}
80
- for tnum , tt := range tcases {
78
+ for _ , tt := range tcases {
81
79
t .Run (tt .name , func (t * testing.T ) {
82
80
flags := & applyPlanFlags {}
83
81
if len (tt .clusterConfig ) > 0 {
84
- tmpfile := fmt . Sprintf ( "/tmp/ kubeadm-upgrade-common-test-%d-%d .yaml", tnum , time . Now (). Unix () )
85
- if err := ioutil . WriteFile ( tmpfile , [] byte ( tt . clusterConfig ), 0666 ); err != nil {
82
+ file , err := ioutil . TempFile ( "" , " kubeadm-upgrade-common-test-* .yaml" )
83
+ if err != nil {
86
84
t .Fatalf ("Failed to create test config file: %+v" , err )
87
85
}
88
- defer os .Remove (tmpfile )
89
86
90
- flags .cfgPath = tmpfile
87
+ tmpFileName := file .Name ()
88
+ defer os .Remove (tmpFileName )
89
+
90
+ _ , err = file .WriteString (tt .clusterConfig )
91
+ file .Close ()
92
+ if err != nil {
93
+ t .Fatalf ("Failed to write test config file contents: %+v" , err )
94
+ }
95
+
96
+ flags .cfgPath = tmpFileName
91
97
}
92
98
93
99
userVersion , err := getK8sVersionFromUserInput (flags , tt .args , tt .isVersionMandatory )
0 commit comments