jpictl/pkg/rings/encode_test.go

package rings

import (
	"fmt"
	"net/netip"
	"testing"
)

func TestRingZeroAddress(t *testing.T) {
	RZNTest(t, "RingZeroAddress", RingZeroAddress, []RZNTestCase{
		{1, 1, 50, MustParseAddr("10.0.17.50")},
		{1, 2, 50, MustParseAddr("10.0.18.50")},
		{2, 3, 1, MustParseAddr("10.0.35.1")},
		{2, 3, 300, netip.Addr{}},
	})
}

func TestRingOneAddress(t *testing.T) {
	RZNTest(t, "RingOneAddress", RingOneAddress, []RZNTestCase{
		{1, 1, 50, MustParseAddr("10.1.16.50")},
		{1, 2, 50, MustParseAddr("10.1.32.50")},
		{2, 3, 300, MustParseAddr("10.2.49.44")},
		{1, 20, 50, netip.Addr{}},
	})
}

type RZNTestCase struct {
	region RegionID
	zone   ZoneID
	node   NodeID
	addr   netip.Addr
}

func RZNTest(t *testing.T,
	fnName string, fn func(RegionID, ZoneID, NodeID) (netip.Addr, error),
	cases []RZNTestCase) {
	//
	for i, tc := range cases {
		s := fmt.Sprintf("%s(%v, %v, %v)", fnName,
			tc.region,
			tc.zone,
			tc.node,
		)

		addr, err := fn(tc.region, tc.zone, tc.node)

		switch {
		case !tc.addr.IsValid():
			// expect error
			if err != nil {
				t.Logf("[%v/%v]: %s → %s", i, len(cases), s, err)
			} else {
				t.Errorf("ERROR: [%v/%v]: %s → %s (expected %s)", i, len(cases), s, addr, "error")
			}
		case err != nil:
			t.Errorf("ERROR: [%v/%v]: %s → %s (expected %s)", i, len(cases), s, err, tc.addr)
		case addr.Compare(tc.addr) != 0:
			t.Errorf("ERROR: [%v/%v]: %s → %s (expected %s)", i, len(cases), s, addr, tc.addr)
		default:
			t.Logf("[%v/%v]: %s → %s", i, len(cases), s, addr)
		}
	}
}