Java Query类代码示例

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
private void prepareMocks() throws ProtocolException {
	mockProtocolSession = EasyMock.createMock(IProtocolSession.class);
	message = new Capture<Query>();
	mockProtocolSession.sendReceive(EasyMock.capture(message));
	EasyMock.expectLastCall().andAnswer(new IAnswer<Object>() {
		public Object answer() throws Throwable {
			return null;
		}
	});

	mockProtocolSessionManager = EasyMock.createMock(IProtocolSessionManager.class);
	EasyMock.expect(mockProtocolSessionManager.getSessionById(EasyMock.anyObject(String.class), EasyMock.anyBoolean())).andReturn(
			mockProtocolSession);
	EasyMock.expect(mockProtocolSessionManager.obtainSessionByProtocol(EasyMock.anyObject(String.class), EasyMock.anyBoolean())).andReturn(
			mockProtocolSession);
	EasyMock.replay(mockProtocolSession);
	EasyMock.replay(mockProtocolSessionManager);
}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
private void sendCloseSession() throws IOException {
	log.debug("Sending close session.");
	Query query = new Query();
	query.setMessageId("" + messageCounter++);
	query.setOperation(Operation.CLOSE_SESSION);
	sendQuery(query);
}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
@Test
public void testBehaviours() {
	Server server = Server.createServerStoringMessages(2222);

	// define behaviours
	Query bQuery = QueryFactory.newGetRouteInformation();
	Reply bReply = new Reply();
	Error error = ErrorFactory.newError(ErrorType.APPLICATION, ErrorTag.OPERATION_FAILED, ErrorSeverity.ERROR, null, null, null, null);
	bReply.addError(error);
	Behaviour behaviour = new Behaviour(bQuery, bReply);
	server.defineBehaviour(behaviour);

	// start the server
	server.startServer();

	try {
		SessionContext sessionContext = new SessionContext();
		sessionContext.setURI(new URI("ssh://user:[email protected]:2222/"));

		INetconfSession session = new NetconfSession(sessionContext);

		session.connect();

		// send a message
		Query query = QueryFactory.newGetRouteInformation();
		Reply reply = session.sendSyncQuery(query);

		// assertions
		Assert.assertEquals("Reply must contain 1 error", 1, reply.getErrors().size());
		Assert.assertEquals("Reply error must be of 'type application'", ErrorType.APPLICATION, reply.getErrors().get(0).getType());
		Assert.assertEquals("Error tag must be 'operation-failed'", ErrorTag.OPERATION_FAILED, reply.getErrors().get(0).getTag());
		Assert.assertEquals("Error severity must be 'error'", ErrorSeverity.ERROR, reply.getErrors().get(0).getSeverity());
	} catch (Exception e) {
		Assert.fail("Error creating session: " + e.getMessage());
	} finally {
		server.stopServer();
	}
}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
@Override
public void asyncSend(Object requestMessage) throws ProtocolException {
	// Send a message to the device, and don't wait for the response
	RPCElement rpcReply = null;
	try {
		Query rpcQuery = (Query) requestMessage;
		netconfSession.sendAsyncQuery(rpcQuery);
	} catch (Exception ex) {
		throw new ProtocolException(ex);
	}

}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
@Override
public Object sendReceive(Object requestMessage) throws ProtocolException {
	// Send a message to the device, and wait for the response
	RPCElement rpcReply = null;
	try {
		Query rpcQuery = (Query) requestMessage;
		rpcReply = netconfSession.sendSyncQuery(rpcQuery);
	} catch (Exception ex) {
		throw new ProtocolException(ex);
	}
	return rpcReply;
}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
@Test
public void testPrepareRestoreAction() throws Exception {
	String uri = System.getProperty("protocol.uri");

	if (uri == null || uri.equals("${protocol.uri}")) {
		uri = "mock://user:[email protected]:2212/mocksubsystem";
	}
	String compare = null;
	String toCompare = null;
	Query query = QueryFactory.newGetConfig("running", "<configuration></configuration>", null);
	Reply reply = sendNetconfMessage(query, uri);
	compare = reply.getContain();

	IAction action = new MockAction();
	action.setActionID("mockAction");
	queueManagerCapability.queueAction(action);
	action = new CorruptedAction();
	action.setActionID("corruptedAction");
	queueManagerCapability.queueAction(action);

	boolean isChecked = false;
	try {
		queueManagerCapability.execute();
	} catch (Exception e) {
		if (e instanceof CapabilityException)
			isChecked = true;
	}
	if (!uri.startsWith("mock:"))
		Assert.assertTrue(isChecked);

	reply = sendNetconfMessage(query, uri);
	toCompare = reply.getContain();

	Assert.assertEquals(compare, toCompare);
}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
public Reply sendNetconfMessage(Query query, String uri) throws Exception {
	SessionContext sessionContext = new SessionContext();
	sessionContext.setURI(new URI(uri));
	NetconfSession session = new NetconfSession(sessionContext);
	session.connect();
	Reply reply = session.sendSyncQuery(query);
	if (reply.containsErrors())
		throw new Exception();
	session.disconnect();
	return reply;

}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
@Override
public ActionResponse execute(IProtocolSessionManager protocolSessionManager) throws ActionException {
	Response response = null;
	try {

		/*
		 * IT INCLUDES AN ERROR IN THE MASK
		 */
		String netconfXML = "<configuration><interfaces>" +
				"<interface>" +
				"<name>fe-0/1/2</name>" +
				"<unit operation=\"replace\">" +
				"<name>2<name>" +
				"<family><inet><address>192.168.1.3/60</address></inet></family>" +
				"</unit>" +
				"</interface></interfaces></configuration>";

		NetconfProtocolSession protocol = (NetconfProtocolSession) protocolSessionManager.obtainSessionByProtocol("netconf", false);
		Query query = QueryFactory.newEditConfig("candidate", null, null, null, netconfXML);
		Reply reply = (Reply) protocol.sendReceive(query);

		// extra control, it checks if is not null the error list
		response = checkResponse(reply, query);
		validateResponse(response);

	} catch (ProtocolException e) {
		throw new ActionException(e);
	}

	ActionResponse actionResponse = new ActionResponse();
	actionResponse.setActionID(actionID);
	actionResponse.addResponse(response);
	return actionResponse;
}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
public void sendFakeConfig(Query configQuery) throws IOException {
	InputStream configFileIs = this.getClass().getResourceAsStream("/router_configs/router_config_A.xml");
	Reply reply = ReplyFactory.newGetConfigReply(configQuery, null, IOUtils.toString(configFileIs));
	sendReply(reply);
}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
private void sendOk(Query query) throws IOException {
	log.debug("Sending OK.");
	sendReply(ReplyFactory.newOk(query, null));
}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
public void sendQuery(Query query) throws IOException {
	send(query.toXML());
}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
public Query getQuery() {
	return query;
}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
@Test
public void testConsumingBehaviours() {
	Server server = Server.createServerStoringMessages(2222);

	// define behaviours
	Query bQuery = QueryFactory.newGetRouteInformation();
	Reply bReply = new Reply();
	Error error = ErrorFactory.newError(ErrorType.APPLICATION, ErrorTag.OPERATION_FAILED, ErrorSeverity.ERROR, null, null, null, null);
	bReply.addError(error);
	Behaviour behaviour = new Behaviour(bQuery, bReply, true);
	server.defineBehaviour(behaviour);

	bQuery = QueryFactory.newDiscardChanges();
	bReply = ReplyFactory.newOk(bQuery, null);
	behaviour = new Behaviour(bQuery, bReply, true);
	server.defineBehaviour(behaviour);

	bQuery = QueryFactory.newGetRouteInformation();
	bReply = ReplyFactory.newOk(bQuery, null);
	behaviour = new Behaviour(bQuery, bReply, true);
	server.defineBehaviour(behaviour);

	// start the server
	server.startServer();

	try {
		SessionContext sessionContext = new SessionContext();
		sessionContext.setURI(new URI("ssh://user:[email protected]:2222/"));

		INetconfSession session = new NetconfSession(sessionContext);

		session.connect();

		// send first query
		Query query = QueryFactory.newGetRouteInformation();
		Reply reply = session.sendSyncQuery(query);

		// assertions
		Assert.assertEquals("Reply must contain 1 error", 1, reply.getErrors().size());
		Assert.assertEquals("Reply error must be of 'type application'", ErrorType.APPLICATION, reply.getErrors().get(0).getType());
		Assert.assertEquals("Error tag must be 'operation-failed'", ErrorTag.OPERATION_FAILED, reply.getErrors().get(0).getTag());
		Assert.assertEquals("Error severity must be 'error'", ErrorSeverity.ERROR, reply.getErrors().get(0).getSeverity());

		// send second query
		query = QueryFactory.newDiscardChanges();
		reply = session.sendSyncQuery(query);

		// assertions
		Assert.assertEquals("Reply must contain 0 errors", 0, reply.getErrors().size());
		Assert.assertEquals("Reply must be OK", true, reply.isOk());

		// send third query
		query = QueryFactory.newGetRouteInformation();
		reply = session.sendSyncQuery(query);

		// assertions
		Assert.assertEquals("Reply must contain 0 errors", 0, reply.getErrors().size());
		Assert.assertEquals("Reply must be OK", true, reply.isOk());

	} catch (Exception e) {
		Assert.fail("Error executing tests: " + e.getMessage());
	} finally {
		server.stopServer();
	}
}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
@Override
public Query message() {
	// setQuery(QueryFactory.newGetInterfaceInformation());
	setQuery(QueryFactory.newGetConfig(source.toString(), netconfXML, attrFilter));
	return query;
}
 

import net.i2cat.netconf.rpc.Query; //导入依赖的package包/类
@Override
public Query message() {
	setQuery(QueryFactory.newLock(target));
	return query;
}