You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
1.5 KiB
63 lines
1.5 KiB
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) |
|
} |
|
} |
|
}
|
|
|