Java Mockery類代碼示例

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testEmptyDriveIsIgnored() throws Exception {
	Mockery context = new Mockery() {{
		setThreadingPolicy(new Synchroniser());
	}};
	final Executor executor = context.mock(Executor.class);
	final SimplexPluginCallback callback =
			context.mock(SimplexPluginCallback.class);
	final RemovableDriveFinder finder =
			context.mock(RemovableDriveFinder.class);
	final RemovableDriveMonitor monitor =
			context.mock(RemovableDriveMonitor.class);

	context.checking(new Expectations() {{
		oneOf(monitor).start(with(any(Callback.class)));
	}});

	RemovableDrivePlugin plugin = new RemovableDrivePlugin(executor,
			callback, finder, monitor, 0);
	plugin.start();

	plugin.driveInserted(testDir);

	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testFilenames() {
	Mockery context = new Mockery() {{
		setThreadingPolicy(new Synchroniser());
	}};
	final Executor executor = context.mock(Executor.class);
	final SimplexPluginCallback callback =
			context.mock(SimplexPluginCallback.class);
	final RemovableDriveFinder finder =
			context.mock(RemovableDriveFinder.class);
	final RemovableDriveMonitor monitor =
			context.mock(RemovableDriveMonitor.class);

	RemovableDrivePlugin plugin = new RemovableDrivePlugin(executor,
			callback, finder, monitor, 0);

	assertFalse(plugin.isPossibleConnectionFilename("abcdefg.dat"));
	assertFalse(plugin.isPossibleConnectionFilename("abcdefghi.dat"));
	assertFalse(plugin.isPossibleConnectionFilename("abcdefgh_dat"));
	assertFalse(plugin.isPossibleConnectionFilename("abcdefgh.rat"));
	assertTrue(plugin.isPossibleConnectionFilename("abcdefgh.dat"));
	assertTrue(plugin.isPossibleConnectionFilename("ABCDEFGH.DAT"));

	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testEmptyFramesAreSkipped() throws Exception {
	Mockery context = new Mockery();
	final StreamDecrypter decrypter = context.mock(StreamDecrypter.class);
	context.checking(new Expectations() {{
		oneOf(decrypter).readFrame(with(any(byte[].class)));
		will(returnValue(0)); // Empty frame
		oneOf(decrypter).readFrame(with(any(byte[].class)));
		will(returnValue(2)); // Non-empty frame with two payload bytes
		oneOf(decrypter).readFrame(with(any(byte[].class)));
		will(returnValue(0)); // Empty frame
		oneOf(decrypter).readFrame(with(any(byte[].class)));
		will(returnValue(-1)); // No more frames
	}});
	StreamReaderImpl r = new StreamReaderImpl(decrypter);
	assertEquals(0, r.read()); // Skip the first empty frame, read a byte
	assertEquals(0, r.read()); // Read another byte
	assertEquals(-1, r.read()); // Skip the second empty frame, reach EOF
	assertEquals(-1, r.read()); // Still at EOF
	context.assertIsSatisfied();
	r.close();
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testMultipleReadsPerFrame() throws Exception {
	Mockery context = new Mockery();
	final StreamDecrypter decrypter = context.mock(StreamDecrypter.class);
	context.checking(new Expectations() {{
		oneOf(decrypter).readFrame(with(any(byte[].class)));
		will(returnValue(MAX_PAYLOAD_LENGTH)); // Nice long frame
		oneOf(decrypter).readFrame(with(any(byte[].class)));
		will(returnValue(-1)); // No more frames
	}});
	StreamReaderImpl r = new StreamReaderImpl(decrypter);
	byte[] buf = new byte[MAX_PAYLOAD_LENGTH / 2];
	// Read the first half of the payload
	assertEquals(MAX_PAYLOAD_LENGTH / 2, r.read(buf));
	// Read the second half of the payload
	assertEquals(MAX_PAYLOAD_LENGTH / 2, r.read(buf));
	// Reach EOF
	assertEquals(-1, r.read(buf, 0, buf.length));
	context.assertIsSatisfied();
	r.close();
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testFlushWithoutBufferedDataWritesFrameAndFlushes()
		throws Exception {
	Mockery context = new Mockery();
	final StreamEncrypter encrypter = context.mock(StreamEncrypter.class);
	StreamWriterImpl w = new StreamWriterImpl(encrypter);
	context.checking(new Expectations() {{
		// Write a non-final frame with an empty payload
		oneOf(encrypter).writeFrame(with(any(byte[].class)), with(0),
				with(0), with(false));
		// Flush the stream
		oneOf(encrypter).flush();
	}});
	w.flush();
	context.assertIsSatisfied();

	// Clean up
	context.checking(new Expectations() {{
		// Closing the writer writes a final frame and flushes again
		oneOf(encrypter).writeFrame(with(any(byte[].class)), with(0),
				with(0), with(true));
		oneOf(encrypter).flush();
	}});
	w.close();
	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testSingleByteWritesWriteFullFrame() throws Exception {
	Mockery context = new Mockery();
	final StreamEncrypter encrypter = context.mock(StreamEncrypter.class);
	StreamWriterImpl w = new StreamWriterImpl(encrypter);
	context.checking(new Expectations() {{
		// Write a full non-final frame
		oneOf(encrypter).writeFrame(with(any(byte[].class)),
				with(MAX_PAYLOAD_LENGTH), with(0), with(false));
	}});
	for (int i = 0; i < MAX_PAYLOAD_LENGTH; i++) w.write(0);
	context.assertIsSatisfied();

	// Clean up
	context.checking(new Expectations() {{
		// Closing the writer writes a final frame and flushes again
		oneOf(encrypter).writeFrame(with(any(byte[].class)), with(0),
				with(0), with(true));
		oneOf(encrypter).flush();
	}});
	w.close();
	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testLargeMultiByteWriteWritesFullFrames() throws Exception {
	Mockery context = new Mockery();
	final StreamEncrypter encrypter = context.mock(StreamEncrypter.class);
	StreamWriterImpl w = new StreamWriterImpl(encrypter);
	context.checking(new Expectations() {{
		// Write two full non-final frames
		exactly(2).of(encrypter).writeFrame(with(any(byte[].class)),
				with(MAX_PAYLOAD_LENGTH), with(0), with(false));
		// Write a final frame with a one-byte payload
		oneOf(encrypter).writeFrame(with(any(byte[].class)), with(1),
				with(0), with(true));
		// Flush the stream
		oneOf(encrypter).flush();
	}});
	// Write two full payloads using one large multi-byte write
	byte[] b = new byte[MAX_PAYLOAD_LENGTH * 2 + 1];
	w.write(b);
	// There should be one byte left in the buffer
	w.close();
	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入依賴的package包/類
private void testCannotStartTransactionDuringTransaction(
		boolean firstTxnReadOnly, boolean secondTxnReadOnly)
		throws Exception {
	Mockery context = new Mockery();
	@SuppressWarnings("unchecked")
	final Database<Object> database = context.mock(Database.class);
	final ShutdownManager shutdown = context.mock(ShutdownManager.class);
	final EventBus eventBus = context.mock(EventBus.class);

	context.checking(new Expectations() {{
		oneOf(database).startTransaction();
		will(returnValue(txn));
	}});

	DatabaseComponent db = createDatabaseComponent(database, eventBus,
			shutdown);

	assertNotNull(db.startTransaction(firstTxnReadOnly));
	try {
		db.startTransaction(secondTxnReadOnly);
		fail();
	} finally {
		context.assertIsSatisfied();
	}
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testEmptyFramesAreSkippedWithBuffer() throws Exception {
	Mockery context = new Mockery();
	final StreamDecrypter decrypter = context.mock(StreamDecrypter.class);
	context.checking(new Expectations() {{
		oneOf(decrypter).readFrame(with(any(byte[].class)));
		will(returnValue(0)); // Empty frame
		oneOf(decrypter).readFrame(with(any(byte[].class)));
		will(returnValue(2)); // Non-empty frame with two payload bytes
		oneOf(decrypter).readFrame(with(any(byte[].class)));
		will(returnValue(0)); // Empty frame
		oneOf(decrypter).readFrame(with(any(byte[].class)));
		will(returnValue(-1)); // No more frames
	}});
	StreamReaderImpl r = new StreamReaderImpl(decrypter);
	byte[] buf = new byte[MAX_PAYLOAD_LENGTH];
	// Skip the first empty frame, read the two payload bytes
	assertEquals(2, r.read(buf));
	// Skip the second empty frame, reach EOF
	assertEquals(-1, r.read(buf));
	// Still at EOF
	assertEquals(-1, r.read(buf));
	context.assertIsSatisfied();
	r.close();
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testMultipleReadsPerFrameWithOffsets() throws Exception {
	Mockery context = new Mockery();
	final StreamDecrypter decrypter = context.mock(StreamDecrypter.class);
	context.checking(new Expectations() {{
		oneOf(decrypter).readFrame(with(any(byte[].class)));
		will(returnValue(MAX_PAYLOAD_LENGTH)); // Nice long frame
		oneOf(decrypter).readFrame(with(any(byte[].class)));
		will(returnValue(-1)); // No more frames
	}});
	StreamReaderImpl r = new StreamReaderImpl(decrypter);
	byte[] buf = new byte[MAX_PAYLOAD_LENGTH];
	// Read the first half of the payload
	assertEquals(MAX_PAYLOAD_LENGTH / 2, r.read(buf, MAX_PAYLOAD_LENGTH / 2,
			MAX_PAYLOAD_LENGTH / 2));
	// Read the second half of the payload
	assertEquals(MAX_PAYLOAD_LENGTH / 2, r.read(buf, 123,
			MAX_PAYLOAD_LENGTH / 2));
	// Reach EOF
	assertEquals(-1, r.read(buf, 0, buf.length));
	context.assertIsSatisfied();
	r.close();
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testOutgoingStreamContextIsNullIfContactIsNotFound()
		throws Exception {
	Mockery context = new Mockery();
	final DatabaseComponent db = context.mock(DatabaseComponent.class);
	final CryptoComponent crypto = context.mock(CryptoComponent.class);
	final Executor dbExecutor = context.mock(Executor.class);
	final ScheduledExecutorService scheduler =
			context.mock(ScheduledExecutorService.class);
	final Clock clock = context.mock(Clock.class);

	final Transaction txn = new Transaction(null, false);

	TransportKeyManager
			transportKeyManager = new TransportKeyManagerImpl(db,
			crypto, dbExecutor, scheduler, clock, transportId, maxLatency);
	assertNull(transportKeyManager.getStreamContext(txn, contactId));

	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testCannotAddLocalIdentityAsContact() throws Exception {
	Mockery context = new Mockery();
	@SuppressWarnings("unchecked")
	final Database<Object> database = context.mock(Database.class);
	final ShutdownManager shutdown = context.mock(ShutdownManager.class);
	final EventBus eventBus = context.mock(EventBus.class);

	context.checking(new Expectations() {{
		oneOf(database).startTransaction();
		will(returnValue(txn));
		oneOf(database).containsLocalAuthor(txn, localAuthorId);
		will(returnValue(true));
		// Contact is a local identity
		oneOf(database).containsLocalAuthor(txn, authorId);
		will(returnValue(true));
		oneOf(database).abortTransaction(txn);
	}});

	DatabaseComponent db = createDatabaseComponent(database, eventBus,
			shutdown);

	Transaction transaction = db.startTransaction(false);
	try {
		db.addContact(transaction, author, localAuthorId, true, true);
		fail();
	} catch (ContactExistsException expected) {
		// Expected
	} finally {
		db.endTransaction(transaction);
	}

	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testWriterIsNullIfNoDriveIsChosen() throws Exception {
	final File drive1 = new File(testDir, "1");
	final File drive2 = new File(testDir, "2");
	final List<File> drives = new ArrayList<>();
	drives.add(drive1);
	drives.add(drive2);

	Mockery context = new Mockery() {{
		setThreadingPolicy(new Synchroniser());
	}};
	final Executor executor = context.mock(Executor.class);
	final SimplexPluginCallback callback =
			context.mock(SimplexPluginCallback.class);
	final RemovableDriveFinder finder =
			context.mock(RemovableDriveFinder.class);
	final RemovableDriveMonitor monitor =
			context.mock(RemovableDriveMonitor.class);

	context.checking(new Expectations() {{
		oneOf(monitor).start(with(any(Callback.class)));
		oneOf(finder).findRemovableDrives();
		will(returnValue(drives));
		oneOf(callback).showChoice(with(any(String[].class)),
				with(any(String[].class)));
		will(returnValue(-1)); // The user cancelled the choice
	}});

	RemovableDrivePlugin plugin = new RemovableDrivePlugin(executor,
			callback, finder, monitor, 0);
	plugin.start();

	assertNull(plugin.createWriter(contactId));
	File[] files = drive1.listFiles();
	assertTrue(files == null || files.length == 0);

	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Before
public void setUp() {
  mockContext = new Mockery() {
    {
      setImposteriser(ClassImposteriser.INSTANCE);
    }
  };
}
 

import org.jmock.Mockery; //導入依賴的package包/類
@Test
public void testReceiveDuplicateMessage() throws Exception {
	Mockery context = new Mockery();
	@SuppressWarnings("unchecked")
	final Database<Object> database = context.mock(Database.class);
	final ShutdownManager shutdown = context.mock(ShutdownManager.class);
	final EventBus eventBus = context.mock(EventBus.class);
	context.checking(new Expectations() {{
		oneOf(database).startTransaction();
		will(returnValue(txn));
		oneOf(database).containsContact(txn, contactId);
		will(returnValue(true));
		oneOf(database).containsMessage(txn, messageId);
		will(returnValue(true));
		oneOf(database).getGroupVisibility(txn, contactId, groupId);
		will(returnValue(VISIBLE));
		// The message wasn't stored but it must still be acked
		oneOf(database).raiseSeenFlag(txn, contactId, messageId);
		oneOf(database).raiseAckFlag(txn, contactId, messageId);
		oneOf(database).commitTransaction(txn);
		// The message was received but not added
		oneOf(eventBus).broadcast(with(any(MessageToAckEvent.class)));
	}});
	DatabaseComponent db = createDatabaseComponent(database, eventBus,
			shutdown);

	Transaction transaction = db.startTransaction(false);
	try {
		db.receiveMessage(transaction, contactId, message);
		db.commitTransaction(transaction);
	} finally {
		db.endTransaction(transaction);
	}

	context.assertIsSatisfied();
}