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 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90
|
// Copyright (C) MongoDB, Inc. 2023-present.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License. You may obtain
// a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
package codecutil
import (
"io"
"testing"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/bsoncodec"
"go.mongodb.org/mongo-driver/bson/bsonrw"
"go.mongodb.org/mongo-driver/internal/assert"
"go.mongodb.org/mongo-driver/internal/require"
)
func testEncFn(t *testing.T) EncoderFn {
t.Helper()
return func(w io.Writer) (*bson.Encoder, error) {
rw, err := bsonrw.NewBSONValueWriter(w)
require.NoError(t, err, "failed to construct BSONValue writer")
enc, err := bson.NewEncoder(rw)
require.NoError(t, err, "failed to construct encoder")
return enc, nil
}
}
func TestMarshalValue(t *testing.T) {
t.Parallel()
tests := []struct {
name string
val interface{}
registry *bsoncodec.Registry
encFn EncoderFn
want string
wantErr error
}{
{
name: "empty",
val: nil,
want: "",
wantErr: ErrNilValue,
encFn: testEncFn(t),
},
{
name: "bson.D",
val: bson.D{{"foo", "bar"}},
want: `{"foo": "bar"}`,
encFn: testEncFn(t),
},
{
name: "map",
val: map[string]interface{}{"foo": "bar"},
want: `{"foo": "bar"}`,
encFn: testEncFn(t),
},
{
name: "struct",
val: struct{ Foo string }{Foo: "bar"},
want: `{"foo": "bar"}`,
encFn: testEncFn(t),
},
{
name: "non-document type",
val: "foo: bar",
want: `"foo: bar"`,
encFn: testEncFn(t),
},
}
for _, test := range tests {
test := test
t.Run(test.name, func(t *testing.T) {
t.Parallel()
value, err := MarshalValue(test.val, test.encFn)
assert.Equal(t, test.wantErr, err, "expected and actual error do not match")
assert.Equal(t, test.want, value.String(), "expected and actual comments are different")
})
}
}
|