mirror of
https://github.com/syncthing/syncthing.git
synced 2024-11-17 19:08:46 -07:00
149 lines
3.2 KiB
Go
149 lines
3.2 KiB
Go
// Copyright (C) 2014 The Syncthing Authors.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify it
|
|
// under the terms of the GNU General Public License as published by the Free
|
|
// Software Foundation, either version 3 of the License, or (at your option)
|
|
// any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful, but WITHOUT
|
|
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
// more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License along
|
|
// with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package discover
|
|
|
|
import (
|
|
"net/url"
|
|
"time"
|
|
|
|
"testing"
|
|
|
|
"github.com/syncthing/syncthing/internal/protocol"
|
|
)
|
|
|
|
type DummyClient struct {
|
|
url *url.URL
|
|
lookups []protocol.DeviceID
|
|
lookupRet []string
|
|
stops int
|
|
statusRet bool
|
|
statusChecks int
|
|
}
|
|
|
|
func (c *DummyClient) Lookup(device protocol.DeviceID) []string {
|
|
c.lookups = append(c.lookups, device)
|
|
return c.lookupRet
|
|
}
|
|
|
|
func (c *DummyClient) StatusOK() bool {
|
|
c.statusChecks++
|
|
return c.statusRet
|
|
}
|
|
|
|
func (c *DummyClient) Stop() {
|
|
c.stops++
|
|
}
|
|
|
|
func (c *DummyClient) Address() string {
|
|
return c.url.String()
|
|
}
|
|
|
|
func TestGlobalDiscovery(t *testing.T) {
|
|
c1 := &DummyClient{
|
|
statusRet: false,
|
|
lookupRet: []string{"test.com:1234"},
|
|
}
|
|
|
|
c2 := &DummyClient{
|
|
statusRet: true,
|
|
lookupRet: []string{},
|
|
}
|
|
|
|
c3 := &DummyClient{
|
|
statusRet: true,
|
|
lookupRet: []string{"best.com:2345"},
|
|
}
|
|
|
|
clients := []*DummyClient{c1, c2}
|
|
|
|
Register("test1", func(uri *url.URL, pkt *Announce) (Client, error) {
|
|
c := clients[0]
|
|
clients = clients[1:]
|
|
c.url = uri
|
|
return c, nil
|
|
})
|
|
|
|
Register("test2", func(uri *url.URL, pkt *Announce) (Client, error) {
|
|
c3.url = uri
|
|
return c3, nil
|
|
})
|
|
|
|
d := NewDiscoverer(device, []string{})
|
|
d.localBcastStart = time.Time{}
|
|
servers := []string{
|
|
"test1://123.123.123.123:1234",
|
|
"test1://23.23.23.23:234",
|
|
"test2://234.234.234.234.2345",
|
|
}
|
|
d.StartGlobal(servers, 1234)
|
|
|
|
if len(d.clients) != 3 {
|
|
t.Fatal("Wrong number of clients")
|
|
}
|
|
|
|
status := d.ExtAnnounceOK()
|
|
|
|
for _, c := range []*DummyClient{c1, c2, c3} {
|
|
if status[c.url.String()] != c.statusRet || c.statusChecks != 1 {
|
|
t.Fatal("Wrong status")
|
|
}
|
|
}
|
|
|
|
addrs := d.Lookup(device)
|
|
if len(addrs) != 2 {
|
|
t.Fatal("Wrong numer of addresses", addrs)
|
|
}
|
|
|
|
for _, addr := range []string{"test.com:1234", "best.com:2345"} {
|
|
found := false
|
|
for _, laddr := range addrs {
|
|
if laddr == addr {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
t.Fatal("Couldn't find", addr)
|
|
}
|
|
}
|
|
|
|
for _, c := range []*DummyClient{c1, c2, c3} {
|
|
if len(c.lookups) != 1 || c.lookups[0] != device {
|
|
t.Fatal("Wrong lookups")
|
|
}
|
|
}
|
|
|
|
addrs = d.Lookup(device)
|
|
if len(addrs) != 2 {
|
|
t.Fatal("Wrong numer of addresses", addrs)
|
|
}
|
|
|
|
// Answer should be cached, so number of lookups should have not incresed
|
|
for _, c := range []*DummyClient{c1, c2, c3} {
|
|
if len(c.lookups) != 1 || c.lookups[0] != device {
|
|
t.Fatal("Wrong lookups")
|
|
}
|
|
}
|
|
|
|
d.StopGlobal()
|
|
|
|
for _, c := range []*DummyClient{c1, c2, c3} {
|
|
if c.stops != 1 {
|
|
t.Fatal("Wrong number of stops")
|
|
}
|
|
}
|
|
}
|