WIP

mongodb · Oct 12, 2023 · 360f89e · 360f89e
1 parent 284715c
commit 360f89e
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/mongo/integration/unified_spec_test.go b/mongo/integration/unified_spec_test.go
@@ -162,14 +162,32 @@ type expectation struct {
 }
 
 type outcome struct {
- Collection *outcomeCollection `bson:"collection"`
+ Collection *outcomeCollection
+ CollectionData []*collectionData
+}
+
+func (outcome *outcome) UnmarshalBSON(b []byte) error {
+ var v struct {
+ Collection *outcomeCollection `bson:"collection"`
+ }
+ err := bson.UnmarshalWithRegistry(specTestRegistry, b, &v)
+ if err == nil {
+ outcome.Collection = v.Collection
+ }
+ return err
 }
 
 type outcomeCollection struct {
  Name string `bson:"name"`
  Data interface{} `bson:"data"`
 }
 
+type collectionData struct {
+ DatabaseName string `bson:"databaseName"`
+ CollectionName string `bson:"collectionName"`
+ Documents []bson.Raw `bson:"documents"`
+}
+
 type operationError struct {
  ErrorContains *string `bson:"errorContains"`
  ErrorCodeName *string `bson:"errorCodeName"`