140 lines
2.7 KiB
Go
140 lines
2.7 KiB
Go
// Copyright (C) 2014 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 http://mozilla.org/MPL/2.0/.
|
|
|
|
package discover
|
|
|
|
import (
|
|
"net/url"
|
|
"time"
|
|
|
|
"testing"
|
|
|
|
"github.com/syncthing/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")
|
|
}
|
|
}
|
|
}
|