Fix integration tests

artie-labs · Dec 20, 2024 · c631dff · c631dff
1 parent 737623f
commit c631dff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integration_tests/mongo/main.go b/integration_tests/mongo/main.go
@@ -122,7 +122,7 @@ func testTypes(ctx context.Context, db *mongo.Database, mongoCfg config.MongoDB)
 	}
 
 	row := rows[0]
-	expectedPartitionKey := map[string]any{"payload": map[string]any{"id": `{"$oid":"66a95fae3776c2f21f0ff568"}`}}
+	expectedPartitionKey := map[string]any{"id": `{"$oid":"66a95fae3776c2f21f0ff568"}`}
 	expectedPkBytes, err := json.Marshal(expectedPartitionKey)
 	if err != nil {
 		return fmt.Errorf("failed to marshal expected partition key: %w", err)