v3po/v3po2vpp/src/test/java/io/fd/honeycomb/translate/v3po/vppclassifier/ClassifyTableReaderTest.java

   1 /*
   2  * Copyright (c) 2016 Cisco and/or its affiliates.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package io.fd.honeycomb.translate.v3po.vppclassifier;
  18
  19 import static org.junit.Assert.assertEquals;
  20 import static org.mockito.Matchers.any;
  21 import static org.mockito.Matchers.anyString;
  22 import static org.mockito.Mockito.doReturn;
  23 import static org.mockito.Mockito.mock;
  24 import static org.mockito.Mockito.times;
  25 import static org.mockito.Mockito.verify;
  26 import static org.mockito.Mockito.when;
  27
  28 import com.google.common.base.Optional;
  29 import io.fd.honeycomb.translate.read.ReadFailedException;
  30 import io.fd.honeycomb.translate.spi.read.ReaderCustomizer;
  31 import io.fd.honeycomb.vpp.test.read.ListReaderCustomizerTest;
  32 import java.util.List;
  33 import org.junit.Test;
  34 import org.mockito.Mock;
  35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.HexString;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.classifier.rev161214.PacketHandlingAction;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.classifier.rev161214.VppClassifierState;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.classifier.rev161214.VppClassifierStateBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.classifier.rev161214.VppNode;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.classifier.rev161214.vpp.classifier.state.ClassifyTable;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.classifier.rev161214.vpp.classifier.state.ClassifyTableBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.classifier.rev161214.vpp.classifier.state.ClassifyTableKey;
  43 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  44 import io.fd.vpp.jvpp.core.dto.ClassifyTableIds;
  45 import io.fd.vpp.jvpp.core.dto.ClassifyTableIdsReply;
  46 import io.fd.vpp.jvpp.core.dto.ClassifyTableInfo;
  47 import io.fd.vpp.jvpp.core.dto.ClassifyTableInfoReply;
  48
  49 public class ClassifyTableReaderTest extends
  50     ListReaderCustomizerTest<ClassifyTable, ClassifyTableKey, ClassifyTableBuilder> {
  51
  52     private static final int TABLE_INDEX_1 = 1;
  53     private static final String TABLE_NAME_1 = "table1";
  54     private static final int TABLE_INDEX_2 = 2;
  55     private static final String TABLE_NAME_2 = "table2";
  56
  57     @Mock
  58     private VppClassifierContextManager classifierContext;
  59
  60     public ClassifyTableReaderTest() {
  61         super(ClassifyTable.class, VppClassifierStateBuilder.class);
  62     }
  63
  64     @Override
  65     protected ReaderCustomizer<ClassifyTable, ClassifyTableBuilder> initCustomizer() {
  66         return new ClassifyTableReader(api, classifierContext);
  67     }
  68
  69     private static InstanceIdentifier<ClassifyTable> getClassifyTableId(final String name) {
  70         return InstanceIdentifier.create(VppClassifierState.class)
  71             .child(ClassifyTable.class, new ClassifyTableKey(name));
  72     }
  73
  74     private static ClassifyTableInfoReply generateClassifyTableInfoReply() {
  75         final ClassifyTableInfoReply reply = new ClassifyTableInfoReply();
  76         reply.tableId = TABLE_INDEX_1;
  77         reply.nbuckets = 2;
  78         reply.skipNVectors = 0;
  79         reply.matchNVectors = 1;
  80         reply.nextTableIndex = ~0;
  81         reply.missNextIndex = ~0;
  82         reply.mask =
  83             new byte[] {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, (byte) 0x01, (byte) 0x02, (byte) 0x03, (byte) 0x04,
  84                 (byte) 0x05, (byte) 0x06, 0x00, 0x00, 0x00, 0x00};
  85         return reply;
  86     }
  87
  88     private void verifyClasifyTableRead(final ClassifyTableBuilder builder) {
  89         verify(builder).setName(TABLE_NAME_1);
  90         verify(builder).setNbuckets(2L);
  91         verify(builder, times(0)).setNextTable(anyString());
  92         verify(builder).setMissNext(new VppNode(PacketHandlingAction.Permit));
  93         verify(builder).setMask(new HexString("00:00:00:00:00:00:01:02:03:04:05:06:00:00:00:00"));
  94         verify(builder).setActiveSessions(0L);
  95     }
  96
  97     @Test
  98     public void testRead() throws ReadFailedException {
  99         doReturn(future(generateClassifyTableInfoReply())).when(api).classifyTableInfo(any(ClassifyTableInfo.class));
 100
 101         when(classifierContext.containsTable(TABLE_NAME_1, mappingContext)).thenReturn(true);
 102         when(classifierContext.getTableIndex(TABLE_NAME_1, mappingContext)).thenReturn(TABLE_INDEX_1);
 103         when(classifierContext.getTableBaseNode(TABLE_NAME_1, mappingContext)).thenReturn(Optional.absent());
 104
 105         final ClassifyTableBuilder builder = mock(ClassifyTableBuilder.class);
 106         getCustomizer().readCurrentAttributes(getClassifyTableId(TABLE_NAME_1), builder, ctx);
 107
 108         verifyClasifyTableRead(builder);
 109     }
 110
 111     @Test
 112     public void testGetAllIds() throws ReadFailedException {
 113         final ClassifyTableIdsReply reply = new ClassifyTableIdsReply();
 114         reply.ids = new int[] {1, 2};
 115         doReturn(future(reply)).when(api).classifyTableIds(any(ClassifyTableIds.class));
 116
 117         when(classifierContext.getTableName(TABLE_INDEX_1, mappingContext)).thenReturn(TABLE_NAME_1);
 118         when(classifierContext.getTableName(TABLE_INDEX_2, mappingContext)).thenReturn(TABLE_NAME_2);
 119
 120         final List<ClassifyTableKey> allIds = getCustomizer().getAllIds(getClassifyTableId(TABLE_NAME_1), ctx);
 121
 122         assertEquals(reply.ids.length, allIds.size());
 123         assertEquals(TABLE_NAME_1, allIds.get(0).getName());
 124         assertEquals(TABLE_NAME_2, allIds.get(1).getName());
 125     }
 126 }