updated per code review

Signed-off-by: Patrick Zheng <[email protected]>
notaryproject · Oct 29, 2024 · 640b430 · 640b430
1 parent 3ebd9c3
commit 640b430
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/request_test.go b/request_test.go
@@ -75,6 +75,9 @@ func TestNewRequest(t *testing.T) {
 
 	defaultRandReader := rand.Reader
 	rand.Reader = &dummyRandReader{}
+	defer func() {
+		rand.Reader = defaultRandReader
+	}()
 	opts = RequestOptions{
 		Content:       message,
 		HashAlgorithm: crypto.SHA256,
@@ -84,7 +87,6 @@ func TestNewRequest(t *testing.T) {
 	if err == nil || !errors.As(err, &malformedRequest) || err.Error() != expectedErrMsg {
 		t.Fatalf("expected error %s, but got %v", expectedErrMsg, err)
 	}
-	rand.Reader = defaultRandReader
 }
 
 func TestRequestMarshalBinary(t *testing.T) {