/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.avro.ipc;
import static org.junit.Assert.assertEquals;
import java.util.concurrent.atomic.AtomicInteger;
import org.apache.avro.AvroRemoteException;
import org.apache.avro.ipc.LocalTransceiver;
import org.apache.avro.ipc.RPCContext;
import org.apache.avro.ipc.RPCPlugin;
import org.apache.avro.ipc.specific.SpecificRequestor;
import org.apache.avro.ipc.specific.SpecificResponder;
import org.apache.avro.test.Mail;
import org.apache.avro.test.Message;
import org.junit.Test;
public class TestRpcPluginOrdering {
private static AtomicInteger orderCounter = new AtomicInteger();
public class OrderPlugin extends RPCPlugin{
public void clientStartConnect(RPCContext context) {
assertEquals(0, orderCounter.getAndIncrement());
}
public void clientSendRequest(RPCContext context) {
assertEquals(1, orderCounter.getAndIncrement());
}
public void clientReceiveResponse(RPCContext context) {
assertEquals(6, orderCounter.getAndIncrement());
}
public void clientFinishConnect(RPCContext context) {
assertEquals(5, orderCounter.getAndIncrement());
}
public void serverConnecting(RPCContext context) {
assertEquals(2, orderCounter.getAndIncrement());
}
public void serverReceiveRequest(RPCContext context) {
assertEquals(3, orderCounter.getAndIncrement());
}
public void serverSendResponse(RPCContext context) {
assertEquals(4, orderCounter.getAndIncrement());
}
}
@Test
public void testRpcPluginOrdering() throws Exception {
OrderPlugin plugin = new OrderPlugin();
SpecificResponder responder = new SpecificResponder(Mail.class, new TestMailImpl());
SpecificRequestor requestor = new SpecificRequestor(Mail.class, new LocalTransceiver(responder));
responder.addRPCPlugin(plugin);
requestor.addRPCPlugin(plugin);
Mail client = SpecificRequestor.getClient(Mail.class, requestor);
Message message = createTestMessage();
client.send(message);
}
private Message createTestMessage() {
Message message = Message.newBuilder().
setTo("me@test.com").
setFrom("you@test.com").
setBody("plugin testing").
build();
return message;
}
private static class TestMailImpl implements Mail{
public String send(Message message) throws AvroRemoteException {
return "Received";
}
public void fireandforget(Message message) {
}
}
}