-
Notifications
You must be signed in to change notification settings - Fork 14
/
mock.go
67 lines (56 loc) · 1.53 KB
/
mock.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
// Package mock contains testing utilities and mocks shared between
// packages.
package restlib
import (
"net/http"
)
// ResponseWriter returns a mock ResponseWriter implementing http.ResponseWriter.
func ResponseWriter() *responseWriter { //nolint:golint
return &responseWriter{}
}
// responseWriter implements http.ResponseWriter.
type responseWriter struct {
LastBody string
LastStatus int
header http.Header
err error
}
// Err sets the error value returned by the Write method.
// Chain it with construction: mock.ResponseWriter().Err(someErr)
func (m *responseWriter) Err(err error) *responseWriter {
m.err = err
return m
}
// Header returns the responseWriters HTTP header.
func (m *responseWriter) Header() http.Header {
if m.header == nil {
m.header = http.Header{}
}
return m.header
}
// WriteHeader writes HTTP Status code to struct field.
func (m *responseWriter) WriteHeader(status int) {
m.LastStatus = status
}
// Write writes response body to struct field.
func (m *responseWriter) Write(b []byte) (int, error) {
m.LastBody = string(b)
return len(b), m.err
}
type readCloser struct{ err error }
func (r *readCloser) Read(_ []byte) (int, error) {
return 0, r.err
}
func (r *readCloser) Close() error {
return r.err
}
func (r *readCloser) Err(err error) *readCloser {
r.err = err
return r
}
// ReadCloser returns a mock io.ReadCloser that can be set to return
// an err when Read is called on it and do nothing when Close is called
// on it.
func ReadCloser() *readCloser { //nolint:golint
return &readCloser{}
}