forked from wgroeneveld/go-jamming
51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
package mocks
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
// neat trick! https://medium.com/@matryer/meet-moq-easily-mock-interfaces-in-go-476444187d10
|
|
type RestClientMock struct {
|
|
GetFunc func(string) (*http.Response, error)
|
|
GetBodyFunc func(string) (string, error)
|
|
PostFunc func(string, string, string) error
|
|
}
|
|
|
|
// although these are still requied to match the rest.Client interface.
|
|
func (m *RestClientMock) Get(url string) (*http.Response, error) {
|
|
return m.GetFunc(url)
|
|
}
|
|
func (m *RestClientMock) GetBody(url string) (string, error) {
|
|
return m.GetBodyFunc(url)
|
|
}
|
|
|
|
func (m *RestClientMock) Post(url string, contentType string, body string) error {
|
|
return m.PostFunc(url, contentType, body)
|
|
}
|
|
|
|
func RelPathGetBodyFunc(t *testing.T, relPath string) func(string) (string, error) {
|
|
return func(url string) (string, error) {
|
|
// url: https://brainbaking.com/something-something.html
|
|
// want: ../../mocks/something-something.html
|
|
mockfile := relPath + strings.ReplaceAll(url, "https://brainbaking.com/", "")
|
|
html, err := ioutil.ReadFile(mockfile)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
return string(html), nil
|
|
}
|
|
}
|
|
|
|
func BodyFunc(t *testing.T, mockfile string) func(string) (string, error) {
|
|
html, err := ioutil.ReadFile(mockfile)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
return func(url string) (string, error) {
|
|
return string(html), nil
|
|
}
|
|
}
|