lisp/lisp2vpp/src/test/java/io/fd/honeycomb/lisp/translate/read/LocalMappingCustomizerTest.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.lisp.translate.read;
  18
  19 import static io.fd.honeycomb.lisp.translate.read.dump.executor.params.MappingsDumpParams.EidType.IPV4;
  20 import static org.junit.Assert.assertEquals;
  21 import static org.junit.Assert.assertNotNull;
  22 import static org.mockito.Matchers.any;
  23 import static org.mockito.Mockito.when;
  24
  25 import com.google.common.collect.ImmutableList;
  26 import io.fd.honeycomb.lisp.context.util.EidMappingContext;
  27 import io.fd.honeycomb.lisp.translate.read.dump.executor.params.MappingsDumpParams;
  28 import io.fd.honeycomb.lisp.translate.util.EidTranslator;
  29 import io.fd.honeycomb.translate.MappingContext;
  30 import io.fd.honeycomb.translate.spi.read.ReaderCustomizer;
  31 import io.fd.honeycomb.translate.vpp.util.NamingContext;
  32 import io.fd.honeycomb.vpp.test.read.ListReaderCustomizerTest;
  33 import java.util.List;
  34 import org.junit.Test;
  35 import org.mockito.Mock;
  36 import org.opendaylight.yang.gen.v1.urn.honeycomb.params.xml.ns.yang.eid.mapping.context.rev160801.contexts.eid.mapping.context.mappings.mapping.EidBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv4;
  39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv4Builder;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.MappingId;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.dp.subtable.grouping.LocalMappings;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.dp.subtable.grouping.LocalMappingsBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.dp.subtable.grouping.local.mappings.LocalMapping;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.dp.subtable.grouping.local.mappings.LocalMappingBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.dp.subtable.grouping.local.mappings.LocalMappingKey;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.dp.subtable.grouping.local.mappings.local.mapping.Eid;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.eid.table.grouping.EidTable;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.eid.table.grouping.eid.table.VniTable;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.eid.table.grouping.eid.table.VniTableKey;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.eid.table.grouping.eid.table.vni.table.BridgeDomainSubtable;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.eid.table.grouping.eid.table.vni.table.VrfSubtable;
  52 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  53 import io.fd.vpp.jvpp.core.dto.LispEidTableDetails;
  54 import io.fd.vpp.jvpp.core.dto.LispEidTableDetailsReplyDump;
  55
  56 public class LocalMappingCustomizerTest extends
  57         ListReaderCustomizerTest<LocalMapping, LocalMappingKey, LocalMappingBuilder> implements EidTranslator {
  58
  59     private static final Ipv4
  60             EID_ADDRESS = new Ipv4Builder().setIpv4(new Ipv4Address("192.168.2.1")).build();
  61
  62     @Mock
  63     private EidMappingContext localMappingContext;
  64
  65     private InstanceIdentifier<LocalMapping> emptyIdentifier;
  66     private InstanceIdentifier<LocalMapping> validIdentifier;
  67
  68     public LocalMappingCustomizerTest() {
  69         super(LocalMapping.class, LocalMappingsBuilder.class);
  70     }
  71
  72
  73     public void setUp() {
  74         emptyIdentifier = InstanceIdentifier.create(EidTable.class)
  75                 .child(VniTable.class, new VniTableKey(12L))
  76                 .child(VrfSubtable.class)
  77                 .child(LocalMappings.class)
  78                 .child(LocalMapping.class);
  79
  80         validIdentifier = InstanceIdentifier.create(EidTable.class)
  81                 .child(VniTable.class, new VniTableKey(12L))
  82                 .child(VrfSubtable.class)
  83                 .child(LocalMappings.class)
  84                 .child(LocalMapping.class, new LocalMappingKey(new MappingId("local-mapping")));
  85
  86         defineDumpData();
  87         defineMappings();
  88     }
  89
  90     private void defineDumpData() {
  91         LispEidTableDetailsReplyDump replyDump = new LispEidTableDetailsReplyDump();
  92         LispEidTableDetails detail = new LispEidTableDetails();
  93         detail.action = 0;
  94         detail.authoritative = 1;
  95         detail.context = 4;
  96         detail.eid = new byte[]{-64, -88, 2, 1};
  97         detail.eidPrefixLen = 32;
  98         detail.eidType = (byte) IPV4.getValue();
  99         detail.isLocal = 1;
 100         detail.locatorSetIndex = 1;
 101         detail.ttl = 7;
 102         detail.vni = 12;
 103
 104         replyDump.lispEidTableDetails = ImmutableList.of(detail);
 105         when(api.lispEidTableDump(any())).thenReturn(future(replyDump));
 106     }
 107
 108     private void defineMappings() {
 109         //eid mapping
 110
 111         when(localMappingContext.getId(any(Eid.class), any(MappingContext.class)))
 112                 .thenReturn(new MappingId("local-mapping"));
 113         when(localMappingContext.containsEid(new MappingId("local-mapping"), mappingContext)).thenReturn(true);
 114         when(localMappingContext.getEid(new MappingId("local-mapping"), mappingContext)).thenReturn(new EidBuilder()
 115                 .setAddress(EID_ADDRESS).build());
 116         //naming context for locator
 117         defineMapping(mappingContext, "loc-set", 1, "locator-set-context");
 118     }
 119
 120     @Test
 121     public void readCurrentAttributes() throws Exception {
 122         LocalMappingBuilder builder = new LocalMappingBuilder();
 123         getCustomizer().readCurrentAttributes(validIdentifier, builder, ctx);
 124
 125         final LocalMapping mapping = builder.build();
 126
 127         assertNotNull(mapping);
 128         assertEquals(true, compareAddresses(EID_ADDRESS, mapping.getEid().getAddress()));
 129         assertEquals("loc-set", mapping.getLocatorSet());
 130     }
 131
 132     @Test
 133     public void getAllIds() throws Exception {
 134         final List<LocalMappingKey> keys = getCustomizer().getAllIds(emptyIdentifier, ctx);
 135
 136         assertEquals(1, keys.size());
 137         assertEquals("local-mapping", keys.get(0).getId().getValue());
 138     }
 139
 140     @Override
 141     protected ReaderCustomizer<LocalMapping, LocalMappingBuilder> initCustomizer() {
 142         return new LocalMappingCustomizer(api, new NamingContext("loc", "locator-set-context"), localMappingContext);
 143     }
 144 }