34 lines
926 B
Go
34 lines
926 B
Go
|
package contract
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ValidateAPIResponseEnvelope checks that the response has success, data/error.
|
||
|
|
func ValidateAPIResponseEnvelope(t *testing.T, body []byte, expectSuccess bool) map[string]interface{} {
|
||
|
|
t.Helper()
|
||
|
|
var resp map[string]interface{}
|
||
|
|
require.NoError(t, json.Unmarshal(body, &resp))
|
||
|
|
success, ok := resp["success"].(bool)
|
||
|
|
require.True(t, ok, "response must have success field")
|
||
|
|
require.Equal(t, expectSuccess, success)
|
||
|
|
if expectSuccess {
|
||
|
|
require.Contains(t, resp, "data")
|
||
|
|
require.Nil(t, resp["error"])
|
||
|
|
} else {
|
||
|
|
require.Contains(t, resp, "error")
|
||
|
|
}
|
||
|
|
return resp
|
||
|
|
}
|
||
|
|
|
||
|
|
// RequireDataKeys checks that data has the required top-level keys.
|
||
|
|
func RequireDataKeys(t *testing.T, data map[string]interface{}, keys ...string) {
|
||
|
|
t.Helper()
|
||
|
|
for _, k := range keys {
|
||
|
|
require.Contains(t, data, k, "data must contain key %q", k)
|
||
|
|
}
|
||
|
|
}
|