update_test.go

package main

import (
	"context"
	"reflect"
	"testing"
	"time"

	"github.com/zerotier/go-ztcentral"
	"github.com/zerotier/go-ztcentral/pkg/spec"
)

func boolPtr(b bool) *bool {
	return &b
}

func stringPtr(s string) *string {
	return &s
}

// this is slightly different than the converters.go version; the result coming
// back from json parsing will always be `float64`, instead of `int`, which we
// pass normally through the type system in converters. See
// pkg/zerotier/converters.go for more information.
func toUintList(i interface{}) []uint {
	ray := []uint{}
	for _, x := range i.([]interface{}) {
		ray = append(ray, uint(x.(float64))) // here
	}
	return ray
}

func toUintDoubleList(i interface{}) [][]uint {
	ray := [][]uint{}
	for _, x := range i.([]interface{}) {
		items := []uint{}
		for _, y := range x.([]interface{}) {
			items = append(items, uint(y.(float64)))
		}
		ray = append(ray, items)
	}
	return ray
}

func modifyMember(ctx context.Context, networkID string, memberID string, updateFunc func(*spec.Member)) error {
	c, err := ztcentral.NewClient(controllerToken)
	if err != nil {
		return err
	}

	member, err := c.GetMember(ctx, networkID, memberID)
	if err != nil {
		return err
	}

	updateFunc(member)

	if _, err := c.UpdateMember(ctx, networkID, memberID, member); err != nil {
		return err
	}

	return nil
}

func TestMemberUpdate(t *testing.T) {
	// see TestNetworkUpdate for the flow of this test.
	tf := getTFTest(t)
	tf.Apply("testdata/plans/basic-member.tf")

	ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
	t.Cleanup(cancel)

	// for each network, perform a set of transformations with the client.
	for _, resource := range a(tf.State()["resources"]) {
		m := h(resource)
		attrs := h(h(a(m["instances"])[0])["attributes"])

		switch m["type"] {
		case "zerotier_member":
			switch m["name"] {
			case "alice":
				err := modifyMember(ctx, attrs["network_id"].(string), attrs["member_id"].(string), func(member *spec.Member) {
					member.Description = stringPtr("This is a new description")
					member.Hidden = boolPtr(false)
					member.Config.ActiveBridge = boolPtr(false)
					member.Config.NoAutoAssignIps = boolPtr(false)
					member.Config.IpAssignments = &[]string{"10.0.0.2"}
					member.Config.Capabilities = &[]int{0, 1, 2}
					member.Config.Tags = &[][]interface{}{{0, 1}}
				})
				if err != nil {
					t.Fatal(err)
				}
			default:
				t.Fatalf("invalid member %q encountered", m["name"])
			}
		}
	}

	tf.Refresh()

	// for each network, validate the transformations were applied.
	for _, resource := range a(tf.State()["resources"]) {
		m := h(resource)
		attrs := h(h(a(m["instances"])[0])["attributes"])

		switch m["type"] {
		case "zerotier_member":
			switch m["name"] {
			case "alice":
				if attrs["description"].(string) != "This is a new description" {
					t.Fatal("description was not set")
				}

				isBool(t, attrs["hidden"], false, "hidden")
				isBool(t, attrs["allow_ethernet_bridging"], false, "allow_ethernet_bridging")
				isBool(t, attrs["no_auto_assign_ips"], false, "no_auto_assign_ips")

				if a(attrs["ip_assignments"])[0].(string) != "10.0.0.2" {
					t.Fatal("ip_assignments was improperly set")
				}

				if !reflect.DeepEqual(toUintList(a(attrs["capabilities"])), []uint{0, 1, 2}) {
					t.Fatalf("capabilities was improperly set: is %#v", a(attrs["capabilities"]))
				}

				if !reflect.DeepEqual(toUintDoubleList(a(attrs["tags"])), [][]uint{{0, 1}}) {
					t.Fatalf("tags was improperly set: is %#v", a(attrs["tags"]))
				}
			default:
				t.Fatalf("invalid member %q encountered", m["name"])
			}
		}
	}

	// reset the settings with terraform. Did it work?
	tf.Apply("testdata/plans/basic-member.tf")

	// for each network, validate the transformations were applied.
	for _, resource := range a(tf.State()["resources"]) {
		m := h(resource)
		attrs := h(h(a(m["instances"])[0])["attributes"])

		switch m["type"] {
		case "zerotier_member":
			switch m["name"] {
			case "alice":
				if attrs["description"].(string) != "Hello, world" {
					t.Fatalf("description was not set; was %q", attrs["description"])
				}

				isBool(t, attrs["hidden"], true, "hidden")
				isBool(t, attrs["allow_ethernet_bridging"], true, "allow_ethernet_bridging")
				isBool(t, attrs["no_auto_assign_ips"], true, "no_auto_assign_ips")

				if !reflect.DeepEqual(toUintList(a(attrs["capabilities"])), []uint{1, 2, 3}) {
					t.Fatalf("capabilities was improperly set: is %#v", a(attrs["capabilities"]))
				}

				if a(attrs["ip_assignments"])[0].(string) != "10.0.0.1" {
					t.Fatal("ip_assignments was improperly set")
				}

				if !reflect.DeepEqual(toUintDoubleList(a(attrs["tags"])), [][]uint{{1000, 100}}) {
					t.Fatalf("tags was improperly set: is %#v", a(attrs["tags"]))
				}
			default:
				t.Fatalf("invalid member %q encountered", m["name"])
			}
		}
	}
}

