briar/test/net/sf/briar/invitation/InvitationWorkerTest.java

package net.sf.briar.invitation;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import junit.framework.TestCase;
import net.sf.briar.TestUtils;
import net.sf.briar.api.invitation.InvitationCallback;
import net.sf.briar.api.invitation.InvitationParameters;

import org.jmock.Expectations;
import org.jmock.Mockery;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class InvitationWorkerTest extends TestCase {

	private final File testDir = TestUtils.getTestDirectory();

	@Before
	public void setUp() {
		testDir.mkdirs();
	}

	@Test
	public void testHaltsIfDestinationDoesNotExist() {
		final File nonExistent = new File(testDir, "does.not.exist");
		Mockery context = new Mockery();
		final InvitationCallback callback =
			context.mock(InvitationCallback.class);
		final InvitationParameters params =
			context.mock(InvitationParameters.class);
		context.checking(new Expectations() {{
			oneOf(params).getChosenLocation();
			will(returnValue(nonExistent));
			oneOf(callback).notFound(nonExistent);
		}});

		new InvitationWorker(callback, params).run();

		context.assertIsSatisfied();
		File[] children = testDir.listFiles();
		assertNotNull(children);
		assertEquals(0, children.length);
	}

	@Test
	public void testHaltsIfDestinationIsNotADirectory() throws IOException {
		final File exists = new File(testDir, "exists");
		TestUtils.createFile(exists, "foo");
		assertFalse(exists.isDirectory());
		Mockery context = new Mockery();
		final InvitationCallback callback =
			context.mock(InvitationCallback.class);
		final InvitationParameters params =
			context.mock(InvitationParameters.class);
		context.checking(new Expectations() {{
			oneOf(params).getChosenLocation();
			will(returnValue(exists));
			oneOf(callback).notDirectory(exists);
		}});

		new InvitationWorker(callback, params).run();

		context.assertIsSatisfied();
		File[] children = testDir.listFiles();
		assertNotNull(children);
		assertEquals(1, children.length);
		assertEquals(exists, children[0]);
	}

	@Test
	public void testCreatesExe() throws IOException {
		testInstallerCreation(true, false);
	}

	@Test
	public void testCreatesJar() throws IOException {
		testInstallerCreation(false, true);
	}

	@Test
	public void testCreatesBoth() throws IOException {
		testInstallerCreation(true, true);
	}

	@Test
	public void testCreatesNeither() throws IOException {
		testInstallerCreation(false, false);
	}

	private void testInstallerCreation(final boolean createExe,
			final boolean createJar) throws IOException {
		final File setup = new File(testDir, "setup.dat");
		TestUtils.createFile(setup, "foo bar baz");
		final File invitation = new File(testDir, "invitation.dat");
		final File exe = new File(testDir, "briar.exe");
		final File jar = new File(testDir, "briar.jar");
		assertTrue(setup.exists());
		assertFalse(invitation.exists());
		assertFalse(exe.exists());
		assertFalse(jar.exists());
		final List<File> expectedFiles = new ArrayList<File>();
		expectedFiles.add(invitation);
		if(createExe) expectedFiles.add(exe);
		if(createJar) expectedFiles.add(jar);
		Mockery context = new Mockery();
		final InvitationCallback callback =
			context.mock(InvitationCallback.class);
		final InvitationParameters params =
			context.mock(InvitationParameters.class);
		context.checking(new Expectations() {{
			oneOf(params).getChosenLocation();
			will(returnValue(testDir));
			allowing(callback).isCancelled();
			will(returnValue(false));
			oneOf(params).getPassword();
			will(returnValue(new char[] {'x', 'y', 'z', 'z', 'y'}));
			oneOf(callback).encryptingFile(invitation);
			oneOf(params).shouldCreateExe();
			will(returnValue(createExe));
			oneOf(params).shouldCreateJar();
			will(returnValue(createJar));
			if(createExe) {
				oneOf(params).getSetupDat();
				will(returnValue(setup));
				oneOf(callback).copyingFile(exe);
			}
			if(createJar) {
				oneOf(params).getSetupDat();
				will(returnValue(setup));
				oneOf(callback).copyingFile(jar);
			}
			oneOf(callback).created(expectedFiles);
		}});

		new InvitationWorker(callback, params).run();

		assertTrue(invitation.exists());
		assertEquals(createExe, exe.exists());
		assertEquals(createJar, jar.exists());
		if(createExe) assertEquals(exe.length(), setup.length());
		if(createJar) assertEquals(jar.length(), setup.length());
	}

	@After
	public void tearDown() throws IOException {
		TestUtils.delete(testDir);
	}
}