tpb/endpoints_test.go at master · odwrtw/tpb · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package tpb

import (
	"reflect"
	"testing"
	"time"
)

func TestNewEndpoints(t *testing.T) {
	urls := []string{
		"http://test1.com/",
		"http://test2.com",
		"http://test3.com/",
		"http://test4.com",
	}

	got := newEndpoints(urls...)
	expected := endpoints{
		{baseURL: urls[0][:16]},
		{baseURL: urls[1]},
		{baseURL: urls[2][:16]},
		{baseURL: urls[3]},
	}

	if !reflect.DeepEqual(got, expected) {
		t.Fatalf("expected:\n%v\ngot:\n%v", expected, got)
	}
}

func TestBestEndpoint(t *testing.T) {
	e1 := &endpoint{lastFailure: time.Now().Add(-1 * time.Minute)}
	e2 := &endpoint{lastFailure: time.Now().Add(-2 * time.Minute)}
	e3 := &endpoint{}
	e4 := &endpoint{lastFailure: time.Now()}

	input := endpoints{e1, e2, e3, e4}
	best := input.best()

	if best != e3 {
		t.Fatalf("failed to find the best endpoint")
	}
}

func TestNoEndpoint(t *testing.T) {
	e := endpoints{}
	if e.best() != nil {
		t.Fatalf("no endpoints should mean no best endpoint")
	}
}