Decompose RPC implementation classes
[openflowplugin.git] / openflowplugin-impl / src / test / java / org / opendaylight / openflowplugin / impl / statistics / services / direct / singlelayer / FlowDirectStatisticsServiceTest.java
index d0eee124fc23dd0241ef1b7e7e9b982ed401d02b..6f44779e706cdc71172d4d86fcb3a02cf1168d1b 100644 (file)
@@ -15,8 +15,7 @@ import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
-import java.math.BigInteger;
-import java.util.Collections;
+import java.util.ArrayList;
 import java.util.List;
 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
 import org.opendaylight.openflowplugin.api.openflow.device.Xid;
@@ -25,6 +24,7 @@ import org.opendaylight.openflowplugin.impl.registry.flow.FlowDescriptorFactory;
 import org.opendaylight.openflowplugin.impl.statistics.services.direct.AbstractDirectStatisticsServiceTest;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Counter32;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Counter64;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.MacAddress;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetFlowStatisticsInput;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.GetFlowStatisticsOutput;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowId;
@@ -32,26 +32,34 @@ import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.F
 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.flow.and.statistics.map.list.FlowAndStatisticsMapList;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.flow.and.statistics.map.list.FlowAndStatisticsMapListBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.multipart.reply.multipart.reply.body.MultipartReplyFlowStatsBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.FlowCookie;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.FlowModFlags;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.InstructionsBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.MatchBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.multipart.request.multipart.request.body.MultipartRequestFlowStats;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.ethernet.match.fields.EthernetDestinationBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.match.EthernetMatchBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.model.statistics.types.rev130925.duration.DurationBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.multipart.types.rev170112.MultipartReply;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.multipart.types.rev170112.MultipartReplyBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.multipart.types.rev170112.MultipartRequest;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
