use of com.adobe.cq.dam.cfm.DataType in project aem-core-wcm-components by Adobe-Marketing-Cloud.
the class ContentFragmentMockAdapter method getMockElement.
/**
* Creates a mock of a content element for a text-only (if {@code model} is {@code null}) or structured
* (if {@code model} is not {@code null}) content fragment.
*/
private ContentElement getMockElement(Resource resource, String name, Resource model) {
// get the respective element
MockElement element;
if (model == null) {
element = getTextOnlyElement(resource, name);
} else {
element = getStructuredElement(resource, model, name);
}
if (element == null) {
return null;
}
/* create mock objects */
// mock data type
DataType dataType = mock(DataType.class, withSettings().lenient());
when(dataType.isMultiValue()).thenReturn(element.isMultiValued);
when(dataType.getTypeString()).thenReturn(element.typeString);
// mock fragment data
FragmentData data = mock(FragmentData.class, withSettings().lenient());
when(data.getValue()).thenReturn(element.isMultiValued ? element.values : element.values[0]);
when(data.getValue(String.class)).thenReturn(element.values[0]);
when(data.getValue(String[].class)).thenReturn(element.values);
when(data.getContentType()).thenReturn(element.contentType);
when(data.getDataType()).thenReturn(dataType);
// mock content element
ContentElement contentElement = mock(ContentElement.class, withSettings().lenient());
when(contentElement.getName()).thenReturn(element.name);
when(contentElement.getTitle()).thenReturn(element.title);
when(contentElement.getContent()).thenReturn(element.values[0]);
when(contentElement.getContentType()).thenReturn(element.contentType);
when(contentElement.getValue()).thenReturn(data);
// mock variations
Map<String, ContentVariation> variations = new LinkedHashMap<>();
for (MockVariation variation : element.variations.values()) {
FragmentData variationData = mock(FragmentData.class, withSettings().lenient());
when(variationData.getValue()).thenReturn(element.isMultiValued ? variation.values : variation.values[0]);
when(variationData.getValue(String.class)).thenReturn(variation.values[0]);
when(variationData.getValue(String[].class)).thenReturn(variation.values);
when(variationData.getContentType()).thenReturn(variation.contentType);
when(variationData.getDataType()).thenReturn(dataType);
ContentVariation contentVariation = mock(ContentVariation.class, withSettings().lenient());
when(contentVariation.getName()).thenReturn(variation.name);
when(contentVariation.getTitle()).thenReturn(variation.title);
when(contentVariation.getContent()).thenReturn(variation.values[0]);
when(contentVariation.getContentType()).thenReturn(variation.contentType);
when(contentVariation.getValue()).thenReturn(variationData);
variations.put(variation.name, contentVariation);
}
when(contentElement.getVariations()).thenReturn(variations.values().iterator());
when(contentElement.getVariation(any(String.class))).thenAnswer(invocation -> {
String variationName = invocation.getArgument(0);
return variations.get(variationName);
});
return contentElement;
}
Aggregations