Add search stage tests

documentdb_tests/compatibility/tests/core/collections/commands/cloneCollectionAsCapped/test_cloneCollectionAsCapped_wc.py

@@ -26,6 +26,7 @@
 )
 from documentdb_tests.framework.executor import execute_command
 from documentdb_tests.framework.parametrize import pytest_params
+from documentdb_tests.framework.property_checks import Eq
 from documentdb_tests.framework.test_constants import (
     DECIMAL128_ZERO,
     FLOAT_INFINITY,
@@ -101,7 +102,7 @@
             "size": 100_000,
             "writeConcern": {"wtimeout": 0},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=0 should succeed",
     ),
     CommandTestCase(
@@ -113,7 +114,7 @@
             "size": 100_000,
             "writeConcern": {"wtimeout": 1000},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=1000 should succeed",
     ),
 ]

documentdb_tests/compatibility/tests/core/collections/commands/convertToCapped/test_convertToCapped_wc.py

@@ -27,6 +27,7 @@
 )
 from documentdb_tests.framework.executor import execute_command
 from documentdb_tests.framework.parametrize import pytest_params
+from documentdb_tests.framework.property_checks import Eq
 from documentdb_tests.framework.test_constants import (
     DECIMAL128_ZERO,
     FLOAT_INFINITY,
@@ -99,11 +100,8 @@
             "size": 100_000,
             "writeConcern": {"wtimeout": v},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg=f"wtimeout={id} should succeed",
-        # A negative wtimeout is accepted cleanly on a standalone server but a
-        # replica set reports a writeConcernError, so gate that case.
-        marks=((pytest.mark.requires(quorum_write_concern=False),) if id == "negative" else ()),
     )
     for id, val in [
         ("zero", 0),

documentdb_tests/compatibility/tests/core/collections/commands/dropDatabase/test_dropDatabase_wc_wtimeout.py

@@ -10,6 +10,7 @@
 from documentdb_tests.framework.error_codes import FAILED_TO_PARSE_ERROR
 from documentdb_tests.framework.executor import execute_command
 from documentdb_tests.framework.parametrize import pytest_params
+from documentdb_tests.framework.property_checks import Eq
 from documentdb_tests.framework.test_constants import (
     DECIMAL128_INFINITY,
     DECIMAL128_NAN,
@@ -32,49 +33,49 @@
 WRITE_CONCERN_WTIMEOUT_ACCEPTED_TESTS: list[CommandTestCase] = [
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": True}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:bool true should be accepted",
         id="wtimeout_bool_true",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": False}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:bool false should be accepted",
         id="wtimeout_bool_false",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": 2.5}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:double should be accepted",
         id="wtimeout_double",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": Int64(42)}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:Int64 should be accepted",
         id="wtimeout_int64",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": Decimal128("99")}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:Decimal128 should be accepted",
         id="wtimeout_decimal128",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": 5}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:int32 should be accepted",
         id="wtimeout_int32",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": "hello"}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:string should be accepted",
         id="wtimeout_string",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": None}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:null should be treated as omitted and succeed",
         id="wtimeout_null",
     ),
@@ -86,7 +87,7 @@
 WRITE_CONCERN_WTIMEOUT_SPECIAL_VALUES_TESTS: list[CommandTestCase] = [
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": INT32_MAX}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:max int32 should be accepted",
         id="wtimeout_max_int32",
     ),
@@ -95,58 +96,55 @@
             "dropDatabase": 1,
             "writeConcern": {"wtimeout": INT64_MIN},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:Int64 min should be accepted",
-        marks=(pytest.mark.requires(quorum_write_concern=False),),
         id="wtimeout_int64_min",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": FLOAT_NAN}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:NaN should be accepted",
         id="wtimeout_float_nan",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": FLOAT_NEGATIVE_NAN}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:negative NaN should be accepted",
         id="wtimeout_float_neg_nan",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": FLOAT_NEGATIVE_INFINITY}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:-Infinity should be accepted",
-        marks=(pytest.mark.requires(quorum_write_concern=False),),
         id="wtimeout_float_neg_infinity",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": DECIMAL128_NAN}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:Decimal128('NaN') should be accepted",
         id="wtimeout_decimal128_nan",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": DECIMAL128_NEGATIVE_NAN}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:Decimal128 negative NaN should be accepted",
         id="wtimeout_decimal128_neg_nan",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": DECIMAL128_NEGATIVE_INFINITY}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:Decimal128('-Infinity') should be accepted",
-        marks=(pytest.mark.requires(quorum_write_concern=False),),
         id="wtimeout_decimal128_neg_infinity",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": DOUBLE_NEGATIVE_ZERO}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:float -0.0 should be accepted",
         id="wtimeout_float_neg_zero",
     ),
     CommandTestCase(
         command={"dropDatabase": 1, "writeConcern": {"wtimeout": DECIMAL128_NEGATIVE_ZERO}},
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout:Decimal128 -0 should be accepted",
         id="wtimeout_decimal128_neg_zero",
     ),

documentdb_tests/compatibility/tests/core/collections/commands/renameCollection/test_renameCollection_wc_wtimeout.py

@@ -18,6 +18,7 @@
 )
 from documentdb_tests.framework.executor import execute_admin_command
 from documentdb_tests.framework.parametrize import pytest_params
