/
luadns_test.go
54 lines (43 loc) · 1.17 KB
/
luadns_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
package luadns_test
import (
"bufio"
"bytes"
"io"
"net/http"
"os"
"path"
"testing"
"github.com/stretchr/testify/assert"
)
func endpointPath(filename string) string {
ext := path.Ext(filename)
name := filename[0 : len(filename)-len(ext)]
return name
}
func readHTTPFixture(name string) ([]byte, error) {
filename := "testdata/http" + name
return os.ReadFile(filename)
}
func sendHTTPFixture(t *testing.T, filename string, w http.ResponseWriter, r *http.Request) {
path := endpointPath(filename)
assert.Equal(t, r.URL.Path, path)
assert.Equal(t, r.Header.Get("Accept"), "application/json")
assert.Equal(t, r.Header.Get("Authorization"), "Basic am9lQGV4YW1wbGUuY29tOnBhc3N3b3Jk")
data, err := readHTTPFixture(filename)
assert.NoError(t, err)
resp, err := http.ReadResponse(bufio.NewReader(bytes.NewReader(data)), r)
assert.NoError(t, err)
flusher, ok := w.(http.Flusher)
if !ok {
panic("expected http.ResponseWriter to be an http.Flusher")
}
for name, values := range resp.Header {
for _, value := range values {
w.Header().Set(name, value)
}
}
w.WriteHeader(resp.StatusCode)
_, err = io.Copy(w, resp.Body)
assert.NoError(t, err)
flusher.Flush()
}