com.seleniumtests.ut.uipage.htmlelements.TestScreenZone.java Source code

Introduction

Here is the source code for com.seleniumtests.ut.uipage.htmlelements.TestScreenZone.java

Source

/**
 * Orignal work: Copyright 2015 www.seleniumtests.com
 * Modified work: Copyright 2016 www.infotel.com
 *             Copyright 2017-2019 B.Hecquet
 *
 * 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 com.seleniumtests.ut.uipage.htmlelements;

import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.awt.AWTException;
import java.awt.event.KeyEvent;
import java.io.File;

import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.openqa.selenium.Rectangle;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.testng.Assert;
import org.testng.ITestContext;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.Test;

import com.seleniumtests.MockitoTest;
import com.seleniumtests.customexception.ImageSearchException;
import com.seleniumtests.customexception.ScenarioException;
import com.seleniumtests.driver.CustomEventFiringWebDriver;
import com.seleniumtests.driver.screenshots.ScreenshotUtil;
import com.seleniumtests.driver.screenshots.ScreenshotUtil.Target;
import com.seleniumtests.uipage.htmlelements.ScreenZone;
import com.seleniumtests.util.imaging.ImageDetector;

@PrepareForTest(CustomEventFiringWebDriver.class)
public class TestScreenZone extends MockitoTest {

    @Mock
    ImageDetector imageDetector;

    @Mock
    ScreenshotUtil screenshotUtil;

    @InjectMocks
    ScreenZone screenZone = new ScreenZone();

    @Test(groups = { "ut" })
    public void testClick() {
        ScreenZone picElement = spy(screenZone);
        picElement.setObjectPictureFile(new File(""));

        PowerMockito.mockStatic(CustomEventFiringWebDriver.class);
        doReturn(screenshotUtil).when(picElement).getScreenshotUtil();
        when(screenshotUtil.capture(Target.SCREEN, File.class, true)).thenReturn(new File(""));
        when(imageDetector.getDetectedRectangle()).thenReturn(new Rectangle(10, 10, 100, 50));
        when(imageDetector.getSizeRatio()).thenReturn(1.0);

        picElement.click();
        verify(picElement).moveAndLeftClick(35, 60);
    }

    @Test(groups = { "ut" })
    public void testDoubleClick() {
        ScreenZone picElement = spy(screenZone);
        picElement.setObjectPictureFile(new File(""));

        PowerMockito.mockStatic(CustomEventFiringWebDriver.class);
        doReturn(screenshotUtil).when(picElement).getScreenshotUtil();
        when(screenshotUtil.capture(Target.SCREEN, File.class, true)).thenReturn(new File(""));
        when(imageDetector.getDetectedRectangle()).thenReturn(new Rectangle(10, 10, 100, 50));
        when(imageDetector.getSizeRatio()).thenReturn(1.0);

        picElement.doubleClickAt(0, 0);
        verify(picElement).moveAndDoubleClick(35, 60);
    }

    @Test(groups = { "ut" })
    public void testRightClick() {
        ScreenZone picElement = spy(screenZone);
        picElement.setObjectPictureFile(new File(""));

        PowerMockito.mockStatic(CustomEventFiringWebDriver.class);
        doReturn(screenshotUtil).when(picElement).getScreenshotUtil();
        when(screenshotUtil.capture(Target.SCREEN, File.class, true)).thenReturn(new File(""));
        when(imageDetector.getDetectedRectangle()).thenReturn(new Rectangle(10, 10, 100, 50));
        when(imageDetector.getSizeRatio()).thenReturn(1.0);

        picElement.rightClickAt(0, 0);
        verify(picElement).moveAndRightClick(35, 60);
    }

    @Test(groups = { "ut" })
    public void testSendKeys() {
        ScreenZone picElement = spy(screenZone);
        picElement.setObjectPictureFile(new File(""));

        PowerMockito.mockStatic(CustomEventFiringWebDriver.class);
        doReturn(screenshotUtil).when(picElement).getScreenshotUtil();
        when(screenshotUtil.capture(Target.SCREEN, File.class, true)).thenReturn(new File(""));
        when(imageDetector.getDetectedRectangle()).thenReturn(new Rectangle(10, 10, 100, 50));
        when(imageDetector.getSizeRatio()).thenReturn(1.0);

        picElement.sendKeys(0, 0, KeyEvent.VK_0);
        verify(picElement).moveAndLeftClick(35, 60);
    }

    @Test(groups = { "ut" })
    public void testPictureNotVisible() throws AWTException {
        ScreenZone picElement = spy(screenZone);
        picElement.setObjectPictureFile(new File(""));
        doReturn(screenshotUtil).when(picElement).getScreenshotUtil();
        when(screenshotUtil.capture(Target.SCREEN, File.class, true)).thenReturn(new File(""));
        doThrow(ImageSearchException.class).when(imageDetector).detectExactZoneWithScale();

        Assert.assertFalse(picElement.isElementPresent());

        verify(picElement).findElement();

    }

    @Test(groups = { "ut" })
    public void testPictureNotVisibleWithReplay() throws AWTException {
        ScreenZone picElement = spy(screenZone);
        picElement.setObjectPictureFile(new File(""));
        doReturn(screenshotUtil).when(picElement).getScreenshotUtil();
        when(screenshotUtil.capture(Target.SCREEN, File.class, true)).thenReturn(new File(""));
        doThrow(ImageSearchException.class).when(imageDetector).detectExactZoneWithScale();

        Assert.assertFalse(picElement.isElementPresent(350));

        verify(picElement, times(2)).findElement();
    }

    @Test(groups = { "ut" })
    public void testPictureVisible() throws AWTException {
        ScreenZone picElement = spy(screenZone);
        picElement.setObjectPictureFile(new File(""));
        doReturn(screenshotUtil).when(picElement).getScreenshotUtil();
        when(screenshotUtil.capture(Target.SCREEN, File.class, true)).thenReturn(new File(""));
        when(imageDetector.getDetectedRectangle()).thenReturn(new Rectangle(10, 10, 100, 50));
        when(imageDetector.getSizeRatio()).thenReturn(1.0);

        Assert.assertTrue(picElement.isElementPresent(2000));
        verify(picElement).findElement();
    }

    @Test(groups = { "ut" }, expectedExceptions = ScenarioException.class)
    public void testTapOnDesktop() throws AWTException {
        screenZone.tap();
    }

    @Test(groups = { "ut" }, expectedExceptions = ScenarioException.class)
    public void testSwipeOnDesktop() throws AWTException {
        screenZone.swipe(0, 0);
    }

    @AfterMethod(groups = { "ut" }, alwaysRun = true)
    public void reset(ITestContext testNGCtx) {
        initThreadContext(testNGCtx);
    }
}