com.applitools.eyes.selenium.universal.mapper.TFloatingRegionMapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of eyes-selenium-java5 Show documentation
Applitools Eyes SDK for Selenium Java WebDriver
There is a newer version: 5.73.0
package com.applitools.eyes.selenium.universal.mapper;

import java.util.List;
import java.util.stream.Collectors;

import com.applitools.eyes.fluent.FloatingRegionByRectangle;
import com.applitools.eyes.fluent.GetRegion;
import com.applitools.eyes.selenium.fluent.FloatingRegionByElement;
import com.applitools.eyes.selenium.fluent.FloatingRegionBySelector;
import com.applitools.eyes.selenium.universal.dto.*;
import com.applitools.eyes.universal.dto.*;
import com.applitools.eyes.universal.mapper.RectangleRegionMapper;

/**
 * floating region mapper
 */
public class TFloatingRegionMapper {

  public static TFloatingRegion toTFloatingRegionDto(GetRegion getFloatingRegion) {
    if (getFloatingRegion == null) {
      return null;
    }

    if (getFloatingRegion instanceof FloatingRegionByRectangle) {
      FloatingRegionByRectangle floatingRegionByRectangle = (FloatingRegionByRectangle) getFloatingRegion;

      RectangleRegionDto rectangleRegionDto = RectangleRegionMapper
              .toRectangleRegionDto(((FloatingRegionByRectangle) getFloatingRegion).getRegion());

      RectangleFloatingRegionDto response = new RectangleFloatingRegionDto();
      response.setRegion(rectangleRegionDto);
      response.setMaxUpOffset(floatingRegionByRectangle.getMaxUpOffset());
      response.setMaxDownOffset(floatingRegionByRectangle.getMaxDownOffset());
      response.setMaxLeftOffset(floatingRegionByRectangle.getMaxLeftOffset());
      response.setMaxRightOffset(floatingRegionByRectangle.getMaxRightOffset());
      response.setRegionId(floatingRegionByRectangle.getRegion().getRegionId());

      return response;
    } else if (getFloatingRegion instanceof FloatingRegionByElement) {
      FloatingRegionByElement floatingRegionByElement = (FloatingRegionByElement) getFloatingRegion;

      ElementRegionDto elementRegionDto = ElementRegionMapper.toElementRegionDto(floatingRegionByElement.getElement());

      ElementFloatingRegionDto elementFloatingRegionDto = new ElementFloatingRegionDto();
      elementFloatingRegionDto.setRegion(elementRegionDto);
      elementFloatingRegionDto.setMaxUpOffset(floatingRegionByElement.getMaxUpOffset());
      elementFloatingRegionDto.setMaxDownOffset(floatingRegionByElement.getMaxDownOffset());
      elementFloatingRegionDto.setMaxLeftOffset(floatingRegionByElement.getMaxLeftOffset());
      elementFloatingRegionDto.setMaxRightOffset(floatingRegionByElement.getMaxRightOffset());
      elementFloatingRegionDto.setRegionId(floatingRegionByElement.getRegionId());

      return elementFloatingRegionDto;
    } else if (getFloatingRegion instanceof FloatingRegionBySelector) {
      FloatingRegionBySelector floatingRegionBySelector = (FloatingRegionBySelector) getFloatingRegion;

      SelectorRegionDto selectorRegionDto = SelectorRegionMapper
              .toSelectorRegionDto(floatingRegionBySelector.getSelector());

      SelectorFloatingRegionDto selectorFloatingRegionDto = new SelectorFloatingRegionDto();
      selectorFloatingRegionDto.setRegion(selectorRegionDto);
      selectorFloatingRegionDto.setMaxUpOffset(floatingRegionBySelector.getMaxUpOffset());
      selectorFloatingRegionDto.setMaxDownOffset(floatingRegionBySelector.getMaxDownOffset());
      selectorFloatingRegionDto.setMaxLeftOffset(floatingRegionBySelector.getMaxLeftOffset());
      selectorFloatingRegionDto.setMaxRightOffset(floatingRegionBySelector.getMaxRightOffset());
      selectorFloatingRegionDto.setRegionId(floatingRegionBySelector.getRegionId());

      return selectorFloatingRegionDto;
    }

    return null;

  }

  public static List toTFloatingRegionDtoList(List getFloatingRegionList) {
    if (getFloatingRegionList == null || getFloatingRegionList.isEmpty()) {
      return null;
    }

    return getFloatingRegionList.stream().map(TFloatingRegionMapper::toTFloatingRegionDto).collect(Collectors.toList());
  }
}