+import org.opendaylight.yangtools.yang.common.Uint16;
+import org.opendaylight.yangtools.yang.common.Uint32;
+import org.opendaylight.yangtools.yang.common.Uint64;
+import org.opendaylight.yangtools.yang.common.Uint8;
 
 public class FlowDirectStatisticsServiceTest extends AbstractDirectStatisticsServiceTest {
-    static final Short TABLE_NO = 1;
-    private FlowDirectStatisticsService service;
+    static final Uint8 TABLE_NO = Uint8.ONE;
+    private SingleGetFlowStatistics service;
 
     @Override
     public void setUp() {
-        service = new FlowDirectStatisticsService(requestContextStack,
+        final DeviceFlowRegistry registry = mock(DeviceFlowRegistry.class);
+        service = new SingleGetFlowStatistics(requestContextStack,
                                                   deviceContext,
                                                   convertorManager,
                                                   multipartWriterProvider);
-        final DeviceFlowRegistry registry = mock(DeviceFlowRegistry.class);
         when(registry.retrieveDescriptor(any())).thenReturn(FlowDescriptorFactory.create(TABLE_NO, new FlowId("1")));
         when(deviceContext.getDeviceFlowRegistry()).thenReturn(registry);
     }
@@ -63,7 +71,7 @@ public class FlowDirectStatisticsServiceTest extends AbstractDirectStatisticsSer
         when(input.getTableId()).thenReturn(TABLE_NO);
 
         final MultipartRequestFlowStats body = (MultipartRequestFlowStats) ((MultipartRequest) service
-            .buildRequest(new Xid(42L), input))
+            .buildRequest(new Xid(Uint32.valueOf(42L)), input))
             .getMultipartRequestBody();
 
         assertEquals(TABLE_NO, body.getFlowStats().getTableId());
@@ -73,30 +81,65 @@ public class FlowDirectStatisticsServiceTest extends AbstractDirectStatisticsSer
     public void testBuildReply() {
         final FlowAndStatisticsMapList flowStat = new FlowAndStatisticsMapListBuilder()
                 .setDuration(new DurationBuilder()
-                        .setSecond(new Counter32(1L))
-                        .setNanosecond(new Counter32(1L))
+                        .setSecond(new Counter32(Uint32.ONE))
+                        .setNanosecond(new Counter32(Uint32.TWO))
                         .build())
-                .setTableId(TABLE_NO)
-                .setByteCount(new Counter64(BigInteger.ONE))
-                .setPacketCount(new Counter64(BigInteger.ONE))
+                .setTableId(Uint8.valueOf(50))
+                .setByteCount(new Counter64(Uint64.valueOf(7094)))
+                .setPacketCount(new Counter64(Uint64.valueOf(63)))
+                .setCookie(new FlowCookie(Uint64.valueOf(134419365)))
                 .setFlags(new FlowModFlags(true, false, false, false, false))
-                .setMatch(new MatchBuilder().build())
+                .setMatch(new MatchBuilder().setEthernetMatch(
+                        new EthernetMatchBuilder()
+                                .setEthernetDestination(
+                                        new EthernetDestinationBuilder()
+                                                .setAddress(new MacAddress("fa:16:3e:92:81:45"))
+                                                .build()).build()).build())
+                .setInstructions(new InstructionsBuilder().build())
+                .setPriority(Uint16.valueOf(20))
                 .build();
 
         final MultipartReply reply = new MultipartReplyBuilder()
                 .setMultipartReplyBody(new MultipartReplyFlowStatsBuilder()
-                        .setFlowAndStatisticsMapList(Collections.singletonList(flowStat))
+                        .setFlowAndStatisticsMapList(List.of(flowStat))
                         .build())
                 .build();
 
-        final List<MultipartReply> input = Collections.singletonList(reply);
 
+        final FlowAndStatisticsMapList flowStat1 = new FlowAndStatisticsMapListBuilder()
+                .setDuration(new DurationBuilder()
+                        .setSecond(new Counter32(Uint32.ONE))
+                        .setNanosecond(new Counter32(Uint32.TWO))
+                        .build())
+                .setTableId(Uint8.valueOf(51))
+                .setByteCount(new Counter64(Uint64.valueOf(9853)))
+                .setPacketCount(new Counter64(Uint64.valueOf(99)))
+                .setCookie(new FlowCookie(Uint64.valueOf(134550437)))
+                .setFlags(new FlowModFlags(true, false, false, false, false))
+                .setMatch(new MatchBuilder().setEthernetMatch(
+                        new EthernetMatchBuilder()
+                                .setEthernetDestination(
+                                        new EthernetDestinationBuilder()
+                                                .setAddress(new MacAddress("fa:16:3e:92:81:45"))
+                                                .build()).build()).build())
+                .setInstructions(new InstructionsBuilder().build())
+                .setPriority(Uint16.valueOf(20))
+                .build();
+
+        final MultipartReply reply1 = new MultipartReplyBuilder()
+                .setMultipartReplyBody(new MultipartReplyFlowStatsBuilder()
+                        .setFlowAndStatisticsMapList(List.of(flowStat1))
+                        .build())
+                .build();
+        final List<MultipartReply> input = new ArrayList<>();
+        input.add(reply);
+        input.add(reply1);
         final GetFlowStatisticsOutput output = service.buildReply(input, true);
         assertTrue(output.getFlowAndStatisticsMapList().size() > 0);
-
-        final FlowAndStatisticsMap stats = output.getFlowAndStatisticsMapList().get(0);
-
-        assertEquals(stats.getTableId(), TABLE_NO);
+        FlowAndStatisticsMap stats = output.nonnullFlowAndStatisticsMapList().iterator().next();
+        assertEquals(Uint8.valueOf(50),stats.getTableId());
+        stats = output.nonnullFlowAndStatisticsMapList().get(1);
+        assertEquals(Uint8.valueOf(51),stats.getTableId());
     }
 
     @Override
@@ -105,11 +148,11 @@ public class FlowDirectStatisticsServiceTest extends AbstractDirectStatisticsSer
         when(stat.getTableId()).thenReturn(TABLE_NO);
         when(stat.getMatch()).thenReturn(new MatchBuilder().build());
 
-        final List<FlowAndStatisticsMapList> stats = Collections.singletonList(stat);
+        final List<FlowAndStatisticsMapList> stats = List.of(stat);
         final GetFlowStatisticsOutput output = mock(GetFlowStatisticsOutput.class);
-        when(output.getFlowAndStatisticsMapList()).thenReturn(stats);
+        when(output.nonnullFlowAndStatisticsMapList()).thenReturn(stats);
 
-        multipartWriterProvider.lookup(MultipartType.OFPMPFLOW).get().write(output, true);
+        multipartWriterProvider.lookup(MultipartType.OFPMPFLOW).orElseThrow().write(output, true);
         verify(deviceContext).writeToTransactionWithParentsSlow(eq(LogicalDatastoreType.OPERATIONAL), any(), any());
     }
 }