Finishing off whois tests

brimstone · Jun 12, 2016 · 8dc5b62 · 8dc5b62
1 parent b917ca3
commit 8dc5b62
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/whois_test.go b/whois_test.go
@@ -1,7 +1,6 @@
 package main_test
 
 import (
-	"fmt"
 	"log"
 	"testing"
 
@@ -83,5 +82,10 @@ func TestGetWhoisInfo(*testing.T) {
 	if err != nil {
 		log.Fatal(err)
 	}
-	fmt.Println(result)
+	if len(result.Nameservers) == 0 {
+		log.Fatal("Expected Nameservers")
+	}
+	if len(result.Emails) == 0 {
+		log.Fatal("Expected Email addresses")
+	}
 }