test and fix

Signed-off-by: Jess Frazelle <acidburn@microsoft.com>
This commit is contained in:
Jess Frazelle
2018-07-05 18:18:31 -04:00
parent f1360a2bf0
commit ae31e7f249
2 changed files with 22 additions and 2 deletions

20
k8scan/main_test.go Normal file
View File

@@ -0,0 +1,20 @@
package main
import "testing"
func TestARINResponse(t *testing.T) {
info, err := getIPInfo("104.40.92.107")
if err != nil {
t.Fatal(err)
}
if info.Net.Organization.Handle != "MSFT" {
t.Fatalf("expected handle to be MSFT, got %s", info.Net.Organization.Handle)
}
if info.Net.Organization.Name != "Microsoft Corporation" {
t.Fatalf("expected handle to be Microsoft Corporation, got %s", info.Net.Organization.Name)
}
if info.Net.Organization.Reference != "https://whois.arin.net/rest/org/MSFT" {
t.Fatalf("expected reference to be https://whois.arin.net/rest/org/MSFT, got %s", info.Net.Organization.Reference)
}
}