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
|
package integration
import (
"context"
"os"
"testing"
"github.com/stretchr/testify/require"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/internal/testutil"
"go.mongodb.org/mongo-driver/mongo/writeconcern"
"go.mongodb.org/mongo-driver/x/bsonx/bsoncore"
"go.mongodb.org/mongo-driver/x/mongo/driver/operation"
)
func TestCompression(t *testing.T) {
comp := os.Getenv("MONGO_GO_DRIVER_COMPRESSOR")
if len(comp) == 0 {
t.Skip("Skipping because no compressor specified")
}
wc := writeconcern.New(writeconcern.WMajority())
collOne := testutil.ColName(t)
testutil.DropCollection(t, testutil.DBName(t), collOne)
testutil.InsertDocs(t, testutil.DBName(t), collOne, wc,
bsoncore.BuildDocument(nil, bsoncore.AppendStringElement(nil, "name", "compression_test")),
)
cmd := operation.NewCommand(bsoncore.BuildDocument(nil, bsoncore.AppendInt32Element(nil, "serverStatus", 1))).
Deployment(testutil.Topology(t)).
Database(testutil.DBName(t))
ctx := context.Background()
err := cmd.Execute(ctx)
noerr(t, err)
result := cmd.Result()
serverVersion, err := result.LookupErr("version")
noerr(t, err)
if testutil.CompareVersions(t, serverVersion.StringValue(), "3.4") < 0 {
t.Skip("skipping compression test for version < 3.4")
}
networkVal, err := result.LookupErr("network")
noerr(t, err)
require.Equal(t, networkVal.Type, bson.TypeEmbeddedDocument)
compressionVal, err := networkVal.Document().LookupErr("compression")
noerr(t, err)
compressorDoc, err := compressionVal.Document().LookupErr(comp)
noerr(t, err)
compressorKey := "compressor"
compareTo36 := testutil.CompareVersions(t, serverVersion.StringValue(), "3.6")
if compareTo36 < 0 {
compressorKey = "compressed"
}
compressor, err := compressorDoc.Document().LookupErr(compressorKey)
noerr(t, err)
bytesIn, err := compressor.Document().LookupErr("bytesIn")
noerr(t, err)
require.True(t, bytesIn.IsNumber())
require.True(t, bytesIn.Int64() > 0)
}
|
