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
|
package registry // import "github.com/docker/docker/testutil/registry"
import (
"net/http"
"net/http/httptest"
"regexp"
"strings"
"sync"
"testing"
)
type handlerFunc func(w http.ResponseWriter, r *http.Request)
// Mock represent a registry mock
type Mock struct {
server *httptest.Server
hostport string
handlers map[string]handlerFunc
mu sync.Mutex
}
// RegisterHandler register the specified handler for the registry mock
func (tr *Mock) RegisterHandler(path string, h handlerFunc) {
tr.mu.Lock()
defer tr.mu.Unlock()
tr.handlers[path] = h
}
// NewMock creates a registry mock
func NewMock(t testing.TB) (*Mock, error) {
t.Helper()
testReg := &Mock{handlers: make(map[string]handlerFunc)}
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
url := r.URL.String()
var matched bool
var err error
for re, function := range testReg.handlers {
matched, err = regexp.MatchString(re, url)
if err != nil {
t.Fatal("Error with handler regexp")
}
if matched {
function(w, r)
break
}
}
if !matched {
t.Fatalf("Unable to match %s with regexp", url)
}
}))
testReg.server = ts
testReg.hostport = strings.Replace(ts.URL, "http://", "", 1)
return testReg, nil
}
// URL returns the url of the registry
func (tr *Mock) URL() string {
return tr.hostport
}
// Close closes mock and releases resources
func (tr *Mock) Close() {
tr.server.Close()
}
|
