Best solution that's type-safe but also extensible

Supply an `io.Writer` but check it against the `WriterResetter`
interface to see if we can call `Reset()`.  Which is a neat solution!

Add a new test to check that a supplied `bytes.Buffer` gets reset
between API calls.
This commit is contained in:
rjp 2023-05-25 11:09:41 +01:00
parent 08be497fae
commit 9921b587a7
2 changed files with 72 additions and 0 deletions

View file

@ -39,6 +39,70 @@ func TestGetFavourites(t *testing.T) {
}
}
func TestGetFavouritesSavedJSONTwice(t *testing.T) {
ourJSON := `[{"content": "foo"}, {"content": "bar"}]`
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Fprintln(w, ourJSON)
}))
defer ts.Close()
client := NewClient(&Config{
Server: ts.URL,
ClientID: "foo",
ClientSecret: "bar",
AccessToken: "zoo",
})
var buf bytes.Buffer
client.JSONWriter = &buf
favs, err := client.GetFavourites(context.Background(), nil)
if err != nil {
t.Fatalf("should not be fail: %v", err)
}
if len(favs) != 2 {
t.Fatalf("result should be two: %d", len(favs))
}
if favs[0].Content != "foo" {
t.Fatalf("want %q but %q", "foo", favs[0].Content)
}
if favs[1].Content != "bar" {
t.Fatalf("want %q but %q", "bar", favs[1].Content)
}
// We get a trailing `\n` from the API which we need to trim
// off before we compare it with our literal above.
theirJSON := strings.TrimSpace(string(buf.Bytes()))
if theirJSON != ourJSON {
t.Fatalf("want %q but %q", ourJSON, theirJSON)
}
// Now we call the API again to see if we get the same or doubled JSON.
favs, err = client.GetFavourites(context.Background(), nil)
if err != nil {
t.Fatalf("should not be fail: %v", err)
}
if len(favs) != 2 {
t.Fatalf("result should be two: %d", len(favs))
}
if favs[0].Content != "foo" {
t.Fatalf("want %q but %q", "foo", favs[0].Content)
}
if favs[1].Content != "bar" {
t.Fatalf("want %q but %q", "bar", favs[1].Content)
}
// We get a trailing `\n` from the API which we need to trim
// off before we compare it with our literal above.
theirJSON = strings.TrimSpace(string(buf.Bytes()))
if theirJSON != ourJSON {
t.Fatalf("want %q but %q", ourJSON, theirJSON)
}
}
func TestGetFavouritesSavedJSON(t *testing.T) {
ourJSON := `[{"content": "foo"}, {"content": "bar"}]`
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {