Update test flag parsing which was broken in go 1.13
@@ -454,7 +454,7 @@ func createApp(t *testing.T, testID string) *scheduler.Context {
return ctxs[0]
}
-func init() {
+func TestMain(m *testing.M) {
flag.IntVar(&snapshotScaleCount,
"snapshot-scale-count",
10,
@@ -464,4 +464,5 @@ func init() {
logrus.Errorf("Setup failed with error: %v", err)
os.Exit(1)
}
+ os.Exit(m.Run())
}
