Added integration test for SetDisableReferral

likexian · May 2, 2023 · 8082fc7 · 8082fc7
1 parent 761ee92
commit 8082fc7
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/whois_test.go b/whois_test.go
@@ -21,6 +21,7 @@ package whois
 
 import (
 	"errors"
+	"strings"
 	"testing"
 	"time"
 
@@ -34,6 +35,20 @@ func TestVersion(t *testing.T) {
 	assert.Contains(t, License(), "Apache License")
 }
 
+func TestClient_SetDisableReferral(t *testing.T) {
+	client := NewClient()
+
+	resp, err := client.Whois("likexian.com")
+	assert.Nil(t, err)
+	assert.Equal(t, strings.Count(resp, "Domain Name: LIKEXIAN.COM"), 2)
+
+	client.SetDisableReferral(true)
+
+	resp, err = client.Whois("likexian.com")
+	assert.Nil(t, err)
+	assert.Equal(t, strings.Count(resp, "Domain Name: LIKEXIAN.COM"), 1)
+}
+
 func TestWhoisFail(t *testing.T) {
 	tests := []struct {
 		domain string