package me.scrobble.example.powermock.testng;

import static org.mockito.Mockito.when;

import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.testng.PowerMockObjectFactory;
import org.testng.Assert;
import org.testng.IObjectFactory;
import org.testng.annotations.ObjectFactory;
import org.testng.annotations.Test;


@PrepareForTest(StaticFactory.class)
public class ObjectFactoryExample {

	/**
	 * We need a special {@link IObjectFactory}.
	 * 
	 * @return {@link PowerMockObjectFactory}.
	 */
	@ObjectFactory
	public IObjectFactory getObjectFactory() {
		return new org.powermock.modules.testng.PowerMockObjectFactory();
	}

	/**
	 * Test if it is possible to create a cipher using
	 * 
	 * @throws Exception
	 *             Thrown if something went wrong.
	 */
	@Test
	public void test() throws Exception {
		Assert.assertEquals("AES", StaticFactory.getMethod());
		PowerMockito.mockStatic(StaticFactory.class);
		when(StaticFactory.getMethod()).thenReturn("DES");
		Assert.assertEquals("DES", StaticFactory.getMethod());
	}
}