feat(aggregation):

- fix

Author: chenmingyong0423

builder/aggregation/bson_construction.go

@@ -25,8 +25,8 @@ func Sum(key string, expression any) bson.D {
 	return bson.D{bson.E{Key: key, Value: bson.D{bson.E{Key: types.AggregationSum, Value: expression}}}}
 }
 
-func Push(expression any) bson.D {
-	return bson.D{bson.E{Key: types.AggregationPush, Value: expression}}
+func Push(key string, expression any) bson.D {
+	return bson.D{bson.E{Key: key, Value: bson.D{bson.E{Key: types.AggregationPush, Value: expression}}}}
 }
 
 func Avg(key string, expression any) bson.D {

builder/aggregation/bson_construction_test.go

@@ -31,24 +31,9 @@ func TestSum(t *testing.T) {
 }
 
 func TestPush(t *testing.T) {
-	testCases := []struct {
-		name       string
-		expression any
-		want       bson.D
-	}{
-		{
-			name:       "test push",
-			expression: bsonx.NewD().Add("item", "$item").Add("quantity", "$quantity").Build(),
-			want:       bson.D{{Key: "$push", Value: bson.D{{Key: "item", Value: "$item"}, {Key: "quantity", Value: "$quantity"}}}},
-		},
-	}
-
-	for _, tc := range testCases {
-		t.Run(tc.name, func(t *testing.T) {
-			got := Push(tc.expression)
-			assert.Equal(t, tc.want, got)
-		})
-	}
+	t.Run("test push", func(t *testing.T) {
+		assert.Equal(t, bson.D{{Key: "allItems", Value: bson.D{bson.E{Key: "$push", Value: "$item"}}}}, Push("allItems", "$item"))
+	})
 }
 
 func TestAvg(t *testing.T) {