rrset_test.go

package udnssdk

import (
	"encoding/json"
	"os"
	"reflect"
	"strings"
	"testing"
)

func Test_RRSets_SelectPre(t *testing.T) {
	if !enableIntegrationTests {
		t.SkipNow()
	}
	if testClient == nil {
		t.Fatalf("TestClient Not Defined?\n")
	}
	if !enableRRSetTests {
		t.SkipNow()
	}

	testClient, err := NewClient(testUsername, testPassword, testBaseURL)
	if err != nil {
		t.Fatal(err)
	}

	r := RRSetKey{
		Zone: testDomain,
		Type: "ANY",
		Name: testHostname,
	}
	t.Logf("Select(%v)", r)
	rrsets, err := testClient.RRSets.Select(r)

	if err != nil {
		t.Fatal(err)
	}
	t.Logf("RRSets: %+v\n", rrsets)
}

func Test_RRSets_Select(t *testing.T) {
	if !enableIntegrationTests {
		t.SkipNow()
	}
	if !enableRRSetTests {
		t.SkipNow()
	}

	testClient, err := NewClient(testUsername, testPassword, testBaseURL)
	if err != nil {
		t.Fatal(err)
	}

	r := RRSetKey{
		Zone: testDomain,
		Type: "ANY",
		Name: "",
	}
	t.Logf("Select(%v)", r)
	rrsets, err := testClient.RRSets.Select(r)

	if err != nil {
		t.Fatal(err)
	}
	t.Logf("RRSets: %+v\n", rrsets)
	t.Logf("Checking for profiles...\n")
	for _, rr := range rrsets {
		if rr.Profile != nil {
			typ := rr.Profile.Context()
			if typ == "" {
				t.Fatalf("Could not get type for profile %+v\n", rr.Profile)
			}
			t.Logf("Found Profile %s for %s\n", rr.Profile.Context(), rr.OwnerName)
			st, er := json.Marshal(rr.Profile)
			t.Logf("Marshal the profile to JSON: %s / %+v", string(st), er)
			p, _ := rr.Profile.GetProfileObject()
			t.Logf("Check the Magic Profile: %+v\n", p)
		}
	}
}

func Test_RRSets_Create(t *testing.T) {
	if !enableIntegrationTests {
		t.SkipNow()
	}
	if !enableRRSetTests {
		t.SkipNow()
	}
	if !enableChanges {
		t.SkipNow()
	}

	testClient, err := NewClient(testUsername, testPassword, testBaseURL)
	if err != nil {
		t.Fatal(err)
	}

	r := RRSetKey{
		Zone: testDomain,
		Type: "A",
		Name: testHostname,
	}
	val := RRSet{
		OwnerName: r.Name,
		RRType:    r.Type,
		TTL:       300,
		RData:     []string{testIP1},
	}
	p := RDPoolProfile{
		Context:     RDPoolSchema,
		Order:       "ROUND_ROBIN",
		Description: "T. migratorius",
	}
	val.Profile = p.RawProfile()
	t.Logf("Create(%v, %v)", r, val)
	resp, err := testClient.RRSets.Create(r, val)

	if err != nil {
		t.Fatal(err)
	}
	t.Logf("Response: %+v\n", resp)
}

// Another Get Test if it matchs the Ip in IP1
func Test_RRSets_SelectMid1(t *testing.T) {
	if !enableIntegrationTests {
		t.SkipNow()
	}
	if !enableRRSetTests {
		t.SkipNow()
	}

	testClient, err := NewClient(testUsername, testPassword, testBaseURL)
	if err != nil {
		t.Fatal(err)
	}

	r := RRSetKey{
		Zone: testDomain,
		Type: "ANY",
		Name: testHostname,
	}
	t.Logf("Select(%v)", r)
	rrsets, err := testClient.RRSets.Select(r)

	if err != nil {
		t.Fatal(err)
	}
	t.Logf("RRSets: %+v\n", rrsets)
	// Do the test v IP1 here
	actual := rrsets[0].RData[0]
	expected := testIP1
	if actual != expected {
		t.Fatalf("actual RData[0]\"%s\" != expected \"%s\"", actual, expected)
	}
}

