migrate code for github

providers/namedotcom/nameservers_test.go

@@ -0,0 +1,150 @@
+package namedotcom
+
+import (
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/StackExchange/dnscontrol/models"
+)
+
+var (
+	mux    *http.ServeMux
+	client *nameDotCom
+	server *httptest.Server
+)
+
+func setup() {
+	mux = http.NewServeMux()
+	server = httptest.NewServer(mux)
+
+	client = &nameDotCom{
+		APIUser: "bob",
+		APIKey:  "123",
+	}
+	apiBase = server.URL
+}
+
+func teardown() {
+	server.Close()
+}
+
+func TestGetNameservers(t *testing.T) {
+	for i, test := range []struct {
+		givenNs, expected string
+	}{
+		{"", ""},
+		{`"foo.ns.tld","bar.ns.tld"`, "bar.ns.tld,foo.ns.tld"},
+		{"ERR", "ERR"},
+		{"MSGERR", "ERR"},
+	} {
+		setup()
+		defer teardown()
+
+		mux.HandleFunc("/domain/get/example.tld", func(w http.ResponseWriter, r *http.Request) {
+			if test.givenNs == "ERR" {
+				http.Error(w, "UH OH", 500)
+				return
+			}
+			if test.givenNs == "MSGERR" {
+				w.Write(nameComError)
+				return
+			}
+			w.Write(domainResponse(test.givenNs))
+		})
+
+		found, err := client.getNameservers("example.tld")
+		if err != nil {
+			if test.expected == "ERR" {
+				continue
+			}
+			t.Errorf("Error on test %d: %s", i, err)
+			continue
+		}
+		if test.expected == "ERR" {
+			t.Errorf("Expected error on test %d, but was none", i)
+			continue
+		}
+		if found != test.expected {
+			t.Errorf("Test %d: Expected '%s', but found '%s'", i, test.expected, found)
+		}
+	}
+}
+
+func TestGetCorrections(t *testing.T) {
+	for i, test := range []struct {
+		givenNs  string
+		expected int
+	}{
+		{"", 1},
+		{`"foo.ns.tld","bar.ns.tld"`, 0},
+		{`"bar.ns.tld","foo.ns.tld"`, 0},
+		{`"foo.ns.tld"`, 1},
+		{`"1.ns.aaa","2.ns.www"`, 1},
+		{"ERR", -1}, //-1 means we expect an error
+		{"MSGERR", -1},
+	} {
+		setup()
+		defer teardown()
+
+		mux.HandleFunc("/domain/get/example.tld", func(w http.ResponseWriter, r *http.Request) {
+			if test.givenNs == "ERR" {
+				http.Error(w, "UH OH", 500)
+				return
+			}
+			if test.givenNs == "MSGERR" {
+				w.Write(nameComError)
+				return
+			}
+			w.Write(domainResponse(test.givenNs))
+		})
+		dc := &models.DomainConfig{
+			Name: "example.tld",
+			Nameservers: []*models.Nameserver{
+				{Name: "foo.ns.tld"},
+				{Name: "bar.ns.tld"},
+			},
+		}
+		corrections, err := client.GetRegistrarCorrections(dc)
+		if err != nil {
+			if test.expected == -1 {
+				continue
+			}
+			t.Errorf("Error on test %d: %s", i, err)
+			continue
+		}
+		if test.expected == -1 {
+			t.Errorf("Expected error on test %d, but was none", i)
+			continue
+		}
+		if len(corrections) != test.expected {
+			t.Errorf("Test %d: Expected '%d', but found '%d'", i, test.expected, len(corrections))
+		}
+	}
+}
+
+func domainResponse(ns string) []byte {
+	return []byte(fmt.Sprintf(`{
+  "result": {
+    "code": 100,
+    "message": "Command Successful"
+  },
+  "domain_name": "example.tld",
+  "create_date": "2015-12-28 18:08:05",
+  "expire_date": "2016-12-28 23:59:59",
+  "locked": true,
+  "nameservers": [%s],
+  "contacts": [],
+  "addons": {
+    "whois_privacy": {
+      "price": "3.99"
+    },
+    "domain\/renew": {
+      "price": "10.99"
+    }
+  }
+}`, ns))
+}
+
+var nameComError = []byte(`{"result":{"code":251,"message":"Authentication Error - Invalid Username Or Api Token"}}`)