Merge pull request #67 from ddymko/dns-priority

DNS Record Update Priority
vultr · Jun 2, 2020 · d79aa63 · d79aa63
2 parents e8d9613 + 2b3525b
commit d79aa63
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/dns_records.go b/dns_records.go
@@ -27,7 +27,7 @@ type DNSRecord struct {
 	Type     string `json:"type"`
 	Name     string `json:"name"`
 	Data     string `json:"data"`
-	Priority int    `json:"priority"`
+	Priority *int   `json:"priority"`
 	TTL      int    `json:"ttl"`
 }
 
@@ -130,8 +130,8 @@ func (d *DNSRecordsServiceHandler) Update(ctx context.Context, domain string, dn
 	if dnsRecord.TTL != 0 {
 		values.Add("ttl", strconv.Itoa(dnsRecord.TTL))
 	}
-	if dnsRecord.Priority != 0 {
-		values.Add("priority", strconv.Itoa(dnsRecord.Priority))
+	if dnsRecord.Priority != nil {
+		values.Add("priority", strconv.Itoa(*dnsRecord.Priority))
 	}
 
 	req, err := d.client.NewRequest(ctx, http.MethodPost, uri, values)

diff --git a/dns_records_test.go b/dns_records_test.go
@@ -51,10 +51,10 @@ func TestDNSRecordsServiceHandler_List(t *testing.T) {
 	if err != nil {
 		t.Errorf("DNSRecord.List returned %+v, expected %+v", err, nil)
 	}
-
+	p := 0
 	expected := []DNSRecord{
-		{Type: "A", Name: "", Data: "127.0.0.1", Priority: 0, RecordID: 1265276, TTL: 300},
-		{Type: "A", Name: "", Data: "127.0.0.1", Priority: 0, RecordID: 1265276, TTL: 300},
+		{Type: "A", Name: "", Data: "127.0.0.1", Priority: &p, RecordID: 1265276, TTL: 300},
+		{Type: "A", Name: "", Data: "127.0.0.1", Priority: &p, RecordID: 1265276, TTL: 300},
 	}
 
 	if !reflect.DeepEqual(records, expected) {
@@ -70,13 +70,13 @@ func TestDNSRecordsServiceHandler_Update(t *testing.T) {
 
 		fmt.Fprint(writer)
 	})
-
+	p := 10
 	params := &DNSRecord{
 		RecordID: 14283638,
 		Name:     "api",
 		Data:     "turnip.data",
 		TTL:      120,
-		Priority: 10,
+		Priority: &p,
 	}
 
 	err := client.DNSRecord.Update(ctx, "turnip.services", params)