chore(ztest/timeout): capture original scale after err handling (#1138)

sashamelentyev · web-flow · commit 4cfaabdda0c2 · 2022-07-25T08:13:47.000-07:00
diff --git a/internal/ztest/timeout.go b/internal/ztest/timeout.go
@@ -42,11 +42,11 @@ func Sleep(base time.Duration) {
 // Initialize checks the environment and alters the timeout scale accordingly.
 // It returns a function to undo the scaling.
 func Initialize(factor string) func() {
-	original := _timeoutScale
 	fv, err := strconv.ParseFloat(factor, 64)
 	if err != nil {
 		panic(err)
 	}
+	original := _timeoutScale
 	_timeoutScale = fv
 	return func() { _timeoutScale = original }
 }

Original file line number	Diff line number	Diff line change
`@@ -42,11 +42,11 @@ func Sleep(base time.Duration) {`
`42`	`42`	`// Initialize checks the environment and alters the timeout scale accordingly.`
`43`	`43`	`// It returns a function to undo the scaling.`
`44`	`44`	`func Initialize(factor string) func() {`
`45`		`- original := _timeoutScale`
`46`	`45`	`fv, err := strconv.ParseFloat(factor, 64)`
`47`	`46`	`if err != nil {`
`48`	`47`	`panic(err)`
`49`	`48`	`}`
	`49`	`+ original := _timeoutScale`
`50`	`50`	`_timeoutScale = fv`
`51`	`51`	`return func() { _timeoutScale = original }`
`52`	`52`	`}`