b37c214e3c
This was done with something along the lines of:
```
mv internal/test testutil
pushd testutil/; grep -IRl "package test" | xargs -I '{}' sed -i -e 's|package test|package testutil|g' {}; popd
mv internal/testutil/*.go testutil/ && rm -rf internal/
grep -IRl "github.com\/docker\/docker\/internal\/test" | xargs -I '{}' sed -i -e 's|github.com/docker/docker/internal/test|github.com/docker/docker/test|g' {}
goimports .
```
I also modified the basic plugin path in testutil/fixtures/plugin.
Signed-off-by: Sam Whited <sam@samwhited.com>
78 lines
2 KiB
Go
78 lines
2 KiB
Go
package request
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
// Options defines request options, like request modifiers and which host to target
|
|
type Options struct {
|
|
host string
|
|
requestModifiers []func(*http.Request) error
|
|
}
|
|
|
|
// Host creates a modifier that sets the specified host as the request URL host
|
|
func Host(host string) func(*Options) {
|
|
return func(o *Options) {
|
|
o.host = host
|
|
}
|
|
}
|
|
|
|
// With adds a request modifier to the options
|
|
func With(f func(*http.Request) error) func(*Options) {
|
|
return func(o *Options) {
|
|
o.requestModifiers = append(o.requestModifiers, f)
|
|
}
|
|
}
|
|
|
|
// Method creates a modifier that sets the specified string as the request method
|
|
func Method(method string) func(*Options) {
|
|
return With(func(req *http.Request) error {
|
|
req.Method = method
|
|
return nil
|
|
})
|
|
}
|
|
|
|
// RawString sets the specified string as body for the request
|
|
func RawString(content string) func(*Options) {
|
|
return RawContent(ioutil.NopCloser(strings.NewReader(content)))
|
|
}
|
|
|
|
// RawContent sets the specified reader as body for the request
|
|
func RawContent(reader io.ReadCloser) func(*Options) {
|
|
return With(func(req *http.Request) error {
|
|
req.Body = reader
|
|
return nil
|
|
})
|
|
}
|
|
|
|
// ContentType sets the specified Content-Type request header
|
|
func ContentType(contentType string) func(*Options) {
|
|
return With(func(req *http.Request) error {
|
|
req.Header.Set("Content-Type", contentType)
|
|
return nil
|
|
})
|
|
}
|
|
|
|
// JSON sets the Content-Type request header to json
|
|
func JSON(o *Options) {
|
|
ContentType("application/json")(o)
|
|
}
|
|
|
|
// JSONBody creates a modifier that encodes the specified data to a JSON string and set it as request body. It also sets
|
|
// the Content-Type header of the request.
|
|
func JSONBody(data interface{}) func(*Options) {
|
|
return With(func(req *http.Request) error {
|
|
jsonData := bytes.NewBuffer(nil)
|
|
if err := json.NewEncoder(jsonData).Encode(data); err != nil {
|
|
return err
|
|
}
|
|
req.Body = ioutil.NopCloser(jsonData)
|
|
req.Header.Set("Content-Type", "application/json")
|
|
return nil
|
|
})
|
|
}
|