Increase test coverage.

Fix possible problem with verificationsWriter set as nil.
storozhukBM · Aug 3, 2018 · bab047b · bab047b
1 parent 52a76f6
commit bab047b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/verifier.go b/verifier.go
@@ -189,7 +189,7 @@ func printWarningOnUncheckedVerification(v *Verify) {
 		return
 	}
 	rawWriter := verificationsWriter.Load()
-	if rawWriter == nil {
+	if rawWriter == nil || rawWriter.(writerWrapper).value == nil {
 		rawWriter = writerWrapper{os.Stdout}
 	}
 	writer := rawWriter.(writerWrapper).value

diff --git a/verifier_test.go b/verifier_test.go
@@ -144,6 +144,12 @@ func TestVerifier_negative_unhandled_success(t *testing.T) {
 	if !strings.HasPrefix(resultBuffer, "[ERROR] found unhandled verification: verification success") {
 		t.Fatalf("unexpected verifier buffer: %s", resultBuffer)
 	}
+	verifier.SetUnhandledVerificationsWriter(nil)
+
+	verify = verifier.New()
+	verify.That(true, "empty string is not nil")
+	runtime.GC()
+	time.Sleep(10 * time.Millisecond)
 }
 
 func TestVerifier_negative_silent(t *testing.T) {