openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/statistics/MultipartReplyTranslatorTest.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.openflowplugin.impl.statistics;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertTrue;
  12 import static org.mockito.Mockito.mock;
  13 import static org.mockito.Mockito.when;
  14
  15 import com.google.common.collect.Lists;
  16 import java.math.BigInteger;
  17 import java.util.Collections;
  18 import java.util.List;
  19 import org.junit.Before;
  20 import org.junit.Test;
  21 import org.opendaylight.openflowplugin.api.OFConstants;
  22 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
  23 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
  24 import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
  25 import org.opendaylight.openflowplugin.impl.common.MultipartReplyTranslatorUtil;
  26 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorManager;
  27 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorManagerFactory;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.FlowAndStatisticsMapList;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupDescStatsReply;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupStatisticsReply;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupTypes;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.group.desc.stats.reply.GroupDescStats;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.model.statistics.types.rev130925.AggregateFlowStatistics;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.GroupId;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.GroupType;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartRequestFlags;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.FeaturesReply;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReplyMessage;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.buckets.grouping.BucketsList;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.MultipartReplyBody;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyAggregateCaseBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyFlowCaseBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyGroupCaseBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyGroupDescCaseBuilder;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyPortStatsCaseBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.aggregate._case.MultipartReplyAggregateBuilder;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.flow._case.MultipartReplyFlowBuilder;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.flow._case.multipart.reply.flow.FlowStats;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group._case.MultipartReplyGroupBuilder;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group._case.multipart.reply.group.GroupStatsBuilder;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group._case.multipart.reply.group.group.stats.BucketStats;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group.desc._case.MultipartReplyGroupDescBuilder;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group.desc._case.multipart.reply.group.desc.GroupDescBuilder;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.port.stats._case.MultipartReplyPortStatsBuilder;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.port.stats._case.multipart.reply.port.stats.PortStatsBuilder;
  57 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.node.connector.statistics.and.port.number.map.NodeConnectorStatisticsAndPortNumberMap;
  58 import org.opendaylight.yangtools.yang.binding.DataContainer;
  59
  60 public class MultipartReplyTranslatorTest {
  61
  62     private static final BigInteger DUMMY_DATAPATH_ID = new BigInteger("21");
  63     private static final Long DUMMY_XID = 1L;
  64     private static final BigInteger DUMMY_BYTE_COUNT = new BigInteger("31");
  65     private static final BigInteger DUMMY_PACKET_COUNT = new BigInteger("41");
  66     private static final Long DUMMY_FLOW_COUNT = 51L;
  67     private static final Long DUMMY_PORT_NO = 154L;
  68     private static final BigInteger DUMMY_RX_BYTES = new BigInteger("155");
  69     private static final BigInteger DUMMY_TX_BYTES = new BigInteger("165");
  70     private static final BigInteger DUMMY_RX_PACKETS = new BigInteger("175");
  71     private static final BigInteger DUMMY_TX_PACKETS = new BigInteger("185");
  72     private static final BigInteger DUMMY_COLLISIONS = new BigInteger("195");
  73     private static final BigInteger DUMMY_RX_CRC_ERR = new BigInteger("205");
  74     private static final BigInteger DUMMY_RX_DROPPED = new BigInteger("215");
  75     private static final BigInteger DUMMY_RX_ERRORS = new BigInteger("225");
  76     private static final BigInteger DUMMY_RX_FRAME_ERR = new BigInteger("235");
  77     private static final BigInteger DUMMY_OVER_ERR = new BigInteger("245");
  78     private static final BigInteger DUMMY_TX_DROPPED = new BigInteger("255");
  79     private static final BigInteger DUMMY_TX_ERRORS = new BigInteger("265");
  80     private static final Long DUMMY_DURATION_SEC = 3453L;
  81     private static final Long DUMMY_DURATION_NSEC = 3343L;
  82     private static final GroupId DUMMY_GROUP_ID = new GroupId(55L);
  83     private static final Long DUMMY_REF_COUNT = 1234L;
  84     private static final GroupTypes DUMMY_GROUPS_TYPE = GroupTypes.GroupAll;
  85     private static final GroupType DUMMY_GROUP_TYPE = GroupType.OFPGTALL;
  86     private static final ConvertorManager CONVERTOR_MANAGER = ConvertorManagerFactory.createDefaultManager();
  87
  88     @Before
  89     public void setUp() {
  90     }
  91
  92     @Test
  93     public void testTranslateFlow() {
  94         DeviceContext mockedDeviceContext = mock(DeviceContext.class);
  95
  96         MultipartReplyMessage multipartReplyMessage =
  97                 prepareMocks(mockedDeviceContext, prepareMultipartReplyFlow(), MultipartType.OFPMPFLOW);
  98
  99         DataContainer result = MultipartReplyTranslatorUtil.translate(
 100             multipartReplyMessage,
 101             mockedDeviceContext.getDeviceInfo(),
 102                 CONVERTOR_MANAGER,
 103             mockedDeviceContext.oook()).get();
 104
 105         DataContainer dataObject = validateOutput(result);
 106         assertTrue(dataObject instanceof FlowAndStatisticsMapList);
 107     }
 108
 109     @Test
 110     public void testTranslateAggregate() {
 111         DeviceContext mockedDeviceContext = mock(DeviceContext.class);
 112
 113         MultipartReplyMessage multipartReplyMessage =
 114                 prepareMocks(mockedDeviceContext, prepareMultipartReplyAggregate(), MultipartType.OFPMPAGGREGATE);
 115
 116         DataContainer result = MultipartReplyTranslatorUtil.translate(
 117             multipartReplyMessage,
 118             mockedDeviceContext.getDeviceInfo(),
 119                 CONVERTOR_MANAGER,
 120             mockedDeviceContext.oook()).get();
 121
 122         DataContainer dataObject = validateOutput(result);
 123         assertTrue(dataObject instanceof AggregateFlowStatistics);
 124         AggregateFlowStatistics message = (AggregateFlowStatistics)dataObject;
 125         assertEquals(DUMMY_BYTE_COUNT, message.getByteCount().getValue());
 126         assertEquals(DUMMY_PACKET_COUNT, message.getPacketCount().getValue());
 127         assertEquals(DUMMY_FLOW_COUNT, message.getFlowCount().getValue());
 128     }
 129
 130     @Test
 131     public void testTranslatePortStats() {
 132         DeviceContext mockedDeviceContext = mock(DeviceContext.class);
 133
 134         MultipartReplyMessage multipartReplyMessage =
 135                 prepareMocks(mockedDeviceContext, prepareMultipartReplyPortStats(), MultipartType.OFPMPPORTSTATS);
 136
 137         DataContainer result = MultipartReplyTranslatorUtil.translate(
 138             multipartReplyMessage,
 139             mockedDeviceContext.getDeviceInfo(),
 140                 CONVERTOR_MANAGER,
 141             mockedDeviceContext.oook()).get();
 142
 143         DataContainer dataObject = validateOutput(result);
 144         assertTrue(dataObject instanceof org.opendaylight.yang.gen.v1.urn
 145                 .opendaylight.port.statistics.rev131214.NodeConnectorStatisticsAndPortNumberMap);
 146         org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.NodeConnectorStatisticsAndPortNumberMap
 147                 nodeConnectorStatisticsUpdate = (org.opendaylight.yang.gen.v1.urn
 148                 .opendaylight.port.statistics.rev131214.NodeConnectorStatisticsAndPortNumberMap) dataObject;
 149         List<NodeConnectorStatisticsAndPortNumberMap> nodeConnectorStatisticsAndPortNumberMaps =
 150                 nodeConnectorStatisticsUpdate.getNodeConnectorStatisticsAndPortNumberMap();
 151         assertEquals(1, nodeConnectorStatisticsAndPortNumberMaps.size());
 152         NodeConnectorStatisticsAndPortNumberMap nodeConnectorStatisticsAndPortNumberMap =
 153                 nodeConnectorStatisticsAndPortNumberMaps.get(0);
 154         assertEquals("openflow:" + DUMMY_DATAPATH_ID + ":" + DUMMY_PORT_NO,
 155                 nodeConnectorStatisticsAndPortNumberMap.getNodeConnectorId().getValue());
 156         assertEquals(DUMMY_RX_BYTES, nodeConnectorStatisticsAndPortNumberMap.getBytes().getReceived());
 157         assertEquals(DUMMY_TX_BYTES, nodeConnectorStatisticsAndPortNumberMap.getBytes().getTransmitted());
 158         assertEquals(DUMMY_RX_PACKETS, nodeConnectorStatisticsAndPortNumberMap.getPackets().getReceived());
 159         assertEquals(DUMMY_TX_PACKETS, nodeConnectorStatisticsAndPortNumberMap.getPackets().getTransmitted());
 160         assertEquals(DUMMY_COLLISIONS, nodeConnectorStatisticsAndPortNumberMap.getCollisionCount());
 161         assertEquals(DUMMY_RX_CRC_ERR, nodeConnectorStatisticsAndPortNumberMap.getReceiveCrcError());
 162         assertEquals(DUMMY_RX_DROPPED, nodeConnectorStatisticsAndPortNumberMap.getReceiveDrops());
 163         assertEquals(DUMMY_RX_ERRORS, nodeConnectorStatisticsAndPortNumberMap.getReceiveErrors());
 164         assertEquals(DUMMY_RX_FRAME_ERR, nodeConnectorStatisticsAndPortNumberMap.getReceiveFrameError());
 165         assertEquals(DUMMY_OVER_ERR, nodeConnectorStatisticsAndPortNumberMap.getReceiveOverRunError());
 166         assertEquals(DUMMY_TX_DROPPED, nodeConnectorStatisticsAndPortNumberMap.getTransmitDrops());
 167         assertEquals(DUMMY_TX_ERRORS, nodeConnectorStatisticsAndPortNumberMap.getTransmitErrors());
 168     }
 169
 170     @Test
 171     public void testTranslateGroup() {
 172         DeviceContext mockedDeviceContext = mock(DeviceContext.class);
 173
 174         MultipartReplyMessage multipartReplyMessage =
 175                 prepareMocks(mockedDeviceContext, prepareMultipartReplyGroup(), MultipartType.OFPMPGROUP);
 176
 177         DataContainer result = MultipartReplyTranslatorUtil.translate(
 178             multipartReplyMessage,
 179             mockedDeviceContext.getDeviceInfo(),
 180                 CONVERTOR_MANAGER,
 181             mockedDeviceContext.oook()).get();
 182
 183         DataContainer dataObject = validateOutput(result);
 184         assertTrue(dataObject instanceof GroupStatisticsReply);
 185         GroupStatisticsReply groupStatisticsUpdate = (GroupStatisticsReply)dataObject;
 186         List<org.opendaylight.yang.gen.v1.urn
 187                 .opendaylight.group.types.rev131018.group.statistics.reply.GroupStats> groupStats =
 188                 groupStatisticsUpdate.getGroupStats();
 189         assertEquals(1, groupStats.size());
 190         org.opendaylight.yang.gen.v1.urn
 191                 .opendaylight.group.types.rev131018.group.statistics.reply.GroupStats groupStat = groupStats.get(0);
 192
 193         assertEquals(DUMMY_BYTE_COUNT, groupStat.getByteCount().getValue());
 194         assertEquals(DUMMY_DURATION_SEC, groupStat.getDuration().getSecond().getValue());
 195         assertEquals(DUMMY_DURATION_NSEC, groupStat.getDuration().getNanosecond().getValue());
 196         assertEquals(DUMMY_GROUP_ID.getValue(), groupStat.getGroupId().getValue());
 197         assertEquals(DUMMY_PACKET_COUNT, groupStat.getPacketCount().getValue());
 198         assertEquals(DUMMY_REF_COUNT, groupStat.getRefCount().getValue());
 199     }
 200
 201     @Test
 202     public void testTranslateGroupDesc() {
 203         DeviceContext mockedDeviceContext = mock(DeviceContext.class);
 204
 205         MultipartReplyMessage multipartReplyMessage =
 206                 prepareMocks(mockedDeviceContext, prepareMultipartReplyGroupDesc(), MultipartType.OFPMPGROUPDESC);
 207
 208         DataContainer result = MultipartReplyTranslatorUtil.translate(
 209             multipartReplyMessage,
 210             mockedDeviceContext.getDeviceInfo(),
 211                 CONVERTOR_MANAGER,
 212             mockedDeviceContext.oook()).get();
 213
 214         DataContainer dataObject = validateOutput(result);
 215         assertTrue(dataObject instanceof GroupDescStatsReply);
 216         GroupDescStatsReply groupStatistics = (GroupDescStatsReply) dataObject;
 217         List<GroupDescStats> groupDescStats = groupStatistics.getGroupDescStats();
 218         assertEquals(1, groupDescStats.size());
 219         GroupDescStats groupDescStat = groupDescStats.get(0);
 220         assertEquals(DUMMY_GROUP_ID.getValue(),groupDescStat.getGroupId().getValue());
 221         assertEquals(DUMMY_GROUPS_TYPE,groupDescStat.getGroupType());
 222     }
 223
 224     private MultipartReplyBody prepareMultipartReplyGroupDesc() {
 225         GroupDescBuilder groupDescBuilder = new GroupDescBuilder();
 226         groupDescBuilder.setGroupId(DUMMY_GROUP_ID);
 227         groupDescBuilder.setBucketsList(Collections.<BucketsList>emptyList());
 228         groupDescBuilder.setType(DUMMY_GROUP_TYPE);
 229         MultipartReplyGroupDescBuilder multipartReplyGroupDescBuilder = new MultipartReplyGroupDescBuilder();
 230         multipartReplyGroupDescBuilder.setGroupDesc(Lists.newArrayList(groupDescBuilder.build()));
 231         MultipartReplyGroupDescCaseBuilder multipartReplyGroupDescCaseBuilder =
 232                 new MultipartReplyGroupDescCaseBuilder();
 233         multipartReplyGroupDescCaseBuilder.setMultipartReplyGroupDesc(multipartReplyGroupDescBuilder.build());
 234         return multipartReplyGroupDescCaseBuilder.build();
 235     }
 236
 237     private MultipartReplyBody prepareMultipartReplyGroup() {
 238         GroupStatsBuilder groupStatsBuilder = new GroupStatsBuilder();
 239         groupStatsBuilder.setByteCount(DUMMY_BYTE_COUNT);
 240         groupStatsBuilder.setBucketStats(Collections.<BucketStats>emptyList());
 241         groupStatsBuilder.setDurationSec(DUMMY_DURATION_SEC);
 242         groupStatsBuilder.setDurationNsec(DUMMY_DURATION_NSEC);
 243         groupStatsBuilder.setGroupId(DUMMY_GROUP_ID);
 244         groupStatsBuilder.setPacketCount(DUMMY_PACKET_COUNT);
 245         groupStatsBuilder.setRefCount(DUMMY_REF_COUNT);
 246         MultipartReplyGroupBuilder multipartReplyGroupBuilder = new MultipartReplyGroupBuilder();
 247         multipartReplyGroupBuilder.setGroupStats(Lists.newArrayList(groupStatsBuilder.build()));
 248         MultipartReplyGroupCaseBuilder multipartReplyGroupCaseBuilder = new MultipartReplyGroupCaseBuilder();
 249         multipartReplyGroupCaseBuilder.setMultipartReplyGroup(multipartReplyGroupBuilder.build());
 250         return multipartReplyGroupCaseBuilder.build();
 251     }
 252
 253     private MultipartReplyBody prepareMultipartReplyPortStats() {
 254         PortStatsBuilder dummyPortStatBuilder = new PortStatsBuilder();
 255         dummyPortStatBuilder.setPortNo(DUMMY_PORT_NO);
 256         dummyPortStatBuilder.setRxBytes(DUMMY_RX_BYTES);
 257         dummyPortStatBuilder.setTxBytes(DUMMY_TX_BYTES);
 258         dummyPortStatBuilder.setRxPackets(DUMMY_RX_PACKETS);
 259         dummyPortStatBuilder.setTxPackets(DUMMY_TX_PACKETS);
 260         dummyPortStatBuilder.setCollisions(DUMMY_COLLISIONS);
 261         dummyPortStatBuilder.setRxCrcErr(DUMMY_RX_CRC_ERR);
 262         dummyPortStatBuilder.setRxDropped(DUMMY_RX_DROPPED);
 263         dummyPortStatBuilder.setRxErrors(DUMMY_RX_ERRORS);
 264         dummyPortStatBuilder.setRxFrameErr(DUMMY_RX_FRAME_ERR);
 265         dummyPortStatBuilder.setRxOverErr(DUMMY_OVER_ERR);
 266         dummyPortStatBuilder.setTxDropped(DUMMY_TX_DROPPED);
 267         dummyPortStatBuilder.setTxErrors(DUMMY_TX_ERRORS);
 268         MultipartReplyPortStatsBuilder multipartReplyPortStatsBuilder = new MultipartReplyPortStatsBuilder();
 269         multipartReplyPortStatsBuilder.setPortStats(Lists.newArrayList(dummyPortStatBuilder.build()));
 270         MultipartReplyPortStatsCaseBuilder multipartReplyPortStatsCaseBuilder =
 271                 new MultipartReplyPortStatsCaseBuilder();
 272         multipartReplyPortStatsCaseBuilder.setMultipartReplyPortStats(multipartReplyPortStatsBuilder.build());
 273         return multipartReplyPortStatsCaseBuilder.build();
 274     }
 275
 276
 277     private MultipartReplyBody prepareMultipartReplyAggregate() {
 278         MultipartReplyAggregateBuilder multipartReplyAggregateBuilder = new MultipartReplyAggregateBuilder();
 279         multipartReplyAggregateBuilder.setByteCount(DUMMY_BYTE_COUNT);
 280         multipartReplyAggregateBuilder.setPacketCount(DUMMY_PACKET_COUNT);
 281         multipartReplyAggregateBuilder.setFlowCount(DUMMY_FLOW_COUNT);
 282         MultipartReplyAggregateCaseBuilder multipartReplyAggregateCaseBuilder =
 283                 new MultipartReplyAggregateCaseBuilder();
 284         multipartReplyAggregateCaseBuilder.setMultipartReplyAggregate(multipartReplyAggregateBuilder.build());
 285         return multipartReplyAggregateCaseBuilder.build();
 286     }
 287
 288     private MultipartReplyBody prepareMultipartReplyFlow() {
 289         MultipartReplyFlowCaseBuilder multipartReplyFlowCaseBuilder = new MultipartReplyFlowCaseBuilder();
 290         MultipartReplyFlowBuilder multipartReplyFlowBuilder = new MultipartReplyFlowBuilder();
 291         multipartReplyFlowBuilder.setFlowStats(Collections.<FlowStats>emptyList());
 292         multipartReplyFlowCaseBuilder.setMultipartReplyFlow(multipartReplyFlowBuilder.build());
 293         return multipartReplyFlowCaseBuilder.build();
 294     }
 295
 296     private MultipartReplyMessage prepareMocks(DeviceContext mockedDeviceContext,
 297                                                MultipartReplyBody multipartReplyBody,
 298                                                final MultipartType multipartType) {
 299         ConnectionContext mockedConnectionContext = mock(ConnectionContext.class);
 300         FeaturesReply mockedFeaturesReply = mock(FeaturesReply.class);
 301         when(mockedFeaturesReply.getVersion()).thenReturn(OFConstants.OFP_VERSION_1_3);
 302         when(mockedFeaturesReply.getDatapathId()).thenReturn(DUMMY_DATAPATH_ID);
 303
 304         DeviceInfo deviceInfo = mock(DeviceInfo.class);
 305         when(deviceInfo.getVersion()).thenReturn(OFConstants.OFP_VERSION_1_3);
 306         when(deviceInfo.getDatapathId()).thenReturn(DUMMY_DATAPATH_ID);
 307         when(deviceInfo.getLOGValue()).thenReturn(DUMMY_DATAPATH_ID.toString());
 308         when(mockedDeviceContext.getDeviceInfo()).thenReturn(deviceInfo);
 309
 310         when(mockedConnectionContext.getFeatures()).thenReturn(mockedFeaturesReply);
 311         when(mockedDeviceContext.getPrimaryConnectionContext()).thenReturn(mockedConnectionContext);
 312
 313         MultipartReplyMessage multipartReplyMessage = mock(MultipartReplyMessage.class);
 314         when(multipartReplyMessage.getType()).thenReturn(multipartType);
 315         when(multipartReplyMessage.getFlags()).thenReturn(new MultipartRequestFlags(true));
 316         when(multipartReplyMessage.getXid()).thenReturn(DUMMY_XID);
 317         when(multipartReplyMessage.getMultipartReplyBody()).thenReturn(multipartReplyBody);
 318         return multipartReplyMessage;
 319     }
 320
 321     private DataContainer validateOutput(DataContainer dataObject) {
 322         return dataObject;
 323     }
 324 }