diff --git a/azure_test.go b/azure_test.go new file mode 100644 index 00000000..49028a69 --- /dev/null +++ b/azure_test.go @@ -0,0 +1,47 @@ +package ipcat + +import ( + "fmt" + "net/http" + "net/http/httptest" + "testing" +) + +func TestAzure(t *testing.T) { + ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + fmt.Fprintln(w, ` + + + + + + + + + + + +`) + + })) + + defer ts.Close() + msazure = ts.URL + b, err := DownloadAzure() + if err != nil { + t.Fatalf("DownloadAzure() error: %v", err) + } + + ipset := NewIntervalSet(100) + err = UpdateAzure(ipset, b) + if err != nil { + t.Fatalf("UpdateAzure error: %v", err) + } + rec, err := ipset.Contains("23.96.0.0") + if err != nil { + t.Fatalf("ipset.Contains(%q) error: %v", "23.96.0.0", err) + } + if rec == nil { + t.Errorf("ipset.Contains(%q) rec = nil, want exists", "23.96.0.0") + } +}