refactor

dns/service/vultr/service_test.go

@@ -0,0 +1,96 @@
+package vultr
+
+import (
+	"errors"
+	"net"
+	"testing"
+
+	"dnsupdater/dns"
+
+	"github.com/vultr/govultr/v3"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestVultrService_List(t *testing.T) {
+	expected := dns.RecordList{
+		{
+			Id:   "656939ee-f942-4ce2-af1d-3bd68c764e96",
+			Name: "sub1",
+			Ip:   net.IPv4(201, 110, 66, 72),
+		},
+		{
+			Id:   "c80118f4-f04c-4ad2-8ec2-16eb15cc8aca",
+			Name: "sub2",
+			Ip:   net.IPv4(242, 124, 218, 187),
+		},
+	}
+
+	service := Service{
+		api: mock{
+			t: t,
+			ListReturn: map[string][]govultr.DomainRecord{
+				"example.com": {
+					{
+						ID:       "656939ee-f942-4ce2-af1d-3bd68c764e96",
+						Type:     "A",
+						Name:     "sub1",
+						Data:     "201.110.66.72",
+						Priority: 2,
+						TTL:      1800,
+					},
+					{
+						ID:       "c80118f4-f04c-4ad2-8ec2-16eb15cc8aca",
+						Type:     "A",
+						Name:     "sub2",
+						Data:     "242.124.218.187",
+						Priority: 1,
+						TTL:      1800,
+					},
+				},
+			},
+		},
+	}
+
+	records, err := service.List("example.com")
+	assert.NoError(t, err)
+
+	assert.Equal(t, expected, records)
+
+	// Fetch invalid
+	_, err = service.List("noexists.com")
+	assert.Error(t, err)
+}
+
+func TestVultrService_Update(t *testing.T) {
+	mockApi := mock{
+		t: t,
+		ListReturn: map[string][]govultr.DomainRecord{
+			"example.com": {
+				{
+					ID:       "6cabe6ba-1ea1-405d-b66d-cd56ecac45ce",
+					Type:     "A",
+					Name:     "www",
+					Data:     "80.17.42.157",
+					Priority: 10,
+					TTL:      360,
+				},
+			},
+		},
+	}
+
+	service := Service{
+		api: &mockApi,
+	}
+
+	err := service.Update("example.com", "6cabe6ba-1ea1-405d-b66d-cd56ecac45ce", net.IPv4(221, 135, 170, 186).String())
+	assert.NoError(t, err)
+
+	mockApi.updateError = errors.New("Error")
+
+	err = service.Update("invalid.com", "332b40fc-0ddf-436c-a0c7-46586b928ac2", net.IPv4(72, 82, 118, 186).String())
+	assert.Error(t, err)
+
+	err = service.Update("example.com", "6cabe6ba-1ea1-405d-b66d-cd56ecac45ce", net.IPv4(221, 135, 170, 186).String())
+	assert.Error(t, err)
+}