Java Mockery.setImposteriser方法代碼示例

import org.jmock.Mockery; //導入方法依賴的package包/類
public IntroducerManagerTest() {
	context = new Mockery();
	context.setImposteriser(ClassImposteriser.INSTANCE);
	messageSender = context.mock(MessageSender.class);
	cryptoComponent = context.mock(CryptoComponent.class);
	clientHelper = context.mock(ClientHelper.class);
	clock = context.mock(Clock.class);
	introductionGroupFactory =
			context.mock(IntroductionGroupFactory.class);

	introducerManager =
			new IntroducerManager(messageSender, clientHelper, clock,
					cryptoComponent, introductionGroupFactory);

	AuthorId authorId1 = new AuthorId(TestUtils.getRandomId());
	Author author1 = new Author(authorId1, "Introducee1",
			TestUtils.getRandomBytes(MAX_PUBLIC_KEY_LENGTH));
	AuthorId localAuthorId1 = new AuthorId(TestUtils.getRandomId());
	ContactId contactId1 = new ContactId(234);
	introducee1 =
			new Contact(contactId1, author1, localAuthorId1, true, true);

	AuthorId authorId2 = new AuthorId(TestUtils.getRandomId());
	Author author2 = new Author(authorId2, "Introducee2",
			TestUtils.getRandomBytes(MAX_PUBLIC_KEY_LENGTH));
	AuthorId localAuthorId2 = new AuthorId(TestUtils.getRandomId());
	ContactId contactId2 = new ContactId(235);
	introducee2 =
			new Contact(contactId2, author2, localAuthorId2, true, true);

	localGroup0 = new Group(new GroupId(TestUtils.getRandomId()),
			getClientId(), new byte[0]);
	introductionGroup1 = new Group(new GroupId(TestUtils.getRandomId()),
			getClientId(), new byte[0]);
	introductionGroup2 = new Group(new GroupId(TestUtils.getRandomId()),
			getClientId(), new byte[0]);

	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入方法依賴的package包/類
@Test
public void testRescheduleAndReconnectOnConnectionClosed()
		throws Exception {
	Mockery context = new Mockery();
	context.setImposteriser(ClassImposteriser.INSTANCE);
	final Executor ioExecutor = new ImmediateExecutor();
	final ScheduledExecutorService scheduler =
			context.mock(ScheduledExecutorService.class);
	final ConnectionManager connectionManager =
			context.mock(ConnectionManager.class);
	final ConnectionRegistry connectionRegistry =
			context.mock(ConnectionRegistry.class);
	final PluginManager pluginManager = context.mock(PluginManager.class);
	final SecureRandom random = context.mock(SecureRandom.class);
	final Clock clock = context.mock(Clock.class);

	final DuplexPlugin plugin = context.mock(DuplexPlugin.class);
	final TransportId transportId = new TransportId("id");
	final DuplexTransportConnection duplexConnection =
			context.mock(DuplexTransportConnection.class);

	context.checking(new Expectations() {{
		allowing(plugin).getId();
		will(returnValue(transportId));
		// reschedule()
		// Get the plugin
		oneOf(pluginManager).getPlugin(transportId);
		will(returnValue(plugin));
		// The plugin supports polling
		oneOf(plugin).shouldPoll();
		will(returnValue(true));
		// Get the plugin
		oneOf(pluginManager).getPlugin(transportId);
		will(returnValue(plugin));
		// The plugin supports polling
		oneOf(plugin).shouldPoll();
		will(returnValue(true));
		// Schedule the next poll
		oneOf(plugin).getPollingInterval();
		will(returnValue(pollingInterval));
		oneOf(clock).currentTimeMillis();
		will(returnValue(now));
		oneOf(scheduler).schedule(with(any(Runnable.class)),
				with((long) pollingInterval), with(MILLISECONDS));
		// connectToContact()
		// Check whether the contact is already connected
		oneOf(connectionRegistry).isConnected(contactId, transportId);
		will(returnValue(false));
		// Connect to the contact
		oneOf(plugin).createConnection(contactId);
		will(returnValue(duplexConnection));
		// Pass the connection to the connection manager
		oneOf(connectionManager).manageOutgoingConnection(contactId,
				transportId, duplexConnection);
	}});

	Poller p = new Poller(ioExecutor, scheduler, connectionManager,
			connectionRegistry, pluginManager, random, clock);

	p.eventOccurred(new ConnectionClosedEvent(contactId, transportId,
			false));

	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入方法依賴的package包/類
@Test
public void testRescheduleOnConnectionOpened() throws Exception {
	Mockery context = new Mockery();
	context.setImposteriser(ClassImposteriser.INSTANCE);
	final Executor ioExecutor = new ImmediateExecutor();
	final ScheduledExecutorService scheduler =
			context.mock(ScheduledExecutorService.class);
	final ConnectionManager connectionManager =
			context.mock(ConnectionManager.class);
	final ConnectionRegistry connectionRegistry =
			context.mock(ConnectionRegistry.class);
	final PluginManager pluginManager = context.mock(PluginManager.class);
	final SecureRandom random = context.mock(SecureRandom.class);
	final Clock clock = context.mock(Clock.class);

	final DuplexPlugin plugin = context.mock(DuplexPlugin.class);
	final TransportId transportId = new TransportId("id");

	context.checking(new Expectations() {{
		allowing(plugin).getId();
		will(returnValue(transportId));
		// Get the plugin
		oneOf(pluginManager).getPlugin(transportId);
		will(returnValue(plugin));
		// The plugin supports polling
		oneOf(plugin).shouldPoll();
		will(returnValue(true));
		// Schedule the next poll
		oneOf(plugin).getPollingInterval();
		will(returnValue(pollingInterval));
		oneOf(clock).currentTimeMillis();
		will(returnValue(now));
		oneOf(scheduler).schedule(with(any(Runnable.class)),
				with((long) pollingInterval), with(MILLISECONDS));
	}});

	Poller p = new Poller(ioExecutor, scheduler, connectionManager,
			connectionRegistry, pluginManager, random, clock);

	p.eventOccurred(new ConnectionOpenedEvent(contactId, transportId,
			false));

	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入方法依賴的package包/類
@Test
public void testRescheduleDoesNotReplaceEarlierTask() throws Exception {
	Mockery context = new Mockery();
	context.setImposteriser(ClassImposteriser.INSTANCE);
	final Executor ioExecutor = new ImmediateExecutor();
	final ScheduledExecutorService scheduler =
			context.mock(ScheduledExecutorService.class);
	final ConnectionManager connectionManager =
			context.mock(ConnectionManager.class);
	final ConnectionRegistry connectionRegistry =
			context.mock(ConnectionRegistry.class);
	final PluginManager pluginManager = context.mock(PluginManager.class);
	final SecureRandom random = context.mock(SecureRandom.class);
	final Clock clock = context.mock(Clock.class);

	final DuplexPlugin plugin = context.mock(DuplexPlugin.class);
	final TransportId transportId = new TransportId("id");

	context.checking(new Expectations() {{
		allowing(plugin).getId();
		will(returnValue(transportId));
		// First event
		// Get the plugin
		oneOf(pluginManager).getPlugin(transportId);
		will(returnValue(plugin));
		// The plugin supports polling
		oneOf(plugin).shouldPoll();
		will(returnValue(true));
		// Schedule the next poll
		oneOf(plugin).getPollingInterval();
		will(returnValue(pollingInterval));
		oneOf(clock).currentTimeMillis();
		will(returnValue(now));
		oneOf(scheduler).schedule(with(any(Runnable.class)),
				with((long) pollingInterval), with(MILLISECONDS));
		// Second event
		// Get the plugin
		oneOf(pluginManager).getPlugin(transportId);
		will(returnValue(plugin));
		// The plugin supports polling
		oneOf(plugin).shouldPoll();
		will(returnValue(true));
		// Don't replace the previously scheduled task, due earlier
		oneOf(plugin).getPollingInterval();
		will(returnValue(pollingInterval));
		oneOf(clock).currentTimeMillis();
		will(returnValue(now + 1));
	}});

	Poller p = new Poller(ioExecutor, scheduler, connectionManager,
			connectionRegistry, pluginManager, random, clock);

	p.eventOccurred(new ConnectionOpenedEvent(contactId, transportId,
			false));
	p.eventOccurred(new ConnectionOpenedEvent(contactId, transportId,
			false));

	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入方法依賴的package包/類
@Test
public void testPollOnTransportEnabled() throws Exception {
	Mockery context = new Mockery();
	context.setImposteriser(ClassImposteriser.INSTANCE);
	final Executor ioExecutor = new ImmediateExecutor();
	final ScheduledExecutorService scheduler =
			context.mock(ScheduledExecutorService.class);
	final ConnectionManager connectionManager =
			context.mock(ConnectionManager.class);
	final ConnectionRegistry connectionRegistry =
			context.mock(ConnectionRegistry.class);
	final PluginManager pluginManager = context.mock(PluginManager.class);
	final SecureRandom random = context.mock(SecureRandom.class);
	final Clock clock = context.mock(Clock.class);

	final Plugin plugin = context.mock(Plugin.class);
	final TransportId transportId = new TransportId("id");
	final List<ContactId> connected = Collections.singletonList(contactId);

	context.checking(new Expectations() {{
		allowing(plugin).getId();
		will(returnValue(transportId));
		// Get the plugin
		oneOf(pluginManager).getPlugin(transportId);
		will(returnValue(plugin));
		// The plugin supports polling
		oneOf(plugin).shouldPoll();
		will(returnValue(true));
		// Schedule a polling task immediately
		oneOf(clock).currentTimeMillis();
		will(returnValue(now));
		oneOf(scheduler).schedule(with(any(Runnable.class)), with(0L),
				with(MILLISECONDS));
		will(new RunAction());
		// Running the polling task schedules the next polling task
		oneOf(plugin).getPollingInterval();
		will(returnValue(pollingInterval));
		oneOf(random).nextDouble();
		will(returnValue(0.5));
		oneOf(clock).currentTimeMillis();
		will(returnValue(now));
		oneOf(scheduler).schedule(with(any(Runnable.class)),
				with((long) (pollingInterval * 0.5)), with(MILLISECONDS));
		// Poll the plugin
		oneOf(connectionRegistry).getConnectedContacts(transportId);
		will(returnValue(connected));
		oneOf(plugin).poll(connected);
	}});

	Poller p = new Poller(ioExecutor, scheduler, connectionManager,
			connectionRegistry, pluginManager, random, clock);

	p.eventOccurred(new TransportEnabledEvent(transportId));

	context.assertIsSatisfied();
}
 

import org.jmock.Mockery; //導入方法依賴的package包/類
public IntroduceeManagerTest() {
	context = new Mockery();
	context.setImposteriser(ClassImposteriser.INSTANCE);
	MessageSender messageSender = context.mock(MessageSender.class);
	db = context.mock(DatabaseComponent.class);
	cryptoComponent = context.mock(CryptoComponent.class);
	clientHelper = context.mock(ClientHelper.class);
	clock = context.mock(Clock.class);
	introductionGroupFactory =
			context.mock(IntroductionGroupFactory.class);
	TransportPropertyManager transportPropertyManager =
			context.mock(TransportPropertyManager.class);
	authorFactory = context.mock(AuthorFactory.class);
	contactManager = context.mock(ContactManager.class);
	IdentityManager identityManager = context.mock(IdentityManager.class);

	introduceeManager = new IntroduceeManager(messageSender, db,
			clientHelper, clock, cryptoComponent, transportPropertyManager,
			authorFactory, contactManager, identityManager,
			introductionGroupFactory);

	AuthorId authorId0 = new AuthorId(TestUtils.getRandomId());
	Author author0 = new Author(authorId0, "Introducer",
			TestUtils.getRandomBytes(MAX_PUBLIC_KEY_LENGTH));
	AuthorId localAuthorId = new AuthorId(TestUtils.getRandomId());
	ContactId contactId0 = new ContactId(234);
	introducer =
			new Contact(contactId0, author0, localAuthorId, true, true);

	AuthorId authorId1 = new AuthorId(TestUtils.getRandomId());
	Author author1 = new Author(authorId1, "Introducee1",
			TestUtils.getRandomBytes(MAX_PUBLIC_KEY_LENGTH));
	AuthorId localAuthorId1 = new AuthorId(TestUtils.getRandomId());
	ContactId contactId1 = new ContactId(234);
	introducee1 =
			new Contact(contactId1, author1, localAuthorId1, true, true);

	AuthorId authorId2 = new AuthorId(TestUtils.getRandomId());
	Author author2 = new Author(authorId2, "Introducee2",
			TestUtils.getRandomBytes(MAX_PUBLIC_KEY_LENGTH));
	ContactId contactId2 = new ContactId(235);
	introducee2 =
			new Contact(contactId2, author2, localAuthorId, true, true);

	ClientId clientId = IntroductionManagerImpl.CLIENT_ID;
	localGroup1 = new Group(new GroupId(TestUtils.getRandomId()),
			clientId, new byte[0]);
	introductionGroup1 = new Group(new GroupId(TestUtils.getRandomId()),
			clientId, new byte[0]);

	sessionId = new SessionId(TestUtils.getRandomId());
	localStateMessage = new Message(
			new MessageId(TestUtils.getRandomId()),
			localGroup1.getId(),
			time,
			TestUtils.getRandomBytes(MESSAGE_HEADER_LENGTH + 1)
	);
	message1 = new Message(
			new MessageId(TestUtils.getRandomId()),
			introductionGroup1.getId(),
			time,
			TestUtils.getRandomBytes(MESSAGE_HEADER_LENGTH + 1)
	);

	txn = new Transaction(null, false);
}
 

import org.jmock.Mockery; //導入方法依賴的package包/類
public IntroductionManagerImplTest() {
	AuthorId authorId1 = new AuthorId(TestUtils.getRandomId());
	Author author1 = new Author(authorId1, "Introducee1",
			new byte[MAX_PUBLIC_KEY_LENGTH]);
	AuthorId localAuthorId1 = new AuthorId(TestUtils.getRandomId());
	ContactId contactId1 = new ContactId(234);
	introducee1 =
			new Contact(contactId1, author1, localAuthorId1, true, true);

	AuthorId authorId2 = new AuthorId(TestUtils.getRandomId());
	Author author2 = new Author(authorId2, "Introducee2",
			new byte[MAX_PUBLIC_KEY_LENGTH]);
	AuthorId localAuthorId2 = new AuthorId(TestUtils.getRandomId());
	ContactId contactId2 = new ContactId(235);
	introducee2 =
			new Contact(contactId2, author2, localAuthorId2, true, true);

	ClientId clientId = new ClientId(TestUtils.getRandomString(5));
	introductionGroup1 = new Group(new GroupId(TestUtils.getRandomId()),
			clientId, new byte[0]);
	introductionGroup2 = new Group(new GroupId(TestUtils.getRandomId()),
			clientId, new byte[0]);

	message1 = new Message(
			new MessageId(TestUtils.getRandomId()),
			introductionGroup1.getId(),
			time,
			TestUtils.getRandomBytes(MESSAGE_HEADER_LENGTH + 1)
	);

	// mock ALL THE THINGS!!!
	context = new Mockery();
	context.setImposteriser(ClassImposteriser.INSTANCE);
	introducerManager = context.mock(IntroducerManager.class);
	introduceeManager = context.mock(IntroduceeManager.class);
	db = context.mock(DatabaseComponent.class);
	clientHelper = context.mock(ClientHelper.class);
	MetadataParser metadataParser = context.mock(MetadataParser.class);
	messageTracker = context.mock(MessageTracker.class);
	introductionGroupFactory = context.mock(IntroductionGroupFactory.class);

	introductionManager = new IntroductionManagerImpl(db, clientHelper,
			metadataParser, messageTracker, introducerManager,
			introduceeManager, introductionGroupFactory);
}
 

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

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