content: /*
   * Copyright 2010-2025 Hyland Software, Inc. and its affiliates.
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   *     http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.activiti.editor.language.xml;
  
  import static org.assertj.core.api.Assertions.assertThat;
  
  import java.util.List;
  import java.util.Map;
  import org.activiti.bpmn.model.ActivitiListener;
  import org.activiti.bpmn.model.BoundaryEvent;
  import org.activiti.bpmn.model.BpmnModel;
  import org.activiti.bpmn.model.ExtensionAttribute;
  import org.activiti.bpmn.model.ExtensionElement;
  import org.activiti.bpmn.model.FieldExtension;
  import org.activiti.bpmn.model.FlowElement;
  import org.activiti.bpmn.model.ImplementationType;
  import org.activiti.bpmn.model.Process;
  import org.activiti.bpmn.model.ServiceTask;
  import org.activiti.bpmn.model.TimerEventDefinition;
  import org.junit.jupiter.api.Test;
  
  public class CustomExtensionsConverterTest extends AbstractConverterTest {
  
      @Test
      public void convertXMLToModel() throws Exception {
          BpmnModel bpmnModel = readXMLFile();
          validateModel(bpmnModel);
      }
  
      @Test
      public void convertModelToXML() throws Exception {
          BpmnModel bpmnModel = readXMLFile();
          BpmnModel parsedModel = exportAndReadXMLFile(bpmnModel);
          validateModel(parsedModel);
          deployProcess(parsedModel);
      }
  
      protected String getResource() {
          return "customextensionsmodel.bpmn";
      }
  
      private void validateModel(BpmnModel model) {
          Process process = model.getMainProcess();
          assertThat(process.getAttributes()).isNotNull();
          assertThat(process.getAttributes()).hasSize(1);
          List<ExtensionAttribute> attributes = process.getAttributes().get("version");
          assertThat(attributes).isNotNull();
          assertThat(attributes).hasSize(1);
          ExtensionAttribute attribute = attributes.get(0);
          // custom:version = "9"
          assertThat(attribute).isNotNull();
          assertThat(attribute.getNamespace()).isEqualTo("http://custom.org/bpmn");
          assertThat(attribute.getNamespacePrefix()).isEqualTo("custom");
          assertThat(attribute.getName()).isEqualTo("version");
          assertThat(attribute.getValue()).isEqualTo("9");
  
          List<ActivitiListener> listeners = model.getMainProcess().getExecutionListeners();
          validateExecutionListeners(listeners);
          Map<String, List<ExtensionElement>> extensionElementMap = model.getMainProcess().getExtensionElements();
          validateExtensionElements(extensionElementMap);
  
          FlowElement flowElement = model.getMainProcess().getFlowElement("servicetask");
          assertThat(flowElement).isNotNull();
          assertThat(flowElement).isInstanceOf(ServiceTask.class);
          assertThat(flowElement.getId()).isEqualTo("servicetask");
          ServiceTask serviceTask = (ServiceTask) flowElement;
          assertThat(serviceTask.getId()).isEqualTo("servicetask");
          assertThat(serviceTask.getName()).isEqualTo("Service task");
  
          List<FieldExtension> fields = serviceTask.getFieldExtensions();
          assertThat(fields).hasSize(2);
          FieldExtension field = (FieldExtension) fields.get(0);
          assertThat(field.getFieldName()).isEqualTo("testField");
          assertThat(field.getStringValue()).isEqualTo("test");
          field = (FieldExtension) fields.get(1);
          assertThat(field.getFieldName()).isEqualTo("testField2");
          assertThat(field.getExpression()).isEqualTo("${test}");
  
          listeners = serviceTask.getExecutionListeners();
          validateExecutionListeners(listeners);
  
          extensionElementMap = serviceTask.getExtensionElements();
          validateExtensionElements(extensionElementMap);
  
          assertThat(serviceTask.getBoundaryEvents()).hasSize(1);
          BoundaryEvent boundaryEvent = serviceTask.getBoundaryEvents().get(0);
          assertThat(boundaryEvent.getId()).isEqualTo("timerEvent");
          assertThat(boundaryEvent.getEventDefinitions()).hasSize(1);
          assertThat(boundaryEvent.getEventDefinitions().get(0)).isInstanceOf(TimerEventDefinition.class);
          extensionElementMap = boundaryEvent.getEventDefinitions().get(0).getExtensionElements();
          validateExtensionElements(extensionElementMap);
      }
  
      protected void validateExecutionListeners(List<ActivitiListener> listeners) {
          assertThat(listeners).hasSize(3);
          ActivitiListener listener = (ActivitiListener) listeners.get(0);
          assertThat(ImplementationType.IMPLEMENTATION_TYPE_CLASS.equals(listener.getImplementationType())).isTrue();
          assertThat(listener.getImplementation()).isEqualTo("org.test.TestClass");
          assertThat(listener.getEvent()).isEqualTo("start");
          assertThat(listener.getOnTransaction()).isEqualTo("before-commit");
          assertThat(listener.getCustomPropertiesResolverImplementation()).isEqualTo("org.test.TestResolverClass");
          listener = (ActivitiListener) listeners.get(1);
          assertThat(ImplementationType.IMPLEMENTATION_TYPE_EXPRESSION.equals(listener.getImplementationType())).isTrue();
          assertThat(listener.getImplementation()).isEqualTo("${testExpression}");
          assertThat(listener.getEvent()).isEqualTo("end");
          assertThat(listener.getOnTransaction()).isEqualTo("committed");
          assertThat(listener.getCustomPropertiesResolverImplementation()).isEqualTo("${testResolverExpression}");
          listener = (ActivitiListener) listeners.get(2);
          assertThat(
              ImplementationType.IMPLEMENTATION_TYPE_DELEGATEEXPRESSION.equals(listener.getImplementationType())
          ).isTrue();
          assertThat(listener.getImplementation()).isEqualTo("${delegateExpression}");
          assertThat(listener.getEvent()).isEqualTo("start");
          assertThat(listener.getOnTransaction()).isEqualTo("rolled-back");
          assertThat(listener.getCustomPropertiesResolverImplementation()).isEqualTo("${delegateResolverExpression}");
      }
  
      protected void validateExtensionElements(Map<String, List<ExtensionElement>> extensionElementMap) {
          assertThat(extensionElementMap).hasSize(1);
  
          List<ExtensionElement> extensionElements = extensionElementMap.get("test");
          assertThat(extensionElements).hasSize(2);
  
          ExtensionElement extensionElement = extensionElements.get(0);
          assertThat(extensionElement).isNotNull();
          assertThat(extensionElement.getName()).isEqualTo("test");
          assertThat(extensionElement.getNamespacePrefix()).isEqualTo("custom");
          assertThat(extensionElement.getNamespace()).isEqualTo("http://custom.org/bpmn");
          assertThat(extensionElement.getAttributes()).hasSize(2);
  
          List<ExtensionAttribute> attributes = extensionElement.getAttributes().get("id");
          assertThat(attributes).hasSize(1);
          ExtensionAttribute attribute = attributes.get(0);
          assertThat(attribute).isNotNull();
          assertThat(attribute.getName()).isEqualTo("id");
          assertThat(attribute.getValue()).isEqualTo("test");
          assertThat(attribute.getNamespace()).isNull();
          assertThat(attribute.getNamespacePrefix()).isNull();
  
          attributes = extensionElement.getAttributes().get("name");
          assertThat(attributes).hasSize(1);
          attribute = attributes.get(0);
          assertThat(attribute).isNotNull();
          assertThat(attribute.getName()).isEqualTo("name");
          assertThat(attribute.getValue()).isEqualTo("test");
  
          assertThat(extensionElement.getChildElements()).hasSize(2);
          List<ExtensionElement> childExtensions = extensionElement.getChildElements().get("name");
          assertThat(childExtensions).hasSize(2);
  
          ExtensionElement childExtension = childExtensions.get(0);
          assertThat(childExtension).isNotNull();
          assertThat(childExtension.getName()).isEqualTo("name");
          assertThat(childExtension.getNamespacePrefix()).isEqualTo("custom");
          assertThat(childExtension.getNamespace()).isEqualTo("http://custom.org/bpmn");
          assertThat(childExtension.getAttributes()).hasSize(0);
          assertThat(childExtension.getChildElements()).hasSize(1);
  
          List<ExtensionElement> subChildExtensions = childExtension.getChildElements().get("test");
          assertThat(subChildExtensions).hasSize(1);
  
          childExtension = subChildExtensions.get(0);
          assertThat(childExtension).isNotNull();
          assertThat(childExtension.getName()).isEqualTo("test");
          assertThat(childExtension.getNamespacePrefix()).isEqualTo("custom");
          assertThat(childExtension.getNamespace()).isEqualTo("http://custom.org/bpmn");
          assertThat(childExtension.getAttributes()).hasSize(0);
          assertThat(childExtension.getChildElements()).hasSize(0);
          assertThat(childExtension.getElementText()).isEqualTo("test");
  
          childExtensions = extensionElement.getChildElements().get("description");
          assertThat(childExtensions).hasSize(1);
          childExtension = childExtensions.get(0);
          assertThat(childExtension).isNotNull();
          assertThat(childExtension.getName()).isEqualTo("description");
          assertThat(childExtension.getAttributes()).hasSize(1);
          attributes = childExtension.getAttributes().get("id");
          attribute = attributes.get(0);
          assertThat(attribute).isNotNull();
          assertThat(attribute.getName()).isEqualTo("id");
          assertThat(attribute.getValue()).isEqualTo("test");
          assertThat(attribute.getNamespacePrefix()).isEqualTo("custom2");
          assertThat(attribute.getNamespace()).isEqualTo("http://custom2.org/bpmn");
  
          extensionElement = extensionElements.get(1);
          assertThat(extensionElement).isNotNull();
          assertThat(extensionElement.getName()).isEqualTo("test");
          assertThat(extensionElement.getNamespacePrefix()).isEqualTo("custom");
          assertThat(extensionElement.getNamespace()).isEqualTo("http://custom.org/bpmn");
          assertThat(extensionElement.getAttributes()).hasSize(2);
  
          attributes = extensionElement.getAttributes().get("id");
          assertThat(attributes).hasSize(1);
          attribute = attributes.get(0);
          assertThat(attribute).isNotNull();
          assertThat(attribute.getName()).isEqualTo("id");
          assertThat(attribute.getValue()).isEqualTo("test2");
          assertThat(attribute.getNamespace()).isNull();
          assertThat(attribute.getNamespacePrefix()).isNull();
  
          attributes = extensionElement.getAttributes().get("name");
          assertThat(attributes).hasSize(1);
          attribute = attributes.get(0);
          assertThat(attribute).isNotNull();
          assertThat(attribute.getName()).isEqualTo("name");
          assertThat(attribute.getValue()).isEqualTo("test2");
      }
  }
  