ExecutableObjectInvisibleMessageTest.java

package com.jenjinstudios.world.client.message;

import com.jenjinstudios.core.io.Message;
import com.jenjinstudios.core.io.MessageRegistry;
import com.jenjinstudios.world.World;
import com.jenjinstudios.world.client.WorldClient;
import com.jenjinstudios.world.collections.WorldObjectList;
import org.testng.annotations.Test;

import static org.mockito.Mockito.*;

/**
 * @author Caleb Brinkman
 */
public class ExecutableObjectInvisibleMessageTest
{
	@Test
	public void testMessageExecution() throws Exception {
		MessageRegistry messageRegistry = MessageRegistry.getInstance();
		Message objectInvisibleMessage = messageRegistry.createMessage("ObjectInvisibleMessage");
		objectInvisibleMessage.setArgument("id", 100);

		WorldClient worldClient = mock(WorldClient.class);
		World world = mock(World.class);
		WorldObjectList worldObjectMap = mock(WorldObjectList.class);
		when(world.getWorldObjects()).thenReturn(worldObjectMap);
		when(worldClient.getWorld()).thenReturn(world);

		ExecutableObjectInvisibleMessage message =
			  new ExecutableObjectInvisibleMessage(worldClient, objectInvisibleMessage);
		message.runImmediate();
		message.runDelayed();

		verify(worldObjectMap).remove(100);
	}
}