syncthing/internal/discover/discover_test.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")
		}
	}
}