97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
// Copyright (C) 2019 The Syncthing Authors.
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
// You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
|
|
package registry
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
func TestRegistry(t *testing.T) {
|
|
r := New()
|
|
|
|
want := func(i int) func(interface{}) bool {
|
|
return func(x interface{}) bool { return x.(int) == i }
|
|
}
|
|
|
|
if res := r.Get("int", want(1)); res != nil {
|
|
t.Error("unexpected")
|
|
}
|
|
|
|
r.Register("int", 1)
|
|
r.Register("int", 11)
|
|
r.Register("int4", 4)
|
|
r.Register("int4", 44)
|
|
r.Register("int6", 6)
|
|
r.Register("int6", 66)
|
|
|
|
if res := r.Get("int", want(1)).(int); res != 1 {
|
|
t.Error("unexpected", res)
|
|
}
|
|
|
|
// int is prefix of int4, so returns 1
|
|
if res := r.Get("int4", want(1)).(int); res != 1 {
|
|
t.Error("unexpected", res)
|
|
}
|
|
|
|
r.Unregister("int", 1)
|
|
|
|
if res := r.Get("int", want(1)).(int); res == 1 {
|
|
t.Error("unexpected", res)
|
|
}
|
|
|
|
if res := r.Get("int6", want(6)).(int); res != 6 {
|
|
t.Error("unexpected", res)
|
|
}
|
|
|
|
// Unregister 11, int should be impossible to find
|
|
r.Unregister("int", 11)
|
|
if res := r.Get("int", want(11)); res != nil {
|
|
t.Error("unexpected")
|
|
}
|
|
|
|
// Unregister a second time does nothing.
|
|
r.Unregister("int", 1)
|
|
|
|
// Can have multiple of the same
|
|
r.Register("int", 1)
|
|
r.Register("int", 1)
|
|
r.Unregister("int", 1)
|
|
|
|
if res := r.Get("int4", want(1)).(int); res != 1 {
|
|
t.Error("unexpected", res)
|
|
}
|
|
}
|
|
|
|
func TestShortSchemeFirst(t *testing.T) {
|
|
r := New()
|
|
r.Register("foo", 0)
|
|
r.Register("foobar", 1)
|
|
|
|
// If we don't care about the value, we should get the one with "foo".
|
|
res := r.Get("foo", func(interface{}) bool { return false })
|
|
if res != 0 {
|
|
t.Error("unexpected", res)
|
|
}
|
|
}
|
|
|
|
func BenchmarkGet(b *testing.B) {
|
|
r := New()
|
|
for _, addr := range []string{"192.168.1.1", "172.1.1.1", "10.1.1.1"} {
|
|
r.Register("tcp", &net.TCPAddr{IP: net.ParseIP(addr)})
|
|
}
|
|
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
r.Get("tcp", func(x interface{}) bool {
|
|
return x.(*net.TCPAddr).IP.IsUnspecified()
|
|
})
|
|
}
|
|
}
|