netbird/management/internals/network_map_db/pgsql/dns_test.go

package networkmap_pgsql

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestRecordTypeAndRdata(t *testing.T) {
	var tests = []struct {
		recordType         string
		expectedRecordType int
		rdata              string
		expectedRdata      string
		expectedErr        error
	}{
		{recordType: "A", expectedRecordType: 1, rdata: "test.com", expectedRdata: "test.com", expectedErr: nil},
		{recordType: "AAAA", expectedRecordType: 28, rdata: "test.com", expectedRdata: "test.com", expectedErr: nil},
		{recordType: "CNAME", expectedRecordType: 5, rdata: "test.com", expectedRdata: "test.com.", expectedErr: nil},
		{recordType: "CNAME", expectedRecordType: 5, rdata: "test.com.", expectedRdata: "test.com.", expectedErr: nil},
		{recordType: "TypeMX", expectedErr: DnsUnsupportedRecordTypeError},
	}

	for _, tt := range tests {
		t.Run(tt.recordType, func(t *testing.T) {
			recordType, rdata, err := recordTypeAndRdata(tt.recordType, tt.rdata)

			if tt.expectedErr != nil {
				assert.ErrorIs(t, err, DnsUnsupportedRecordTypeError)
				return
			}

			assert.NoError(t, err)
			assert.Equal(t, recordType, tt.expectedRecordType)
			assert.Equal(t, rdata, tt.expectedRdata)
		})
	}
}