func Test_RRSets_Update(t *testing.T) {
	if !enableIntegrationTests {
		t.SkipNow()
	}
	if !enableRRSetTests {
		t.SkipNow()
	}
	if !enableChanges {
		t.SkipNow()
	}

	testClient, err := NewClient(testUsername, testPassword, testBaseURL)
	if err != nil {
		t.Fatal(err)
	}

	r := RRSetKey{
		Zone: testDomain,
		Type: "A",
		Name: testHostname,
	}
	val := RRSet{
		OwnerName: r.Name,
		RRType:    r.Type,
		TTL:       300,
		RData:     []string{testIP2},
	}
	p := RDPoolProfile{
		Context:     RDPoolSchema,
		Order:       "RANDOM",
		Description: "T. migratorius",
	}
	val.Profile = p.RawProfile()
	t.Logf("Update(%v, %v)", r, val)
	resp, err := testClient.RRSets.Update(r, val)

	if err != nil {
		t.Fatal(err)
	}
	t.Logf("Response: %+v\n", resp)
}

// Another Get Test if it matches the Ip in IP2
func Test_RRSets_SelectMid(t *testing.T) {
	if !enableIntegrationTests {
		t.SkipNow()
	}
	if !enableRRSetTests {
		t.SkipNow()
	}

	testClient, err := NewClient(testUsername, testPassword, testBaseURL)
	if err != nil {
		t.Fatal(err)
	}

	r := RRSetKey{
		Zone: testDomain,
		Type: "ANY",
		Name: testHostname,
	}
	t.Logf("Select(%v)", r)
	rrsets, err := testClient.RRSets.Select(r)

	if err != nil {
		t.Fatal(err)
	}
	t.Logf("RRSets: %+v\n", rrsets)
	// Do the test v IP2 here
	if rrsets[0].RData[0] != testIP2 {
		t.Fatalf("RData[0]\"%s\" != testIP2\"%s\"", rrsets[0].RData[0], testIP2)
	}
	p, _ := rrsets[0].Profile.GetProfileObject()
	t.Logf("Check the Magic Profile: %+v\n", p)
}

func Test_RRSet_Delete(t *testing.T) {
	if !enableIntegrationTests {
		t.SkipNow()
	}
	if !enableRRSetTests {
		t.SkipNow()
	}
	if !enableChanges {
		t.SkipNow()
	}
	if testHostname == "" ||
		testHostname[0] == '*' ||
		testHostname[0] == '@' ||
		testHostname == "www" ||
		testHostname[0] == '.' {
		t.Fatalf("Invalid testHostname defined: %v", testHostname)
		os.Exit(1)
	}

	testClient, err := NewClient(testUsername, testPassword, testBaseURL)
	if err != nil {
		t.Fatal(err)
	}

	r := RRSetKey{
		Zone: testDomain,
		Type: "ANY",
		Name: testHostname,
	}
	t.Logf("Select(%v)", r)
	rrsets, err := testClient.RRSets.Select(r)

	if err != nil {
		t.Fatal(err)
	}
	t.Logf("RRSets: %+v\n", rrsets)
	for _, e := range rrsets {
		r := RRSetKey{
			Zone: testDomain,
			Type: e.RRType,
			Name: e.OwnerName,
		}
		if strings.Index(r.Type, " ") != -1 {
			t.Logf("Stripping whitespace rom Type: %v\n", r.Type)
			r.Type = strings.Fields(r.Type)[0]
		}
		t.Logf("Delete(%v)", r)
		resp, err := testClient.RRSets.Delete(r)
		t.Logf("Response: %+v\n", resp)
		if err != nil {
			t.Fatal(err)
		}
	}
}

func Test_RRSet_SelectPost(t *testing.T) {
	if !enableIntegrationTests {
		t.SkipNow()
	}
	if !enableRRSetTests {
		t.SkipNow()
	}

	testClient, err := NewClient(testUsername, testPassword, testBaseURL)
	if err != nil {
		t.Fatal(err)
	}

	r := RRSetKey{
		Zone: testDomain,
		Type: "ANY",
		Name: testHostname,
	}
	t.Logf("Select(%v)", r)
	rrsets, err := testClient.RRSets.Select(r)

	if err != nil {
		t.Fatal(err)
	}
	t.Logf("RRSets: %+v\n", rrsets)
}

