OFMessageDamperMockSwitch.java

/*
 * Copyright (c) 2013 Big Switch Networks, Inc.
 *
 * Licensed under the Eclipse Public License, Version 1.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.eclipse.org/legal/epl-v10.html
 *
 * 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.sdnplatform.util;

import static org.junit.Assert.*;
import java.io.IOException;
import java.net.SocketAddress;

import java.util.Collection;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Future;
import java.util.concurrent.locks.Lock;


import org.jboss.netty.channel.Channel;
import org.openflow.protocol.OFFeaturesReply;
import org.openflow.protocol.OFMessage;
import org.openflow.protocol.OFPhysicalPort;
import org.openflow.protocol.OFStatisticsRequest;
import org.openflow.protocol.statistics.OFDescriptionStatistics;
import org.openflow.protocol.statistics.OFStatistics;
import org.sdnplatform.core.ListenerContext;
import org.sdnplatform.core.IOFMessageListener;
import org.sdnplatform.core.IOFSwitch;
import org.sdnplatform.core.IControllerService.Role;
import org.sdnplatform.core.internal.Controller;
import org.sdnplatform.threadpool.IThreadPoolService;
import org.sdnplatform.util.OFMessageDamper;


/**
 * A mock implementation of IFOSwitch we use for {@link OFMessageDamper}
 * 
 * We need to mock equals() and hashCode() but alas, EasyMock doesn't support
 * this. Sigh. And of course this happens to be the interface with the most
 * methods. 
 * @author gregor
 *
 */
public class OFMessageDamperMockSwitch implements IOFSwitch {
    OFMessage writtenMessage;
    ListenerContext writtenContext;
    
    public OFMessageDamperMockSwitch() {
        reset();
    }
    
    /* reset this mock. I.e., clear the stored message previously written */
    public void reset() {
        writtenMessage = null;
        writtenContext = null;
    }
    
    /* assert that a message was written to this switch and that the 
     * written message and context matches the expected values 
     * @param expected
     * @param expectedContext
     */
    public void assertMessageWasWritten(OFMessage expected, 
                                        ListenerContext expectedContext) {
        assertNotNull("No OFMessage was written", writtenMessage);
        assertEquals(expected, writtenMessage);
        assertEquals(expectedContext, writtenContext);
    }
    
    /*
     * assert that no message was written 
     */
    public void assertNoMessageWritten() {
        assertNull("OFMessage was written but didn't expect one", 
                      writtenMessage);
        assertNull("There was a context but didn't expect one", 
                      writtenContext);
    }
    
    /*
     * use hashCode() and equals() from Object
     */
    
    
    //-------------------------------------------------------
    // IOFSwitch: mocked methods
    @Override
    public void write(OFMessage m, ListenerContext bc) throws IOException {
        assertNull("write() called but already have message", writtenMessage);
        assertNull("write() called but already have context", writtenContext);
        writtenContext = bc;
        writtenMessage = m;
    }
    
    //-------------------------------------------------------
    // IOFSwitch: not-implemented methods
    @Override
    public void write(List<OFMessage> msglist, ListenerContext bc) 
            throws IOException {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public void disconnectOutputStream() {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public void setFeaturesReply(OFFeaturesReply featuresReply) {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public void setSwitchProperties(OFDescriptionStatistics description) {
        assertTrue("Unexpected method call", false);
        // TODO Auto-generated method stub
    }
    
    @Override
    public Collection<OFPhysicalPort> getEnabledPorts() {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public Collection<Short> getEnabledPortNumbers() {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public OFPhysicalPort getPort(short portNumber) {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public OFPhysicalPort getPort(String portName) {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public void setPort(OFPhysicalPort port) {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public void deletePort(short portNumber) {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public void deletePort(String portName) {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public Collection<OFPhysicalPort> getPorts() {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public boolean portEnabled(short portName) {
        assertTrue("Unexpected method call", false);
        return false;
    }
    
    @Override
    public boolean portEnabled(String portName) {
        assertTrue("Unexpected method call", false);
        return false;
    }
    
    @Override
    public boolean portEnabled(OFPhysicalPort port) {
        assertTrue("Unexpected method call", false);
        return false;
    }
    
    @Override
    public long getId() {
        assertTrue("Unexpected method call", false);
        return 0;
    }
    
    @Override
    public String getStringId() {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public Map<Object, Object> getAttributes() {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public Date getConnectedSince() {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public int getNextTransactionId() {
        assertTrue("Unexpected method call", false);
        return 0;
    }
    
    @Override
    public Future<List<OFStatistics>>
            getStatistics(OFStatisticsRequest request) throws IOException {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public boolean isConnected() {
        assertTrue("Unexpected method call", false);
        return false;
    }
    
    @Override
    public SocketAddress getInetAddress() {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public void setConnected(boolean connected) {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public Role getHARole() {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public void deliverStatisticsReply(OFMessage reply) {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public void cancelStatisticsReply(int transactionId) {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public void cancelAllStatisticsReplies() {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public boolean hasAttribute(String name) {
        assertTrue("Unexpected method call", false);
        return false;
    }
    
    @Override
    public Object getAttribute(String name) {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public void setAttribute(String name, Object value) {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public Object removeAttribute(String name) {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public void clearAllFlowMods() {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public boolean updateBroadcastCache(Long entry, Short port) {
        assertTrue("Unexpected method call", false);
        return false;
    }
    
    @Override
    public Map<Short, Long> getPortBroadcastHits() {
        assertTrue("Unexpected method call", false);
        return null;
    }
    
    @Override
    public void sendStatsQuery(OFStatisticsRequest request, int xid,
                               IOFMessageListener caller)
                                                         throws IOException {
        assertTrue("Unexpected method call", false);
    }
    
    @Override
    public void flush() {
        assertTrue("Unexpected method call", false);
    }

    @Override
    public Future<OFFeaturesReply> querySwitchFeaturesReply()
            throws IOException {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public void deliverOFFeaturesReply(OFMessage reply) {
        // TODO Auto-generated method stub

    }

    @Override
    public void cancelFeaturesReply(int transactionId) {
        // TODO Auto-generated method stub

    }

    @Override
    public int getBuffers() {
        // TODO Auto-generated method stub
        return 0;
    }

    @Override
    public int getActions() {
        // TODO Auto-generated method stub
        return 0;
    }

    @Override
    public int getCapabilities() {
        // TODO Auto-generated method stub
        return 0;
    }

    @Override
    public byte getTables() {
        // TODO Auto-generated method stub
        return 0;
    }

    @Override
    public void setChannel(Channel channel) {
        // TODO Auto-generated method stub
        
    }

    @Override
    public void setControllerProvider(Controller controller) {
        // TODO Auto-generated method stub
        
    }

    @Override
    public void setThreadPoolService(IThreadPoolService threadPool) {
        // TODO Auto-generated method stub
        
    }

    @Override
    public Lock getListenerReadLock() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Lock getListenerWriteLock() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public void setHARole(Role role, boolean haRoleReplyReceived) {
        // TODO Auto-generated method stub
        
    }

    @Override
    public OFPortType getPortType(short port_num) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public boolean isFastPort(short port_num) {
        // TODO Auto-generated method stub
        return false;
    }

    @Override
    public List<Short> getUplinkPorts() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public boolean attributeEquals(String name, Object other) {
        fail("Unexpected method call");
        return false;
    }

}