Skip to content

Commit 6195819

Browse files
committed
Make NUMBER_OF_VMS optional
defaultNumberOfVms was not used when the environment variable is not set. Signed-off-by: Kazuyoshi Kato <katokazu@amazon.com>
1 parent 990b3f5 commit 6195819

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

runtime/service_integ_test.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -251,10 +251,10 @@ func TestMultipleVMs_Isolated(t *testing.T) {
251251
require.NoError(t, err, "failed to get a namespace")
252252

253253
// numberOfVmsEnvName = NUMBER_OF_VMS ENV and is configurable from buildkite
254-
numberOfVms, err := strconv.Atoi(os.Getenv(numberOfVmsEnvName))
255-
require.NoError(t, err, "failed to get NUMBER_OF_VMS env")
256-
if numberOfVms == 0 {
257-
numberOfVms = defaultNumberOfVms
254+
numberOfVms := defaultNumberOfVms
255+
if str := os.Getenv(numberOfVmsEnvName); str != "" {
256+
numberOfVms, err = strconv.Atoi(str)
257+
require.NoError(t, err, "failed to get NUMBER_OF_VMS env")
258258
}
259259
t.Logf("TestMultipleVMs_Isolated: will run %d vm's", numberOfVms)
260260

0 commit comments

Comments
 (0)