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
|
package funcs
import (
"bytes"
"context"
"strconv"
"testing"
"github.com/stretchr/testify/assert"
)
func TestCreateBase64Funcs(t *testing.T) {
for i := 0; i < 10; i++ {
// Run this a bunch to catch race conditions
t.Run(strconv.Itoa(i), func(t *testing.T) {
t.Parallel()
ctx := context.Background()
fmap := CreateBase64Funcs(ctx)
actual := fmap["base64"].(func() interface{})
assert.Same(t, ctx, actual().(*Base64Funcs).ctx)
})
}
}
func TestBase64Encode(t *testing.T) {
bf := &Base64Funcs{}
assert.Equal(t, "Zm9vYmFy", must(bf.Encode("foobar")))
}
func TestBase64Decode(t *testing.T) {
bf := &Base64Funcs{}
assert.Equal(t, "foobar", must(bf.Decode("Zm9vYmFy")))
}
func TestBase64DecodeBytes(t *testing.T) {
bf := &Base64Funcs{}
out, err := bf.DecodeBytes("Zm9vYmFy")
assert.NoError(t, err)
assert.Equal(t, "foobar", string(out))
}
func TestToBytes(t *testing.T) {
assert.Equal(t, []byte{0, 1, 2, 3}, toBytes([]byte{0, 1, 2, 3}))
buf := &bytes.Buffer{}
buf.WriteString("hi")
assert.Equal(t, []byte("hi"), toBytes(buf))
assert.Equal(t, []byte{}, toBytes(nil))
assert.Equal(t, []byte("42"), toBytes(42))
}
|
