briar/briar-tests/src/org/briarproject/plugins/DuplexClientTest.java

package org.briarproject.plugins;

import org.briarproject.api.contact.ContactId;
import org.briarproject.api.crypto.PseudoRandom;
import org.briarproject.api.plugins.duplex.DuplexPluginCallback;
import org.briarproject.api.plugins.duplex.DuplexTransportConnection;
import org.briarproject.api.properties.TransportProperties;
import org.briarproject.api.settings.Settings;

import java.io.IOException;
import java.util.Map;

import static org.briarproject.api.invitation.InvitationConstants.CONNECTION_TIMEOUT;

public abstract class DuplexClientTest extends DuplexTest {

	protected ClientCallback callback = null;

	protected void run() throws IOException {
		assert plugin != null;
		// Start the plugin
		System.out.println("Starting plugin");
		if (!plugin.start()) {
			System.out.println("Plugin failed to start");
			return;
		}
		try {
			// Try to connect to the server
			System.out.println("Creating connection");
			DuplexTransportConnection d = plugin.createConnection(contactId);
			if (d == null) {
				System.out.println("Connection failed");
				return;
			} else {
				System.out.println("Connection created");
				receiveChallengeSendResponse(d);
			}
			if (!plugin.supportsInvitations()) {
				System.out.println("Skipping invitation test");
				return;
			}
			// Try to create an invitation connection
			System.out.println("Creating invitation connection");
			PseudoRandom r = getPseudoRandom(123);
			d = plugin.createInvitationConnection(r, CONNECTION_TIMEOUT);
			if (d == null) {
				System.out.println("Connection failed");
			} else {
				System.out.println("Connection created");
				sendChallengeReceiveResponse(d);
			}
		} finally {
			// Stop the plugin
			System.out.println("Stopping plugin");
			plugin.stop();
		}
	}

	protected static class ClientCallback implements DuplexPluginCallback {

		private Settings settings = null;
		private TransportProperties local = null;
		private Map<ContactId, TransportProperties> remote = null;

		public ClientCallback(Settings settings, TransportProperties local,
				Map<ContactId, TransportProperties> remote) {
			this.settings = settings;
			this.local = local;
			this.remote = remote;
		}

		public Settings getSettings() {
			return settings;
		}

		public TransportProperties getLocalProperties() {
			return local;
		}

		public Map<ContactId, TransportProperties> getRemoteProperties() {
			return remote;
		}

		public void mergeSettings(Settings s) {
			settings = s;
		}

		public void mergeLocalProperties(TransportProperties p) {
			local = p;
		}

		public int showChoice(String[] options, String... message) {
			return -1;
		}

		public boolean showConfirmationMessage(String... message) {
			return false;
		}

		public void showMessage(String... message) {}

		public void incomingConnectionCreated(DuplexTransportConnection d) {}

		public void outgoingConnectionCreated(ContactId contactId,
				DuplexTransportConnection d) {}

		public void transportEnabled() {}

		public void transportDisabled() {}
	}
}