+from documentdb_tests.framework.property_checks import Eq
 from documentdb_tests.framework.test_constants import (
     DECIMAL128_INFINITY,
     DECIMAL128_NAN,
@@ -93,7 +94,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": 0},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=0 should be accepted",
     ),
     CommandTestCase(
@@ -104,7 +105,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": 1000},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=1000 should be accepted",
     ),
     CommandTestCase(
@@ -115,7 +116,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": INT32_OVERFLOW - 1},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=INT32_MAX should be accepted",
     ),
     CommandTestCase(
@@ -126,9 +127,8 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": -1},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=-1 (negative) should be accepted",
-        marks=(pytest.mark.requires(quorum_write_concern=False),),
     ),
     CommandTestCase(
         "wc_wtimeout_neg_infinity",
@@ -138,9 +138,8 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": FLOAT_NEGATIVE_INFINITY},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=-Infinity should be accepted",
-        marks=(pytest.mark.requires(quorum_write_concern=False),),
     ),
     CommandTestCase(
         "wc_wtimeout_nan",
@@ -150,7 +149,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": FLOAT_NAN},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=NaN should be accepted",
     ),
     CommandTestCase(
@@ -161,7 +160,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": "1000"},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout as string should be accepted",
     ),
     CommandTestCase(
@@ -172,7 +171,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": True},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout as bool should be accepted",
     ),
     CommandTestCase(
@@ -183,7 +182,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": None},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=null should be accepted",
     ),
     CommandTestCase(
@@ -194,7 +193,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": [1000]},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout as array should be accepted",
     ),
     CommandTestCase(
@@ -205,7 +204,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": {"ms": 1000}},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout as object should be accepted",
     ),
     CommandTestCase(
@@ -216,7 +215,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": Int64(1000)},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout as Int64 should be accepted",
     ),
     CommandTestCase(
@@ -227,7 +226,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": Decimal128("1000")},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout as Decimal128 should be accepted",
     ),
     CommandTestCase(
@@ -238,7 +237,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": FLOAT_NEGATIVE_NAN},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=negative NaN should be accepted",
     ),
     CommandTestCase(
@@ -249,7 +248,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": DECIMAL128_NAN},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=Decimal128('NaN') should be accepted",
     ),
     CommandTestCase(
@@ -260,7 +259,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": DECIMAL128_NEGATIVE_NAN},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=Decimal128 negative NaN should be accepted",
     ),
     CommandTestCase(
@@ -271,9 +270,8 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": DECIMAL128_NEGATIVE_INFINITY},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=Decimal128('-Infinity') should be accepted",
-        marks=(pytest.mark.requires(quorum_write_concern=False),),
     ),
     CommandTestCase(
         "wc_wtimeout_neg_zero",
@@ -283,7 +281,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": DOUBLE_NEGATIVE_ZERO},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=-0.0 should be accepted",
     ),
     CommandTestCase(
@@ -294,7 +292,7 @@
             "to": f"{ctx.database}.{ctx.collection}_dest",
             "writeConcern": {"wtimeout": DECIMAL128_NEGATIVE_ZERO},
         },
-        expected={"ok": 1.0},
+        expected={"ok": Eq(1.0)},
         msg="wtimeout=Decimal128('-0') should be accepted",
     ),
 ]