64 lines
1.7 KiB
Go
64 lines
1.7 KiB
Go
|
package httpmock
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil" //nolint: staticcheck
|
||
|
|
)
|
||
|
|
|
||
|
|
// File is a file name. The contents of this file is loaded on demand
|
||
|
|
// by the following methods.
|
||
|
|
//
|
||
|
|
// Note that:
|
||
|
|
//
|
||
|
|
// file := httpmock.File("file.txt")
|
||
|
|
// fmt.Printf("file: %s\n", file)
|
||
|
|
//
|
||
|
|
// prints the content of file "file.txt" as [File.String] method is used.
|
||
|
|
//
|
||
|
|
// To print the file name, and not its content, simply do:
|
||
|
|
//
|
||
|
|
// file := httpmock.File("file.txt")
|
||
|
|
// fmt.Printf("file: %s\n", string(file))
|
||
|
|
type File string
|
||
|
|
|
||
|
|
// MarshalJSON implements [encoding/json.Marshaler].
|
||
|
|
//
|
||
|
|
// Useful to be used in conjunction with [NewJsonResponse] or
|
||
|
|
// [NewJsonResponder] as in:
|
||
|
|
//
|
||
|
|
// httpmock.NewJsonResponder(200, httpmock.File("body.json"))
|
||
|
|
func (f File) MarshalJSON() ([]byte, error) {
|
||
|
|
return f.bytes()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f File) bytes() ([]byte, error) {
|
||
|
|
return ioutil.ReadFile(string(f))
|
||
|
|
}
|
||
|
|
|
||
|
|
// Bytes returns the content of file as a []byte. If an error occurs
|
||
|
|
// during the opening or reading of the file, it panics.
|
||
|
|
//
|
||
|
|
// Useful to be used in conjunction with [NewBytesResponse] or
|
||
|
|
// [NewBytesResponder] as in:
|
||
|
|
//
|
||
|
|
// httpmock.NewBytesResponder(200, httpmock.File("body.raw").Bytes())
|
||
|
|
func (f File) Bytes() []byte {
|
||
|
|
b, err := f.bytes()
|
||
|
|
if err != nil {
|
||
|
|
panic(fmt.Sprintf("Cannot read %s: %s", string(f), err))
|
||
|
|
}
|
||
|
|
return b
|
||
|
|
}
|
||
|
|
|
||
|
|
// String implements [fmt.Stringer] and returns the content of file as
|
||
|
|
// a string. If an error occurs during the opening or reading of the
|
||
|
|
// file, it panics.
|
||
|
|
//
|
||
|
|
// Useful to be used in conjunction with [NewStringResponse] or
|
||
|
|
// [NewStringResponder] as in:
|
||
|
|
//
|
||
|
|
// httpmock.NewStringResponder(200, httpmock.File("body.txt").String())
|
||
|
|
func (f File) String() string {
|
||
|
|
return string(f.Bytes())
|
||
|
|
}
|