-
-
Notifications
You must be signed in to change notification settings - Fork 109
/
assertx.go
63 lines (50 loc) · 1.63 KB
/
assertx.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0
package assertx
import (
"bytes"
"encoding/json"
"math"
"strings"
"testing"
"time"
"github.com/tidwall/sjson"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func PrettifyJSONPayload(t *testing.T, payload interface{}) string {
o, err := json.MarshalIndent(payload, "", " ")
require.NoError(t, err)
return string(o)
}
func EqualAsJSON(t *testing.T, expected, actual interface{}, args ...interface{}) {
var eb, ab bytes.Buffer
if len(args) == 0 {
args = []interface{}{PrettifyJSONPayload(t, actual)}
}
require.NoError(t, json.NewEncoder(&eb).Encode(expected), args...)
require.NoError(t, json.NewEncoder(&ab).Encode(actual), args...)
assert.JSONEq(t, strings.TrimSpace(eb.String()), strings.TrimSpace(ab.String()), args...)
}
func EqualAsJSONExcept(t *testing.T, expected, actual interface{}, except []string, args ...interface{}) {
var eb, ab bytes.Buffer
if len(args) == 0 {
args = []interface{}{PrettifyJSONPayload(t, actual)}
}
require.NoError(t, json.NewEncoder(&eb).Encode(expected), args...)
require.NoError(t, json.NewEncoder(&ab).Encode(actual), args...)
var err error
ebs, abs := eb.String(), ab.String()
for _, k := range except {
ebs, err = sjson.Delete(ebs, k)
require.NoError(t, err)
abs, err = sjson.Delete(abs, k)
require.NoError(t, err)
}
assert.JSONEq(t, strings.TrimSpace(ebs), strings.TrimSpace(abs), args...)
}
func TimeDifferenceLess(t *testing.T, t1, t2 time.Time, seconds int) {
t.Helper()
delta := math.Abs(float64(t1.Unix()) - float64(t2.Unix()))
assert.Less(t, delta, float64(seconds))
}