func Test_DirPoolProfile_RawProfile(t *testing.T) {
	p := DirPoolProfile{}
	expected := RawProfile{
		"@context":    ProfileSchema(""),
		"description": "",
		// "rdataInfo":   []DPRDataInfo(nil),
		"rdataInfo": []interface{}{},
	}

	actual := p.RawProfile()

	if !reflect.DeepEqual(actual, expected) {
		for k := range expected {
			if !reflect.DeepEqual(actual[k], expected[k]) {
				t.Fatalf("RawProfile[%v] want %#v got %#v", k, expected[k], actual[k])
			}
		}
		for k := range actual {
			if !reflect.DeepEqual(actual[k], expected[k]) {
				t.Fatalf("RawProfile[%v] want %#v got %#v", k, expected[k], actual[k])
			}
		}
		t.Fatalf("want %#v got %#v", expected, actual)
	}
}

func Test_RDPoolProfile_RawProfile(t *testing.T) {
	p := RDPoolProfile{}
	expected := RawProfile{
		"@context":    ProfileSchema(""),
		"description": "",
		"order":       "",
	}

	actual := p.RawProfile()

	if !reflect.DeepEqual(actual, expected) {
		for k := range expected {
			if !reflect.DeepEqual(actual[k], expected[k]) {
				t.Fatalf("RawProfile[%v] want %#v got %#v", k, expected[k], actual[k])
			}
		}
		for k := range actual {
			if !reflect.DeepEqual(actual[k], expected[k]) {
				t.Fatalf("RawProfile[%v] want %#v got %#v", k, expected[k], actual[k])
			}
		}
		t.Fatalf("want %#v got %#v", expected, actual)
	}
}

func Test_SBPoolProfile_RawProfile(t *testing.T) {
	p := SBPoolProfile{}
	expected := RawProfile{
		"@context":    ProfileSchema(""),
		"description": "",
		"actOnProbes": false,
		// "rdataInfo":   []SBRDataInfo(nil),
		"rdataInfo": []interface{}{},
		"runProbes": false,
		// "backupRecords": []BackupRecord(nil),
		"backupRecords": []interface{}{},
	}

	actual := p.RawProfile()

	if !reflect.DeepEqual(actual, expected) {
		for k := range expected {
			if !reflect.DeepEqual(actual[k], expected[k]) {
				t.Fatalf("RawProfile[%v] want %#v got %#v", k, expected[k], actual[k])
			}
		}
		for k := range actual {
			if !reflect.DeepEqual(actual[k], expected[k]) {
				t.Fatalf("RawProfile[%v] want %#v got %#v", k, expected[k], actual[k])
			}
		}
		t.Fatalf("want %#v got %#v", expected, actual)
	}
}

func Test_RawProfile_TCPoolProfile(t *testing.T) {
	input := RawProfile{
		"@context":    TCPoolSchema,
		"description": "",
		"actOnProbes": false,
		"rdataInfo":   []SBRDataInfo{},
		"runProbes":   false,
	}
	expected := TCPoolProfile{
		Context:     TCPoolSchema,
		Description: "",
		ActOnProbes: false,
		RunProbes:   false,
		RDataInfo:   []SBRDataInfo{},
	}

	actual, err := input.TCPoolProfile()

	if err != nil {
		t.Fatal(err)
	}
	if !reflect.DeepEqual(actual, expected) {
		t.Fatalf("want %#v got %#v", expected, actual)
	}
}

func Test_TCPoolProfile_RawProfile(t *testing.T) {
	input := TCPoolProfile{}
	expected := RawProfile{
		"@context":    ProfileSchema(""),
		"description": "",
		"actOnProbes": false,
		// "rdataInfo":   []SBRDataInfo(nil),
		"rdataInfo": []interface{}{},
		"runProbes": false,
	}

	actual := input.RawProfile()

	if !reflect.DeepEqual(actual, expected) {
		for k := range expected {
			if !reflect.DeepEqual(actual[k], expected[k]) {
				t.Fatalf("RawProfile[%v] want %#v got %#v", k, expected[k], actual[k])
			}
		}
		t.Fatalf("want %#v got %#v", expected, actual)
	}
}