54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package rings
|
|
|
|
import (
|
|
"fmt"
|
|
"net/netip"
|
|
"testing"
|
|
)
|
|
|
|
func TestDecodeRingZeroAddress(t *testing.T) {
|
|
RZNDecodeTest(t, "DecodeRingZeroAddress", DecodeRingZeroAddress, []RZNDecodeTestCase{
|
|
{1, 1, 50, MustParseAddr("10.0.17.50"), true},
|
|
{1, 2, 50, MustParseAddr("10.0.18.50"), true},
|
|
{2, 3, 1, MustParseAddr("10.0.35.1"), true},
|
|
})
|
|
}
|
|
|
|
func TesDecodetRingOneAddress(t *testing.T) {
|
|
RZNDecodeTest(t, "DecodeRingOneAddress", DecodeRingOneAddress, []RZNDecodeTestCase{
|
|
{1, 1, 50, MustParseAddr("10.1.16.50"), true},
|
|
{1, 2, 50, MustParseAddr("10.1.32.50"), true},
|
|
{2, 3, 300, MustParseAddr("10.2.49.44"), true},
|
|
})
|
|
}
|
|
|
|
type RZNDecodeTestCase struct {
|
|
region RegionID
|
|
zone ZoneID
|
|
node NodeID
|
|
addr netip.Addr
|
|
ok bool
|
|
}
|
|
|
|
func RZNDecodeTest(t *testing.T,
|
|
fnName string, fn func(netip.Addr) (RegionID, ZoneID, NodeID, bool),
|
|
cases []RZNDecodeTestCase) {
|
|
//
|
|
for i, tc := range cases {
|
|
s := fmt.Sprintf("%s(%q)", fnName, tc.addr)
|
|
|
|
r, z, n, ok := fn(tc.addr)
|
|
|
|
switch {
|
|
case ok != tc.ok, r != tc.region, z != tc.zone, n != tc.node:
|
|
t.Errorf("ERROR: [%v/%v]: %s → %v %v %v %v (expected %v %v %v %v)",
|
|
i, len(cases), s,
|
|
r, z, n, ok,
|
|
tc.region, tc.zone, tc.node, tc.ok)
|
|
default:
|
|
t.Logf("[%v/%v]: %s → %v %v %v %v", i, len(cases), s,
|
|
r, z, n, ok)
|
|
}
|
|
}
|
|
}
|