MessageConnectionMock.java

/*
 * JBoss, Home of Professional Open Source
 *
 * Copyright 2013 Red Hat, Inc. and/or its affiliates.
 *
 * Licensed 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.xnio.mock;

import java.io.IOException;
import java.net.SocketAddress;

import org.jmock.Expectations;
import org.jmock.Mockery;
import org.jmock.integration.junit4.JUnit4Mockery;
import org.xnio.MessageConnection;
import org.xnio.Option;
import org.xnio.OptionMap;
import org.xnio.XnioIoThread;
import org.xnio.conduits.MessageSinkConduit;
import org.xnio.conduits.MessageSourceConduit;
import org.xnio.conduits.ReadReadyHandler;
import org.xnio.conduits.WriteReadyHandler;

/**
 * {@link MessageConnection} mock.
 * 
 * @author <a href="mailto:frainone@redhat.com">Flavia Rainone</a>
 *
 */
public class MessageConnectionMock extends MessageConnection implements Mock {

    // the peer address
    private final SocketAddress peerAddress;
    // the local address
    private final SocketAddress localAddress;
    // the option map
    private final OptionMap optionMap;
    // info used to check correct delegation from API to mocks
    private String info;

    protected MessageConnectionMock(final XnioIoThread thread, SocketAddress localAddress, SocketAddress peerAddress, OptionMap optionMap) {
        super(thread);
        this.localAddress = localAddress;
        this.peerAddress = peerAddress;
        this.optionMap = optionMap;
        final Mockery context = new JUnit4Mockery();
        final MessageSourceConduit sourceConduit = context.mock(MessageSourceConduit.class, "source conduit");
        final MessageSinkConduit sinkConduit = context.mock(MessageSinkConduit.class, "sink conduit");
        context.checking(new Expectations() {{
            allowing(sourceConduit).getReadThread();
            will(returnValue(thread));
            allowing(sourceConduit).setReadReadyHandler(with(any(ReadReadyHandler.class)));
            allowing(sourceConduit).getWorker();
            will(returnValue(thread.getWorker()));

            allowing(sinkConduit).getWriteThread();
            will(returnValue(thread));
            allowing(sinkConduit).setWriteReadyHandler(with(any(WriteReadyHandler.class)));
            allowing(sinkConduit).getWorker();
            will(returnValue(thread.getWorker()));
        }});
        setSourceConduit(sourceConduit);
        setSinkConduit(sinkConduit);
    }

    @Override
    public SocketAddress getPeerAddress() {
        return peerAddress;
    }

    @Override
    public SocketAddress getLocalAddress() {
        return localAddress;
    }

    @Override
    protected void notifyWriteClosed() {
        throw new UnsupportedOperationException("operation not implemented by mock");
    }

    @Override
    protected void notifyReadClosed() {
        throw new UnsupportedOperationException("operation not implemented by mock");
    }

    @Override
    public void setInfo(String info) {
        this.info = info;
    }

    @Override
    public String getInfo() {
        return info;
    }

    @Override
    public OptionMap getOptionMap() {
        return optionMap;
    }

    @Override
    public <T> T getOption(Option<T> option) throws IOException {
        return optionMap.get(option);
    }
}