add assert error message

likexian · Apr 29, 2024 · 9dd719a · 9dd719a
1 parent d233eed
commit 9dd719a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/whois_test.go b/whois_test.go
@@ -114,9 +114,10 @@ func TestNewClient(t *testing.T) {
 	c := NewClient()
 	var err error
 
-	c.SetTimeout(10 * time.Microsecond)
+	c.SetTimeout(10 * time.Millisecond)
 	_, err = c.Whois("likexian.com")
 	assert.NotNil(t, err)
+	assert.Contains(t, err.Error(), "timeout")
 
 	c.SetTimeout(10 * time.Second)
 	_, err = c.Whois("likexian.com")