mirror of
https://github.com/hanzoai/docdb.git
synced 2026-10-30 18:39:39 +00:00
zeekayandHanzo Dev
06d5eca873
zip's registration is a verb method on the group, so an operation reads as the group answering it rather than a function taking the group, and a route that is not a typed operation says which method it answers with. The package-level form and the Router interface it existed to accept any target through are both gone. The toolchain and the pin move together, deliberately. Below zip v1.36.50 the JSON encoder was selected on a build tag, so a repo whose go directive rises first writes a wire that is neither the old release's nor the new one's until the pin catches up. Co-authored-by: Hanzo Dev <[email protected]>
780 lines
23 KiB
Go
780 lines
23 KiB
Go
// Copyright 2021 DocDB Inc.
|
|
//
|
|
// 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
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package integration
|
|
|
|
import (
|
|
"math"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
|
|
"github.com/hanzoai/docdb/integration/setup"
|
|
"github.com/hanzoai/docdb/integration/shareddata"
|
|
)
|
|
|
|
func TestQueryBitwiseAllClear(t *testing.T) {
|
|
t.Parallel()
|
|
ctx, collection := setup.Setup(t, shareddata.Scalars)
|
|
|
|
// Move to compat.
|
|
// TODO https://github.com/hanzoai/docdb/issues/508
|
|
_, err := collection.DeleteOne(ctx, bson.D{{"_id", "binary"}})
|
|
require.NoError(t, err)
|
|
_, err = collection.DeleteOne(ctx, bson.D{{"_id", "binary-empty"}})
|
|
require.NoError(t, err)
|
|
|
|
for name, tc := range map[string]struct {
|
|
value any // required, used for $bitsAllClear filter value
|
|
expectedIDs []any // optional
|
|
|
|
err *mongo.CommandError // optional, expected error from MongoDB
|
|
altMessage string // optional, alternative error message for DocDB, ignored if empty
|
|
failsForDocDB string
|
|
}{
|
|
"Array": {
|
|
value: primitive.A{1, 5},
|
|
expectedIDs: []any{
|
|
"decimal128-int-zero", "decimal128-zero",
|
|
"double-1", "double-3",
|
|
"double-big", "double-zero",
|
|
"int32-min", "int32-zero",
|
|
"int64-3", "int64-big", "int64-double-big",
|
|
"int64-min", "int64-zero",
|
|
},
|
|
},
|
|
"ArrayNegativeBitPositionValue": {
|
|
value: primitive.A{int32(-1)},
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: "Failed to parse bit position. Expected a non-negative number in: 0: -1",
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/241",
|
|
},
|
|
"ArrayFloatWholeBitPositionValue": {
|
|
value: primitive.A{1.0},
|
|
expectedIDs: []any{
|
|
"decimal128-int-zero", "decimal128-zero",
|
|
"double-1", "double-3", "double-big",
|
|
"double-zero",
|
|
"int32-1", "int32-2", "int32-3", "int32-min", "int32-zero",
|
|
"int64-1", "int64-2", "int64-3", "int64-big",
|
|
"int64-double-big", "int64-min", "int64-zero",
|
|
},
|
|
},
|
|
"ArrayFloatNotWholeBitPositionValue": {
|
|
value: primitive.A{1.2},
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: "Failed to parse bit position. Expected an integer: 0: 1.2",
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/278",
|
|
},
|
|
"ArrayStringBitPositionValue": {
|
|
value: primitive.A{"123"},
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: `Failed to parse bit position. Expected a number in: 0: "123"`,
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/241",
|
|
},
|
|
|
|
"Double": {
|
|
value: 1.2,
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected an integer: $bitsAllClear: 1.2",
|
|
},
|
|
},
|
|
"DoubleWhole": {
|
|
value: 2.0,
|
|
expectedIDs: []any{
|
|
"decimal128-int-zero", "decimal128-zero",
|
|
"double-1", "double-3",
|
|
"double-big", "double-zero",
|
|
"int32-1", "int32-2", "int32-3", "int32-min", "int32-zero",
|
|
"int64-1", "int64-2", "int64-3",
|
|
"int64-big", "int64-double-big", "int64-min", "int64-zero",
|
|
},
|
|
},
|
|
"DoubleNegativeValue": {
|
|
value: float64(-1),
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected a non-negative number in: $bitsAllClear: -1.0",
|
|
},
|
|
altMessage: "Expected a positive number in: $bitsAllClear: -1.0",
|
|
},
|
|
"DoubleZero": {
|
|
value: 0.0,
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-double", "decimal128-int", "decimal128-int-zero", "decimal128-whole", "decimal128-zero",
|
|
"double-1", "double-2", "double-3", "double-big", "double-whole",
|
|
"double-zero", "int32", "int32-1", "int32-2", "int32-3",
|
|
"int32-max", "int32-min", "int32-zero", "int64", "int64-1",
|
|
"int64-2", "int64-3", "int64-big", "int64-double-big", "int64-max",
|
|
"int64-min", "int64-zero",
|
|
},
|
|
},
|
|
|
|
"String": {
|
|
value: "123",
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: `v takes an Array, a number, or a BinData but received: $bitsAllClear: "123"`,
|
|
},
|
|
altMessage: `value takes an Array, a number, or a BinData but received: $bitsAllClear: "123"`,
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/241",
|
|
},
|
|
|
|
"Binary": {
|
|
value: primitive.Binary{Data: []byte{2}},
|
|
expectedIDs: []any{
|
|
"decimal128-int-zero", "decimal128-zero",
|
|
"double-1", "double-3",
|
|
"double-big", "double-zero",
|
|
"int32-1", "int32-2", "int32-3", "int32-min", "int32-zero",
|
|
"int64-1", "int64-2", "int64-3",
|
|
"int64-big", "int64-double-big", "int64-min", "int64-zero",
|
|
},
|
|
},
|
|
"BinaryWithZeroBytes": {
|
|
value: primitive.Binary{Data: []byte{0, 0, 2}},
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-int", "decimal128-int-zero", "decimal128-whole", "decimal128-zero",
|
|
"double-1", "double-2", "double-big",
|
|
"double-whole", "double-zero",
|
|
"int32", "int32-1", "int32-min", "int32-zero",
|
|
"int64", "int64-1", "int64-big", "int64-double-big",
|
|
"int64-min", "int64-zero",
|
|
},
|
|
},
|
|
"Binary9Bytes": {
|
|
value: primitive.Binary{Data: []byte{1, 2, 3, 4, 5, 6, 7, 8, 9}},
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-int", "decimal128-int-zero", "decimal128-whole", "decimal128-zero",
|
|
"double-big", "double-whole", "double-zero",
|
|
"int32", "int32-zero",
|
|
"int64", "int64-1", "int64-big", "int64-double-big", "int64-zero",
|
|
},
|
|
},
|
|
|
|
"Int32": {
|
|
value: int32(2),
|
|
expectedIDs: []any{
|
|
"decimal128-int-zero", "decimal128-zero",
|
|
"double-1", "double-3", "double-big",
|
|
"double-zero",
|
|
"int32-1", "int32-2", "int32-3",
|
|
"int32-min", "int32-zero",
|
|
"int64-1", "int64-2", "int64-3", "int64-big",
|
|
"int64-double-big", "int64-min", "int64-zero",
|
|
},
|
|
},
|
|
"Int32NegativeValue": {
|
|
value: int32(-1),
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected a non-negative number in: $bitsAllClear: -1",
|
|
},
|
|
altMessage: "Expected a positive number in: $bitsAllClear: -1.0",
|
|
},
|
|
|
|
"Int64Max": {
|
|
value: math.MaxInt64,
|
|
expectedIDs: []any{
|
|
"decimal128-int-zero", "decimal128-zero",
|
|
"double-1",
|
|
"double-zero",
|
|
"int32-zero",
|
|
"int64-min",
|
|
"int64-zero",
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/257",
|
|
},
|
|
"Int64NegativeValue": {
|
|
value: int64(-1),
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected a non-negative number in: $bitsAllClear: -1",
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/258",
|
|
},
|
|
} {
|
|
t.Run(name, func(tt *testing.T) {
|
|
tt.Parallel()
|
|
|
|
var t testing.TB = tt
|
|
|
|
if tc.failsForDocDB != "" {
|
|
t = setup.FailsForDocDB(tt, tc.failsForDocDB)
|
|
}
|
|
|
|
filter := bson.D{{"v", bson.D{{"$bitsAllClear", tc.value}}}}
|
|
cursor, err := collection.Find(ctx, filter, options.Find().SetSort(bson.D{{"_id", 1}}))
|
|
if tc.err != nil {
|
|
assert.Nil(t, cursor)
|
|
AssertEqualAltCommandError(t, *tc.err, tc.altMessage, err)
|
|
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
var actual []bson.D
|
|
err = cursor.All(ctx, &actual)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, tc.expectedIDs, CollectIDs(t, actual))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestQueryBitwiseAllSet(t *testing.T) {
|
|
t.Parallel()
|
|
ctx, collection := setup.Setup(t, shareddata.Scalars)
|
|
|
|
// Move to compat.
|
|
// TODO https://github.com/hanzoai/docdb/issues/508
|
|
_, err := collection.DeleteOne(ctx, bson.D{{"_id", "binary"}})
|
|
require.NoError(t, err)
|
|
_, err = collection.DeleteOne(ctx, bson.D{{"_id", "binary-empty"}})
|
|
require.NoError(t, err)
|
|
|
|
for name, tc := range map[string]struct {
|
|
value any // required, used for $bitsAllSet filter value
|
|
expectedIDs []any // optional
|
|
|
|
err *mongo.CommandError // optional, expected error from MongoDB
|
|
altMessage string // optional, alternative error message for DocDB, ignored if empty
|
|
failsForDocDB string
|
|
}{
|
|
"Array": {
|
|
value: primitive.A{1, 5},
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-double", "decimal128-int", "decimal128-whole", "double-2", "double-whole", "int32", "int32-max", "int64", "int64-max",
|
|
},
|
|
},
|
|
"ArrayNegativeBitPositionValue": {
|
|
value: primitive.A{-1},
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: "Failed to parse bit position. Expected a non-negative number in: 0: -1",
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/241",
|
|
},
|
|
"ArrayBadValue": {
|
|
value: primitive.A{"123"},
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: `Failed to parse bit position. Expected a number in: 0: "123"`,
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/241",
|
|
},
|
|
|
|
"Double": {
|
|
value: 1.2,
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected an integer: $bitsAllSet: 1.2",
|
|
},
|
|
},
|
|
"DoubleWhole": {
|
|
value: 2.0,
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-double", "decimal128-int", "decimal128-whole", "double-2", "double-whole", "int32", "int32-max", "int64", "int64-max",
|
|
},
|
|
},
|
|
"DoubleNegativeValue": {
|
|
value: -1.0,
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected a non-negative number in: $bitsAllSet: -1.0",
|
|
},
|
|
altMessage: "Expected a positive number in: $bitsAllSet: -1.0",
|
|
},
|
|
"LongZero": {
|
|
value: int64(0),
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-double", "decimal128-int", "decimal128-int-zero", "decimal128-whole", "decimal128-zero",
|
|
"double-1", "double-2", "double-3", "double-big", "double-whole",
|
|
"double-zero", "int32", "int32-1", "int32-2", "int32-3",
|
|
"int32-max", "int32-min", "int32-zero", "int64", "int64-1",
|
|
"int64-2", "int64-3", "int64-big", "int64-double-big", "int64-max",
|
|
"int64-min", "int64-zero",
|
|
},
|
|
},
|
|
|
|
"String": {
|
|
value: "123",
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: `v takes an Array, a number, or a BinData but received: $bitsAllSet: "123"`,
|
|
},
|
|
altMessage: `value takes an Array, a number, or a BinData but received: $bitsAllSet: "123"`,
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/241",
|
|
},
|
|
|
|
"Binary": {
|
|
value: primitive.Binary{Data: []byte{2}},
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-double", "decimal128-int", "decimal128-whole", "double-2", "double-whole", "int32", "int32-max", "int64", "int64-max",
|
|
},
|
|
},
|
|
"BinaryWithZeroBytes": {
|
|
value: primitive.Binary{Data: []byte{0, 0, 2}},
|
|
expectedIDs: []any{"decimal128-double", "double-3", "int32-2", "int32-3", "int32-max", "int64-2", "int64-3", "int64-max"},
|
|
},
|
|
|
|
"Int32": {
|
|
value: int32(2),
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-double", "decimal128-int", "decimal128-whole", "double-2", "double-whole", "int32", "int32-max", "int64", "int64-max",
|
|
},
|
|
},
|
|
"Int32NegativeValue": {
|
|
value: int32(-1),
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected a non-negative number in: $bitsAllSet: -1",
|
|
},
|
|
altMessage: "Expected a positive number in: $bitsAllSet: -1.0",
|
|
},
|
|
|
|
"Int64Max": {
|
|
value: math.MaxInt64,
|
|
expectedIDs: []any{"int64-max"},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/257",
|
|
},
|
|
"Int64NegativeValue": {
|
|
value: int64(-1),
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected a non-negative number in: $bitsAllSet: -1",
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/258",
|
|
},
|
|
} {
|
|
t.Run(name, func(tt *testing.T) {
|
|
tt.Parallel()
|
|
|
|
var t testing.TB = tt
|
|
|
|
if tc.failsForDocDB != "" {
|
|
t = setup.FailsForDocDB(tt, tc.failsForDocDB)
|
|
}
|
|
|
|
require.NotNil(t, tc.value, "value must not be nil")
|
|
|
|
filter := bson.D{{"v", bson.D{{"$bitsAllSet", tc.value}}}}
|
|
cursor, err := collection.Find(ctx, filter, options.Find().SetSort(bson.D{{"_id", 1}}))
|
|
if tc.err != nil {
|
|
assert.Nil(t, cursor)
|
|
AssertEqualAltCommandError(t, *tc.err, tc.altMessage, err)
|
|
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
var actual []bson.D
|
|
err = cursor.All(ctx, &actual)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, tc.expectedIDs, CollectIDs(t, actual))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestQueryBitwiseAnyClear(t *testing.T) {
|
|
t.Parallel()
|
|
ctx, collection := setup.Setup(t, shareddata.Scalars)
|
|
|
|
// Move to compat.
|
|
// TODO https://github.com/hanzoai/docdb/issues/508
|
|
_, err := collection.DeleteOne(ctx, bson.D{{"_id", "binary"}})
|
|
require.NoError(t, err)
|
|
_, err = collection.DeleteOne(ctx, bson.D{{"_id", "binary-empty"}})
|
|
require.NoError(t, err)
|
|
|
|
for name, tc := range map[string]struct {
|
|
value any // required, used for $bitsAnyClear filter value
|
|
expectedIDs []any // optional
|
|
|
|
err *mongo.CommandError // optional, expected error from MongoDB
|
|
altMessage string // optional, alternative error message for DocDB, ignored if empty
|
|
failsForDocDB string
|
|
}{
|
|
"Array": {
|
|
value: primitive.A{1, 5},
|
|
expectedIDs: []any{
|
|
"decimal128-int-zero", "decimal128-zero",
|
|
"double-1", "double-3",
|
|
"double-big", "double-zero",
|
|
"int32-1", "int32-2", "int32-3",
|
|
"int32-min", "int32-zero",
|
|
"int64-1", "int64-2", "int64-3",
|
|
"int64-big", "int64-double-big", "int64-min", "int64-zero",
|
|
},
|
|
},
|
|
"ArrayNegativeBitPositionValue": {
|
|
value: primitive.A{-1},
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: "Failed to parse bit position. Expected a non-negative number in: 0: -1",
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/241",
|
|
},
|
|
"ArrayBadValue": {
|
|
value: primitive.A{"123"},
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: `Failed to parse bit position. Expected a number in: 0: "123"`,
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/241",
|
|
},
|
|
|
|
"Double": {
|
|
value: 1.2,
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected an integer: $bitsAnyClear: 1.2",
|
|
},
|
|
},
|
|
"DoubleWhole": {
|
|
value: 2.0,
|
|
expectedIDs: []any{
|
|
"decimal128-int-zero", "decimal128-zero",
|
|
"double-1", "double-3",
|
|
"double-big", "double-zero",
|
|
"int32-1", "int32-2", "int32-3",
|
|
"int32-min", "int32-zero",
|
|
"int64-1", "int64-2", "int64-3",
|
|
"int64-big", "int64-double-big", "int64-min", "int64-zero",
|
|
},
|
|
},
|
|
"DoubleNegativeValue": {
|
|
value: -1.0,
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected a non-negative number in: $bitsAnyClear: -1.0",
|
|
},
|
|
altMessage: "Expected a positive number in: $bitsAnyClear: -1.0",
|
|
},
|
|
"DoubleZero": {
|
|
value: 0.0,
|
|
expectedIDs: []any{},
|
|
},
|
|
|
|
"String": {
|
|
value: "123",
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: `v takes an Array, a number, or a BinData but received: $bitsAnyClear: "123"`,
|
|
},
|
|
altMessage: `value takes an Array, a number, or a BinData but received: $bitsAnyClear: "123"`,
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/241",
|
|
},
|
|
|
|
"Binary": {
|
|
value: primitive.Binary{Data: []byte{2}},
|
|
expectedIDs: []any{
|
|
"decimal128-int-zero", "decimal128-zero",
|
|
"double-1", "double-3",
|
|
"double-big", "double-zero",
|
|
"int32-1", "int32-2", "int32-3",
|
|
"int32-min", "int32-zero",
|
|
"int64-1", "int64-2", "int64-3",
|
|
"int64-big", "int64-double-big", "int64-min", "int64-zero",
|
|
},
|
|
},
|
|
"BinaryWithZeroBytes": {
|
|
value: primitive.Binary{Data: []byte{0, 0, 2}},
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-int", "decimal128-int-zero", "decimal128-whole", "decimal128-zero",
|
|
"double-1", "double-2",
|
|
"double-big", "double-whole", "double-zero",
|
|
"int32", "int32-1", "int32-min", "int32-zero",
|
|
"int64", "int64-1", "int64-big", "int64-double-big",
|
|
"int64-min", "int64-zero",
|
|
},
|
|
},
|
|
|
|
"Int32": {
|
|
value: int32(2),
|
|
expectedIDs: []any{
|
|
"decimal128-int-zero", "decimal128-zero",
|
|
"double-1", "double-3",
|
|
"double-big", "double-zero",
|
|
"int32-1", "int32-2", "int32-3",
|
|
"int32-min", "int32-zero",
|
|
"int64-1", "int64-2", "int64-3",
|
|
"int64-big", "int64-double-big", "int64-min", "int64-zero",
|
|
},
|
|
},
|
|
"Int32NegativeValue": {
|
|
value: int32(-1),
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected a non-negative number in: $bitsAnyClear: -1",
|
|
},
|
|
altMessage: "Expected a positive number in: $bitsAnyClear: -1.0",
|
|
},
|
|
|
|
"Int64Max": {
|
|
value: math.MaxInt64,
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-double", "decimal128-int", "decimal128-int-zero", "decimal128-whole", "decimal128-zero",
|
|
"double-1", "double-2", "double-3",
|
|
"double-big", "double-whole", "double-zero",
|
|
"int32", "int32-1", "int32-2", "int32-3", "int32-max", "int32-min", "int32-zero",
|
|
"int64", "int64-1", "int64-2", "int64-3", "int64-big", "int64-double-big",
|
|
"int64-min", "int64-zero",
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/257",
|
|
},
|
|
"Int64NegativeValue": {
|
|
value: int64(-1),
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected a non-negative number in: $bitsAnyClear: -1",
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/258",
|
|
},
|
|
} {
|
|
t.Run(name, func(tt *testing.T) {
|
|
tt.Parallel()
|
|
|
|
var t testing.TB = tt
|
|
|
|
if tc.failsForDocDB != "" {
|
|
t = setup.FailsForDocDB(tt, tc.failsForDocDB)
|
|
}
|
|
|
|
require.NotNil(t, tc.value, "value must not be nil")
|
|
|
|
filter := bson.D{{"v", bson.D{{"$bitsAnyClear", tc.value}}}}
|
|
cursor, err := collection.Find(ctx, filter, options.Find().SetSort(bson.D{{"_id", 1}}))
|
|
if tc.err != nil {
|
|
assert.Nil(t, cursor)
|
|
AssertEqualAltCommandError(t, *tc.err, tc.altMessage, err)
|
|
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
var actual []bson.D
|
|
err = cursor.All(ctx, &actual)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, tc.expectedIDs, CollectIDs(t, actual))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestQueryBitwiseAnySet(t *testing.T) {
|
|
t.Parallel()
|
|
ctx, collection := setup.Setup(t, shareddata.Scalars)
|
|
|
|
// Move to compat.
|
|
// TODO https://github.com/hanzoai/docdb/issues/508
|
|
_, err := collection.DeleteOne(ctx, bson.D{{"_id", "binary"}})
|
|
require.NoError(t, err)
|
|
_, err = collection.DeleteOne(ctx, bson.D{{"_id", "binary-empty"}})
|
|
require.NoError(t, err)
|
|
|
|
for name, tc := range map[string]struct {
|
|
value any // required, used for $bitsAnySet filter value
|
|
expectedIDs []any // optional
|
|
|
|
err *mongo.CommandError // optional, expected error from MongoDB
|
|
altMessage string // optional, alternative error message for DocDB, ignored if empty
|
|
failsForDocDB string
|
|
}{
|
|
"Array": {
|
|
value: primitive.A{1, 5},
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-double", "decimal128-int", "decimal128-whole",
|
|
"double-2", "double-whole",
|
|
"int32", "int32-1", "int32-2", "int32-3", "int32-max",
|
|
"int64", "int64-1", "int64-2", "int64-max",
|
|
},
|
|
},
|
|
"ArrayNegativeBitPositionValue": {
|
|
value: primitive.A{-1},
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: "Failed to parse bit position. Expected a non-negative number in: 0: -1",
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/241",
|
|
},
|
|
"ArrayBadValue": {
|
|
value: primitive.A{"123"},
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: `Failed to parse bit position. Expected a number in: 0: "123"`,
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/241",
|
|
},
|
|
|
|
"Double": {
|
|
value: 1.2,
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected an integer: $bitsAnySet: 1.2",
|
|
},
|
|
},
|
|
"DoubleWhole": {
|
|
value: 2.0,
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-double", "decimal128-int", "decimal128-whole",
|
|
"double-2", "double-whole",
|
|
"int32", "int32-max",
|
|
"int64", "int64-max",
|
|
},
|
|
},
|
|
"DoubleNegativeValue": {
|
|
value: -1.0,
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected a non-negative number in: $bitsAnySet: -1.0",
|
|
},
|
|
altMessage: "Expected a positive number in: $bitsAnySet: -1.0",
|
|
},
|
|
"DoubleZero": {
|
|
value: 0.0,
|
|
expectedIDs: []any{},
|
|
},
|
|
|
|
"String": {
|
|
value: "123",
|
|
err: &mongo.CommandError{
|
|
Code: 2,
|
|
Name: "BadValue",
|
|
Message: `v takes an Array, a number, or a BinData but received: $bitsAnySet: "123"`,
|
|
},
|
|
altMessage: `value takes an Array, a number, or a BinData but received: $bitsAnySet: "123"`,
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/241",
|
|
},
|
|
|
|
"Binary": {
|
|
value: primitive.Binary{Data: []byte{2}},
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-double", "decimal128-int", "decimal128-whole", "double-2", "double-whole", "int32", "int32-max", "int64", "int64-max",
|
|
},
|
|
},
|
|
"BinaryWithZeroBytes": {
|
|
value: primitive.Binary{Data: []byte{0, 0, 2}},
|
|
expectedIDs: []any{"decimal128-double", "double-3", "int32-2", "int32-3", "int32-max", "int64-2", "int64-3", "int64-max"},
|
|
},
|
|
|
|
"Int32": {
|
|
value: int32(2),
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-double", "decimal128-int", "decimal128-whole",
|
|
"double-2", "double-whole",
|
|
"int32", "int32-max",
|
|
"int64", "int64-max",
|
|
},
|
|
},
|
|
"Int32NegativeValue": {
|
|
value: int32(-1),
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected a non-negative number in: $bitsAnySet: -1",
|
|
},
|
|
altMessage: "Expected a positive number in: $bitsAnySet: -1.0",
|
|
},
|
|
|
|
"Int64Max": {
|
|
value: math.MaxInt64,
|
|
expectedIDs: []any{
|
|
"decimal128", "decimal128-double", "decimal128-int", "decimal128-whole",
|
|
"double-2", "double-3",
|
|
"double-big", "double-whole",
|
|
"int32", "int32-1", "int32-2", "int32-3", "int32-max", "int32-min",
|
|
"int64", "int64-1", "int64-2", "int64-3", "int64-big",
|
|
"int64-double-big", "int64-max",
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/257",
|
|
},
|
|
"Int64NegativeValue": {
|
|
value: int64(-1),
|
|
err: &mongo.CommandError{
|
|
Code: 9,
|
|
Name: "FailedToParse",
|
|
Message: "Expected a non-negative number in: $bitsAnySet: -1",
|
|
},
|
|
failsForDocDB: "https://github.com/hanzoai/docdb-DocumentDB/issues/258",
|
|
},
|
|
} {
|
|
t.Run(name, func(tt *testing.T) {
|
|
tt.Parallel()
|
|
|
|
var t testing.TB = tt
|
|
if tc.failsForDocDB != "" {
|
|
t = setup.FailsForDocDB(tt, tc.failsForDocDB)
|
|
}
|
|
|
|
require.NotNil(t, tc.value, "value must not be nil")
|
|
|
|
filter := bson.D{{"v", bson.D{{"$bitsAnySet", tc.value}}}}
|
|
cursor, err := collection.Find(ctx, filter, options.Find().SetSort(bson.D{{"_id", 1}}))
|
|
if tc.err != nil {
|
|
assert.Nil(t, cursor)
|
|
AssertEqualAltCommandError(t, *tc.err, tc.altMessage, err)
|
|
|
|
return
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
|
|
var actual []bson.D
|
|
err = cursor.All(ctx, &actual)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, tc.expectedIDs, CollectIDs(t, actual))
|
|
})
|
|
}
|
|
}
|