Bugfix: count outgoing control messages if CPMan is activated

Previous implementation omits all outgoing messages when
the role is configured as Master.
With the revised version, the controller would more precisely
count the number of outgoing control messages.

Change-Id: I90c06097cb297535bce37e4a0ed5734cf01871bf

 /*
 /*
- * Copyright 2014-2015 Open Networking Laboratory
+ * Copyright 2014-2016 Open Networking Laboratory
  *
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * you may not use this file except in compliance with the License.
@@ -164,7 +164,9 @@ public abstract class AbstractOpenFlowSwitch extends AbstractHandlerBehaviour
                          dpid, role, channel.isConnected(), msgs);
                          dpid, role, channel.isConnected(), msgs);
             }
             }
         }
         }
+    }
 
 
+    private void countOutgoingMsg(List<OFMessage> msgs) {
         // listen to outgoing control messages only if listeners are registered
         // listen to outgoing control messages only if listeners are registered
         if (ofOutgoingMsgListener.size() != 0) {
         if (ofOutgoingMsgListener.size() != 0) {
             msgs.forEach(m -> {
             msgs.forEach(m -> {
@@ -180,6 +182,7 @@ public abstract class AbstractOpenFlowSwitch extends AbstractHandlerBehaviour
     private void sendMsgsOnChannel(List<OFMessage> msgs) {
     private void sendMsgsOnChannel(List<OFMessage> msgs) {
         if (channel.isConnected()) {
         if (channel.isConnected()) {
             channel.write(msgs);
             channel.write(msgs);
+            countOutgoingMsg(msgs);
         } else {
         } else {
             log.warn("Dropping messages for switch {} because channel is not connected: {}",
             log.warn("Dropping messages for switch {} because channel is not connected: {}",
                      dpid, msgs);
                      dpid, msgs);

@@ -16,13 +16,18 @@
 package org.onosproject.openflow.controller.driver;
 package org.onosproject.openflow.controller.driver;
 
 
 import org.jboss.netty.channel.Channel;
 import org.jboss.netty.channel.Channel;
+import org.jboss.netty.channel.ChannelConfig;
+import org.jboss.netty.channel.ChannelFactory;
+import org.jboss.netty.channel.ChannelFuture;
+import org.jboss.netty.channel.ChannelPipeline;
 import org.junit.Before;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.Test;
-import org.onosproject.core.netty.ChannelAdapter;
 import org.onosproject.openflow.controller.Dpid;
 import org.onosproject.openflow.controller.Dpid;
 import org.onosproject.openflow.controller.OpenFlowEventListener;
 import org.onosproject.openflow.controller.OpenFlowEventListener;
+import org.onosproject.openflow.controller.RoleState;
 import org.projectfloodlight.openflow.protocol.OFMessage;
 import org.projectfloodlight.openflow.protocol.OFMessage;
 
 
+import java.net.SocketAddress;
 import java.util.ArrayList;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.List;
 
 
@@ -65,8 +70,9 @@ public class AbstractOpenFlowSwitchTest {
 
 
         executorService = new TestExecutorService();
         executorService = new TestExecutorService();
         ofSwitch.executorMsgs = executorService;
         ofSwitch.executorMsgs = executorService;
-        Channel channel = new ChannelAdapter();
+        Channel channel = new MockChannel();
         ofSwitch.setChannel(channel);
         ofSwitch.setChannel(channel);
+        ofSwitch.role = RoleState.MASTER;
         ofSwitch.addEventListener(new OpenFlowEventListenerAdapter());
         ofSwitch.addEventListener(new OpenFlowEventListenerAdapter());
     }
     }
 
 
@@ -130,4 +136,148 @@ public class AbstractOpenFlowSwitchTest {
         public void handleMessage(Dpid dpid, OFMessage msg) {
         public void handleMessage(Dpid dpid, OFMessage msg) {
         }
         }
     }
     }
+
+    private class MockChannel implements Channel {
+
+        @Override
+        public Integer getId() {
+            return null;
+        }
+
+        @Override
+        public ChannelFactory getFactory() {
+            return null;
+        }
+
+        @Override
+        public Channel getParent() {
+            return null;
+        }
+
+        @Override
+        public ChannelConfig getConfig() {
+            return null;
+        }
+
+        @Override
+        public ChannelPipeline getPipeline() {
+            return null;
+        }
+
+        @Override
+        public boolean isOpen() {
+            return false;
+        }
+
+        @Override
+        public boolean isBound() {
+            return false;
+        }
+
+        @Override
+        public boolean isConnected() {
+            // we assume that the channel is connected
+            return true;
+        }
+
+        @Override
+        public SocketAddress getLocalAddress() {
+            return null;
+        }
+
+        @Override
+        public SocketAddress getRemoteAddress() {
+            return null;
+        }
+
+        @Override
+        public ChannelFuture write(Object message) {
+            return null;
+        }
+
+        @Override
+        public ChannelFuture write(Object message, SocketAddress remoteAddress) {
+            return null;
+        }
+
+        @Override
+        public ChannelFuture bind(SocketAddress localAddress) {
+            return null;
+        }
+
+        @Override
+        public ChannelFuture connect(SocketAddress remoteAddress) {
+            return null;
+        }
+
+        @Override
+        public ChannelFuture disconnect() {
+            return null;
+        }
+
+        @Override
+        public ChannelFuture unbind() {
+            return null;
+        }
+
+        @Override
+        public ChannelFuture close() {
+            return null;
+        }
+
+        @Override
+        public ChannelFuture getCloseFuture() {
+            return null;
+        }
+
+        @Override
+        public int getInterestOps() {
+            return 0;
+        }
+
+        @Override
+        public boolean isReadable() {
+            return false;
+        }
+
+        @Override
+        public boolean isWritable() {
+            return false;
+        }
+
+        @Override
+        public ChannelFuture setInterestOps(int interestOps) {
+            return null;
+        }
+
+        @Override
+        public ChannelFuture setReadable(boolean readable) {
+            return null;
+        }
+
+        @Override
+        public boolean getUserDefinedWritability(int index) {
+            return false;
+        }
+
+        @Override
+        public void setUserDefinedWritability(int index, boolean isWritable) {
+
+        }
+
+        @Override
+        public Object getAttachment() {
+            return null;
+        }
+
+        @Override
+        public void setAttachment(Object attachment) {
+
+        }
+
+        @Override
+        public int compareTo(Channel o) {
+            return 0;
+        }
+    }
 }
 }