forked from pierrre/imageserver
/
server_test.go
111 lines (98 loc) · 2.12 KB
/
server_test.go
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package imageserver_test
import (
"testing"
. "github.com/pierrre/imageserver"
"github.com/pierrre/imageserver/testdata"
)
var _ Server = ServerFunc(nil)
func TestServerFunc(t *testing.T) {
called := false
server := ServerFunc(func(params Params) (*Image, error) {
called = true
return testdata.Medium, nil
})
server.Get(Params{})
if !called {
t.Fatal("not called")
}
}
var _ Server = &SourceServer{}
func TestSourceServer(t *testing.T) {
server := &SourceServer{
Server: testdata.Server,
}
im, err := server.Get(Params{SourceParam: testdata.MediumFileName})
if err != nil {
t.Fatal(err)
}
if im != testdata.Medium {
t.Fatal("not equal")
}
}
func TestSourceServerParam(t *testing.T) {
server := &SourceServer{
Server: ServerFunc(func(params Params) (*Image, error) {
if !params.Has(SourceParam) {
t.Fatal("no source param")
}
if params.Has("foo") {
t.Fatal("unexpected param")
}
return testdata.Medium, nil
}),
}
server.Get(Params{
SourceParam: testdata.MediumFileName,
"foo": "bar",
})
}
func TestSourceServerError(t *testing.T) {
server := &SourceServer{
Server: testdata.Server,
}
_, err := server.Get(Params{SourceParam: "foobar"})
if err == nil {
t.Fatal("no error")
}
}
func TestSourceServerErrorNoSource(t *testing.T) {
server := &SourceServer{
Server: testdata.Server,
}
_, err := server.Get(Params{})
if err == nil {
t.Fatal("no error")
}
if _, ok := err.(*ParamError); !ok {
t.Fatalf("unexpected error type: %T", err)
}
}
func TestNewLimitServer(t *testing.T) {
// TODO test limit
server := NewLimitServer(&StaticServer{Image: testdata.Medium}, 1)
im, err := server.Get(Params{})
if err != nil {
t.Fatal(err)
}
if im != testdata.Medium {
t.Fatal("not equal")
}
}
func TestNewLimitServerZero(t *testing.T) {
// TODO ?
NewLimitServer(&StaticServer{Image: testdata.Medium}, 0)
}
var _ Server = &StaticServer{}
func TestStaticServer(t *testing.T) {
s := &StaticServer{
Image: testdata.Medium,
Error: nil,
}
im, err := s.Get(Params{})
if err != nil {
t.Fatal(err)
}
if !ImageEqual(im, testdata.Medium) {
t.Fatal("not equal")
}
}