feat: publish standalone worker

Separate worker packaging and service lifecycle from the control plane.

internal/netaddr/cidr_test.go

@@ -0,0 +1,116 @@
+package netaddr
+
+import (
+	"bytes"
+	"net"
+	"testing"
+
+	_ "github.com/lib/pq"
+)
+
+func TestCidr(t *testing.T) {
+	db := openTestConn(t)
+	defer db.Close()
+
+	cidr := Cidr{}
+
+	// Test scanning NULL values
+	err := db.QueryRow("SELECT NULL::cidr").Scan(&cidr)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if cidr.Valid {
+		t.Fatalf("expected null result")
+	}
+
+	// Test setting NULL values
+	err = db.QueryRow("SELECT $1::cidr", cidr).Scan(&cidr)
+	if err != nil {
+		t.Fatalf("re-query null value failed: %s", err.Error())
+	}
+	if cidr.Valid {
+		t.Fatalf("expected null result")
+	}
+
+	// test encoding in query params, then decoding during Scan
+	testBidirectional := func(c Cidr, label string) {
+		err = db.QueryRow("SELECT $1::cidr", c).Scan(&cidr)
+		if err != nil {
+			t.Fatalf("re-query %s cidr failed: %s", label, err.Error())
+		}
+		if !cidr.Valid {
+			t.Fatalf("expected non-null value, got null for %s", label)
+		}
+		if !net.IP.Equal(c.Cidr.IP, cidr.Cidr.IP) {
+			t.Fatalf("expected IP addresses to match, but did not for %s - %s %s", label, c.Cidr.IP.String(), cidr.Cidr.IP.String())
+		}
+		if !bytes.Equal(c.Cidr.Mask, cidr.Cidr.Mask) {
+			t.Fatalf("expected net masks to match, but did not for %s", label)
+		}
+	}
+
+	// a few example CIDRs to test out
+	_, exampleCidr, err := net.ParseCIDR("135.104.0.0/32")
+	if err != nil {
+		t.Fatalf("Fatal error while building simple IP example - %s", err.Error())
+	}
+	simpleIP4 := Cidr{Cidr: *exampleCidr, Valid: true}
+	testBidirectional(simpleIP4, "Simple IPv4")
+
+	_, exampleCidr, err = net.ParseCIDR("0.0.0.0/24")
+	if err != nil {
+		t.Fatalf("Fatal error while building Zero IP example - %s", err.Error())
+	}
+	zeroIP4Subnet := Cidr{Cidr: *exampleCidr, Valid: true}
+	testBidirectional(zeroIP4Subnet, "Zero IPv4 Subnet")
+
+	_, exampleCidr, err = net.ParseCIDR("135.104.0.0/24")
+	if err != nil {
+		t.Fatalf("Fatal error while building simple IPv4 subnet example - %s", err.Error())
+	}
+	simpleIP4Subnet := Cidr{Cidr: *exampleCidr, Valid: true}
+	testBidirectional(simpleIP4Subnet, "Simple IPv4 Subnet")
+
+	_, exampleCidr, err = net.ParseCIDR("::1/128")
+	if err != nil {
+		t.Fatalf("Fatal error while building simple IPv6 loopback example - %s", err.Error())
+	}
+	ip6Loopback := Cidr{Cidr: *exampleCidr, Valid: true}
+	testBidirectional(ip6Loopback, "IPv6 Loopback")
+
+	_, exampleCidr, err = net.ParseCIDR("abcd:2345::/65")
+	if err != nil {
+		t.Fatalf("Fatal error while building simple IPv6 subnet example - %s", err.Error())
+	}
+	ip6Subnet := Cidr{Cidr: *exampleCidr, Valid: true}
+	testBidirectional(ip6Subnet, "IPv6 Subnet #1")
+
+	_, exampleCidr, err = net.ParseCIDR("abcd:2300::/24")
+	if err != nil {
+		t.Fatalf("Fatal error while building simple IPv6 subnet #2 example - %s", err.Error())
+	}
+	ip6Subnet2 := Cidr{Cidr: *exampleCidr, Valid: true}
+	testBidirectional(ip6Subnet2, "IPv6 Subnet #2")
+
+	_, exampleCidr, err = net.ParseCIDR("2001:DB8::1/48")
+	if err != nil {
+		t.Fatalf("Fatal error while building simple IPv6 subnet #3 example - %s", err.Error())
+	}
+	ip6Subnet3 := Cidr{Cidr: *exampleCidr, Valid: true}
+	testBidirectional(ip6Subnet3, "IPv6 Subnet #3")
+
+	// Error handling
+
+	// Bad argument
+	cidr = Cidr{}
+	err = cidr.Scan(456)
+	if err == nil {
+		t.Fatal("Expected error for non-byte[] argument to Scan")
+	}
+
+	cidr = Cidr{}
+	err = cidr.Scan([]byte(""))
+	if err == nil {
+		t.Fatalf("Expected error for invalid CIDR")
+	}
+}