ExportAsFileActionTest.java

/*******************************************************************************
 * ATE, Automation Test Engine
 *
 * Copyright 2015, Montreal PROT, or individual contributors as
 * indicated by the @author tags or express copyright attribution
 * statements applied by the authors.  All third-party contributions are
 * distributed under license by Montreal PROT.
 *
 * 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.bigtester.ate.test.model.page.elementaction;

import static org.mockito.Mockito.when;

import org.bigtester.ate.model.io.IDiskFileOperation;
import org.bigtester.ate.model.page.elementaction.ExportAsFileAction;
import org.bigtester.ate.test.BigtesterProjectTest;
import org.mockito.Mockito;
import org.springframework.test.context.ContextConfiguration;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

// TODO: Auto-generated Javadoc
/**
 * This class ExportAsFileAction defines ....
 * @author Peidong Hu
 *
 */
@ContextConfiguration(locations = { "classpath:bigtesterTestNG/testSuite01/loginSuccess.xml" })

public class ExportAsFileActionTest extends BigtesterProjectTest{
	
	/** The Constant FILENAME. */
	final public static String FILENAME = "exportFile"; 
	
	/** The file opr. */
	private transient IDiskFileOperation fileOpr = ateMock(IDiskFileOperation.class);
	
	/**
	 * Gets the file opr.
	 *
	 * @return the fileOpr
	 */
	public final IDiskFileOperation getFileOpr() {
		return fileOpr;
	}
	
	/**
	 * Sets the file opr.
	 *
	 * @param fileOpr the fileOpr to set
	 */
	public final void setFileOpr(IDiskFileOperation fileOpr) {
		this.fileOpr = fileOpr;
	}
	/**
	 * Setup.
	 */
	@BeforeClass
	public void setup() {
		when(getMyMockedDriver().getWebDriverInstance()).thenReturn(getMockedDriver());
		when(getMockedDriver().manage()).thenReturn(getOptions());
		//when(getOptions().getCookies()).thenReturn(cookiesData.getCookies());
		
	}
	
	/**
	 * Do action test.
	 */
	@Test
	public void doActionTest() {
		ExportAsFileAction eafa = new ExportAsFileAction(getMyMockedDriver(), this.FILENAME);
		eafa.doAction(fileOpr);
		Mockito.verify(fileOpr).setExportFileNameWithAbsolutePath(this.FILENAME);
		Mockito.verify(fileOpr).saveToSingleFile();
	}
}