remove duplicate validation

Signed-off-by: phuhung273 <phuhung.tranpham@gmail.com>

Author: phuhung273

packages/aws-cdk-lib/aws-stepfunctions/lib/states/distributed-map.ts

@@ -3,7 +3,7 @@ import { ItemBatcher } from './distributed-map/item-batcher';
 import { IItemReader } from './distributed-map/item-reader';
 import { ResultWriter, ResultWriterV2 } from './distributed-map/result-writer';
 import { MapBase, MapBaseJsonataOptions, MapBaseJsonPathOptions, MapBaseOptions, MapBaseProps } from './map-base';
-import { Annotations, FeatureFlags, ValidationError } from '../../../core';
+import { Annotations, FeatureFlags } from '../../../core';
 import { FieldUtils } from '../fields';
 import { StateGraph } from '../state-graph';
 import { StateMachineType } from '../state-machine';
@@ -314,11 +314,6 @@ export class DistributedMap extends MapBase implements INextable {
    * Return the Amazon States Language object for this state
    */
   public toStateJson(stateMachineQueryLanguage?: QueryLanguage): object {
-    // https://docs.aws.amazon.com/step-functions/latest/dg/statemachine-structure.html
-    if (stateMachineQueryLanguage == QueryLanguage.JSONATA && this.queryLanguage == QueryLanguage.JSON_PATH) {
-      throw new ValidationError('When StateMachine query language is JSONata, individual states cannot use JSONPath.', this);
-    }
-
     let rendered: any = super.toStateJson(stateMachineQueryLanguage);
     if (rendered.ItemProcessor.ProcessorConfig.ExecutionType) {
       Annotations.of(this).addWarningV2('@aws-cdk/aws-stepfunctions:propertyIgnored', 'Property \'ProcessorConfig.executionType\' is ignored, use the \'mapExecutionType\' in the \'DistributedMap\' class instead.');
@@ -327,9 +322,7 @@ export class DistributedMap extends MapBase implements INextable {
     rendered.ItemProcessor.ProcessorConfig.ExecutionType = this.mapExecutionType;
     // ItemReader and ResultWriter configuration will base on the Map's query language.
     // If Map's query language is not specified, then use state machine's query language.
-    const stateQueryLanguage = stateMachineQueryLanguage == QueryLanguage.JSONATA
-      ? QueryLanguage.JSONATA
-      : this.queryLanguage ?? stateMachineQueryLanguage;
+    const stateQueryLanguage = this.queryLanguage ?? stateMachineQueryLanguage;
 
     const renderedResultWriter = this.renderResultWriter(stateQueryLanguage);
     this.addWarningIfResultWriterIsEmpty(renderedResultWriter);

packages/aws-cdk-lib/aws-stepfunctions/test/distributed-map.test.ts

@@ -1619,8 +1619,6 @@ describe('Distributed Map State', () => {
     expect(() => {
       // GIVEN
       const stack = new cdk.Stack();
-
-      stack.node.setContext(STEPFUNCTIONS_USE_DISTRIBUTED_MAP_RESULT_WRITER_V2, true);
       const writerBucket = new s3.Bucket(stack, 'TestBucket');
 
       // WHEN
@@ -1637,7 +1635,7 @@ describe('Distributed Map State', () => {
         definition: map,
         queryLanguage: stepfunctions.QueryLanguage.JSONATA,
       });
-    }).toThrow(/When StateMachine query language is JSONata, individual states cannot use JSONPath./);
+    }).toThrow(/'queryLanguage' can not be 'JSONPath' if set to 'JSONata' for whole state machine/);
   }),
 
   test('State Machine JSONPath With Distributed Map State and ItemReader, ResultWriterV2 in JSONATA', () => {