func modifyNetwork(ctx context.Context, id string, updateFunc func(*spec.Network)) error {
	c, err := ztcentral.NewClient(controllerToken)
	if err != nil {
		return err
	}
	net, err := c.GetNetwork(ctx, id)
	if err != nil {
		return err
	}

	updateFunc(net)

	if _, err := c.UpdateNetwork(ctx, *net.Id, net); err != nil {
		return err
	}

	return nil
}

func TestNetworkUpdate(t *testing.T) { // nolint:gocyclo
	// this test uses the same plan as BasicNetwork, but then inverts some values
	// back to defaults in each one to ensure that terraform picks them up on
	// refresh.
	tf := getTFTest(t)
	tf.Apply("testdata/plans/basic-network.tf")

	ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
	t.Cleanup(cancel)

	// for each network, perform a set of transformations with the client.
	for _, resource := range a(tf.State()["resources"]) {
		m := h(resource)
		attrs := h(h(a(m["instances"])[0])["attributes"])

		switch m["type"] {
		case "zerotier_network":
			switch m["name"] {
			// XXX please, before you modify any of this, read the comments in provision_test.go.
			case "multicast_limit":
				// not updateable
			case "description":
				err := modifyNetwork(ctx, attrs["id"].(string), func(net *spec.Network) {
					net.Description = nil
				})

				if err != nil {
					t.Fatal(err)
				}
			case "assign_off":
				err := modifyNetwork(ctx, attrs["id"].(string), func(net *spec.Network) {
					net.Config.V4AssignMode = &spec.IPV4AssignMode{Zt: boolPtr(false)}
					net.Config.V6AssignMode = &spec.IPV6AssignMode{Zt: boolPtr(false), N6plane: boolPtr(true), Rfc4193: boolPtr(true)}
				})

				if err != nil {
					t.Fatal(err)
				}
			case "private":
				err := modifyNetwork(ctx, attrs["id"].(string), func(net *spec.Network) {
					net.Config.Private = boolPtr(false)
				})

				if err != nil {
					t.Fatal(err)
				}
			case "no_broadcast":
				err := modifyNetwork(ctx, attrs["id"].(string), func(net *spec.Network) {
					net.Config.EnableBroadcast = boolPtr(true)
				})

				if err != nil {
					t.Fatal(err)
				}
			case "flow_rules":
				c, err := ztcentral.NewClient(controllerToken)
				if err != nil {
					t.Fatal(err)
				}
				rules, err := c.UpdateNetworkRules(ctx, attrs["id"].(string), "accept;")
				if err != nil {
					t.Fatal(err)
				}

				if rules != "accept;" {
					t.Fatal("rules were not alterered on set")
				}
			case "dns_settings":
				err := modifyNetwork(ctx, attrs["id"].(string), func(net *spec.Network) {
					domain := "testing.home"
					net.Config.Dns = &spec.DNS{
						Domain:  &domain,
						Servers: &[]string{"1.1.1.1", "8.8.8.8"},
					}
				})
				if err != nil {
					t.Fatal(err)
				}
			case "alice", "bobs_garage":
				// this is a collection of defaults; not sure testing this is really worth the effort.
			default:
				t.Fatalf("Unexpected network %q in plan", m["name"])
			}
		}
	}

	tf.Refresh()

	// for each network, validate the transformations were applied.
	for _, resource := range a(tf.State()["resources"]) {
		m := h(resource)
		attrs := h(h(a(m["instances"])[0])["attributes"])

		switch m["type"] {
		case "zerotier_network":
			switch m["name"] {
			case "multicast_limit":
				// not updateable
			case "description":
				if attrs["description"].(string) != "My description is changed!" {
					t.Fatal("description was not updated")
				}
			case "flow_rules":
				if attrs["flow_rules"].(string) != "accept;" {
					t.Fatal("flow_rules were not updated")
				}
			case "assign_off":
				isBool(t, h(a(attrs["assign_ipv4"])[0])["zerotier"], false, "assign_ipv4/zerotier")

				table := map[string]bool{
					"zerotier": false,
					"sixplane": true,
					"rfc4193":  true,
				}

				for name, val := range table {
					isBool(t, h(a(attrs["assign_ipv6"])[0])[name], val, "assign_ipv6/"+name)
				}
			case "private":
				isBool(t, attrs["private"], false, "private")
			case "no_broadcast":
				isBool(t, attrs["enable_broadcast"], true, "private")
			case "dns_settings":
				if attrs["dns"] == nil {
					t.Fatal("dns settings were missing")
				}

				domain := h(a(attrs["dns"])[0])["domain"].(string)
				if domain != "testing.home" {
					t.Fatalf("dns domain was not updated, was %q", domain)
				}

				if len(a(h(a(attrs["dns"])[0])["servers"])) != 2 {
					t.Fatal("dns servers were not updated")
				}

				if a(h(a(attrs["dns"])[0])["servers"])[0] != "1.1.1.1" {
					t.Fatal("dns servers were incorrect")
				}

				if a(h(a(attrs["dns"])[0])["servers"])[1] != "8.8.8.8" {
					t.Fatal("dns servers were incorrect")
				}
			case "alice", "bobs_garage":
				// this is a collection of defaults; not sure testing this is really worth the effort.
			default:
				t.Fatalf("Unexpected network %q in plan", m["name"])
			}
		}
	}
}