diff options
author | 2017-08-16 10:37:55 +0200 | |
---|---|---|
committer | 2017-08-16 13:20:28 +0200 | |
commit | c4ec83562d4c4582be64de637e76826718e2beaa (patch) | |
tree | 542d71d7c56955035af78683fd60333b2c1c8529 /nat/nat2vpp/src/test/java/io | |
parent | 8b051e21ed6e4e4c24415c40fd0b6c6b684c071d (diff) |
HC2VPP-105: add support for nat64 pool read
Change-Id: If458e0b812435de5ae41af1bdbf53e2ffaf94e35
Signed-off-by: Marek Gradzki <mgradzki@cisco.com>
Diffstat (limited to 'nat/nat2vpp/src/test/java/io')
-rw-r--r-- | nat/nat2vpp/src/test/java/io/fd/hc2vpp/nat/read/ExternalIpPoolCustomizerTest.java | 156 |
1 files changed, 114 insertions, 42 deletions
diff --git a/nat/nat2vpp/src/test/java/io/fd/hc2vpp/nat/read/ExternalIpPoolCustomizerTest.java b/nat/nat2vpp/src/test/java/io/fd/hc2vpp/nat/read/ExternalIpPoolCustomizerTest.java index 8413bce30..e6dfeb86f 100644 --- a/nat/nat2vpp/src/test/java/io/fd/hc2vpp/nat/read/ExternalIpPoolCustomizerTest.java +++ b/nat/nat2vpp/src/test/java/io/fd/hc2vpp/nat/read/ExternalIpPoolCustomizerTest.java @@ -20,14 +20,18 @@ import static org.hamcrest.MatcherAssert.assertThat; import static org.hamcrest.collection.IsEmptyCollection.empty; import static org.hamcrest.core.IsCollectionContaining.hasItems; import static org.junit.Assert.assertEquals; +import static org.mockito.ArgumentMatchers.any; import static org.mockito.Mockito.when; +import static org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.nat.rev170804.NatPoolType.Nat44; +import static org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.nat.rev170804.NatPoolType.Nat64; import io.fd.hc2vpp.common.test.read.ListReaderCustomizerTest; import io.fd.honeycomb.translate.spi.read.ReaderCustomizer; -import io.fd.honeycomb.translate.util.read.cache.DumpCacheManager; -import io.fd.honeycomb.translate.util.read.cache.EntityDumpExecutor; +import io.fd.vpp.jvpp.snat.dto.Nat64PoolAddrDetails; +import io.fd.vpp.jvpp.snat.dto.Nat64PoolAddrDetailsReplyDump; import io.fd.vpp.jvpp.snat.dto.SnatAddressDetails; import io.fd.vpp.jvpp.snat.dto.SnatAddressDetailsReplyDump; +import io.fd.vpp.jvpp.snat.future.FutureJVppSnatFacade; import java.util.Arrays; import java.util.List; import java.util.stream.LongStream; @@ -41,99 +45,167 @@ import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev1509 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.nat.state.nat.instances.NatInstanceKey; import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.nat.state.nat.instances.nat.instance.NatCurrentConfig; import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.nat.state.nat.instances.nat.instance.NatCurrentConfigBuilder; +import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.nat.rev170804.ExternalIpAddressPoolStateAugmentation; import org.opendaylight.yangtools.yang.binding.InstanceIdentifier; public class ExternalIpPoolCustomizerTest extends ListReaderCustomizerTest<ExternalIpAddressPool, ExternalIpAddressPoolKey, ExternalIpAddressPoolBuilder> { - private InstanceIdentifier<ExternalIpAddressPool> externalPoolIdDefaultNatInstance; - private InstanceIdentifier<ExternalIpAddressPool> externalPoolIdDifferentNatInstance; - private DumpCacheManager<SnatAddressDetailsReplyDump, Void> dumpCacheManager; + private static final InstanceIdentifier<NatCurrentConfig> NAT_CONFIG_ID = + InstanceIdentifier.create(NatInstances.class) + .child(NatInstance.class, new NatInstanceKey(NatInstanceCustomizer.DEFAULT_VRF_ID)) + .child(NatCurrentConfig.class); + + private static final InstanceIdentifier<ExternalIpAddressPool> NAT_DEFAULT_POOL_WILDCARDED_ID = + NAT_CONFIG_ID.child(ExternalIpAddressPool.class); + + private static final InstanceIdentifier<ExternalIpAddressPool> NAT_NON_DEFAULT_POOL_WILDCARDED_ID = + InstanceIdentifier.create(NatInstances.class).child(NatInstance.class, new NatInstanceKey(7L)) + .child(NatCurrentConfig.class).child(ExternalIpAddressPool.class); @Mock - private EntityDumpExecutor<SnatAddressDetailsReplyDump, Void> executor; + private FutureJVppSnatFacade jvppSnat; public ExternalIpPoolCustomizerTest() { super(ExternalIpAddressPool.class, NatCurrentConfigBuilder.class); } @Override - protected void setUp() throws Exception { - externalPoolIdDefaultNatInstance = InstanceIdentifier - .create(NatInstances.class) - .child(NatInstance.class, new NatInstanceKey(NatInstanceCustomizer.DEFAULT_VRF_ID)) - .child(NatCurrentConfig.class) - .child(ExternalIpAddressPool.class, new ExternalIpAddressPoolKey(2L)); - - externalPoolIdDifferentNatInstance = InstanceIdentifier - .create(NatInstances.class) - .child(NatInstance.class, new NatInstanceKey(7L)) - .child(NatCurrentConfig.class) - .child(ExternalIpAddressPool.class, new ExternalIpAddressPoolKey(2L)); - - dumpCacheManager = new DumpCacheManager.DumpCacheManagerBuilder<SnatAddressDetailsReplyDump, Void>() - .withExecutor(executor) - .acceptOnly(SnatAddressDetailsReplyDump.class) - .build(); + protected ReaderCustomizer<ExternalIpAddressPool, ExternalIpAddressPoolBuilder> initCustomizer() { + return new ExternalIpPoolCustomizer(jvppSnat); } - @Override - protected ReaderCustomizer<ExternalIpAddressPool, ExternalIpAddressPoolBuilder> initCustomizer() { - return new ExternalIpPoolCustomizer(dumpCacheManager); + @Test + public void testReadAttributesNat44() throws Exception { + when(jvppSnat.snatAddressDump(any())).thenReturn(future(dumpReplyNat44NonEmpty())); + final long poolId = 2; + final ExternalIpAddressPoolBuilder builder = new ExternalIpAddressPoolBuilder(); + getCustomizer().readCurrentAttributes(getId(poolId), builder, ctx); + + assertEquals("192.168.44.3/32", builder.getExternalIpPool().getValue()); + assertEquals(poolId, builder.getPoolId().longValue()); + assertEquals(Nat44, builder.getAugmentation(ExternalIpAddressPoolStateAugmentation.class).getPoolType()); + } + + @Test + public void testReadAttributesNat64() throws Exception { + when(jvppSnat.snatAddressDump(any())).thenReturn(future(dumpReplyNat44Empty())); + when(jvppSnat.nat64PoolAddrDump(any())).thenReturn(future(dumpReplyNat64NonEmpty())); + final long poolId = 2; + + final ExternalIpAddressPoolBuilder builder = new ExternalIpAddressPoolBuilder(); + getCustomizer().readCurrentAttributes(getId(poolId), builder, ctx); + + assertEquals("192.168.64.3/32", builder.getExternalIpPool().getValue()); + assertEquals(poolId, builder.getPoolId().longValue()); + assertEquals(Nat64, builder.getAugmentation(ExternalIpAddressPoolStateAugmentation.class).getPoolType()); } @Test public void testReadAttributes() throws Exception { - when(executor.executeDump(externalPoolIdDefaultNatInstance, null)).thenReturn(dumpReplyNonEmpty()); + when(jvppSnat.snatAddressDump(any())).thenReturn(future(dumpReplyNat44NonEmpty())); + when(jvppSnat.nat64PoolAddrDump(any())).thenReturn(future(dumpReplyNat64NonEmpty())); + final long poolId = 5; final ExternalIpAddressPoolBuilder builder = new ExternalIpAddressPoolBuilder(); - getCustomizer().readCurrentAttributes(externalPoolIdDefaultNatInstance, builder, ctx); + getCustomizer().readCurrentAttributes(getId(poolId), builder, ctx); + + assertEquals("192.168.64.3/32", builder.getExternalIpPool().getValue()); + assertEquals(poolId, builder.getPoolId().longValue()); + assertEquals(Nat64, builder.getAugmentation(ExternalIpAddressPoolStateAugmentation.class).getPoolType()); + } + + @Test + public void testGetAllNat44() throws Exception { + when(jvppSnat.snatAddressDump(any())).thenReturn(future(dumpReplyNat44NonEmpty())); + when(jvppSnat.nat64PoolAddrDump(any())).thenReturn(future(dumpReplyNat64Empty())); - assertEquals("192.168.2.3/32", builder.getExternalIpPool().getValue()); - assertEquals(2L, builder.getPoolId().longValue()); + final List<ExternalIpAddressPoolKey> allIds = getCustomizer().getAllIds(NAT_DEFAULT_POOL_WILDCARDED_ID, ctx); + assertThat(allIds, hasItems( + LongStream.range(0, 2).mapToObj(ExternalIpAddressPoolKey::new) + .toArray(ExternalIpAddressPoolKey[]::new))); + } + + @Test + public void testGetAllNat64() throws Exception { + when(jvppSnat.snatAddressDump(any())).thenReturn(future(dumpReplyNat44Empty())); + when(jvppSnat.nat64PoolAddrDump(any())).thenReturn(future(dumpReplyNat64NonEmpty())); + + final List<ExternalIpAddressPoolKey> allIds = getCustomizer().getAllIds(NAT_DEFAULT_POOL_WILDCARDED_ID, ctx); + assertThat(allIds, hasItems( + LongStream.range(0, 2).mapToObj(ExternalIpAddressPoolKey::new) + .toArray(ExternalIpAddressPoolKey[]::new))); } @Test public void testGetAll() throws Exception { - when(executor.executeDump(externalPoolIdDefaultNatInstance, null)).thenReturn(dumpReplyNonEmpty()); + when(jvppSnat.snatAddressDump(any())).thenReturn(future(dumpReplyNat44NonEmpty())); + when(jvppSnat.nat64PoolAddrDump(any())).thenReturn(future(dumpReplyNat64NonEmpty())); - final List<ExternalIpAddressPoolKey> allIds = getCustomizer().getAllIds(externalPoolIdDefaultNatInstance, ctx); + final List<ExternalIpAddressPoolKey> allIds = getCustomizer().getAllIds(NAT_DEFAULT_POOL_WILDCARDED_ID, ctx); assertThat(allIds, hasItems( - LongStream.range(0, 2).mapToObj(ExternalIpAddressPoolKey::new).toArray(ExternalIpAddressPoolKey[]::new))); + LongStream.range(0, 5).mapToObj(ExternalIpAddressPoolKey::new) + .toArray(ExternalIpAddressPoolKey[]::new))); } @Test public void testGetAllDifferentInstance() throws Exception { - assertThat(getCustomizer().getAllIds(externalPoolIdDifferentNatInstance, ctx), empty()); + assertThat(getCustomizer().getAllIds(NAT_NON_DEFAULT_POOL_WILDCARDED_ID, ctx), empty()); } @Test public void testGetAllNoDump() throws Exception { - when(executor.executeDump(externalPoolIdDefaultNatInstance, null)).thenReturn(dumpReplyEmpty()); - assertThat(getCustomizer().getAllIds(externalPoolIdDefaultNatInstance, ctx), empty()); + when(jvppSnat.snatAddressDump(any())).thenReturn(future(dumpReplyNat44Empty())); + when(jvppSnat.nat64PoolAddrDump(any())).thenReturn(future(dumpReplyNat64Empty())); + assertThat(getCustomizer().getAllIds(NAT_DEFAULT_POOL_WILDCARDED_ID, ctx), empty()); + } + + private static InstanceIdentifier<ExternalIpAddressPool> getId(final long id) { + return NAT_CONFIG_ID.child(ExternalIpAddressPool.class, new ExternalIpAddressPoolKey(id)); } - private static SnatAddressDetailsReplyDump dumpReplyEmpty() { + private static SnatAddressDetailsReplyDump dumpReplyNat44Empty() { return new SnatAddressDetailsReplyDump(); } - private static SnatAddressDetailsReplyDump dumpReplyNonEmpty() { - SnatAddressDetailsReplyDump replyDump = dumpReplyEmpty(); + private static SnatAddressDetailsReplyDump dumpReplyNat44NonEmpty() { + SnatAddressDetailsReplyDump replyDump = dumpReplyNat44Empty(); SnatAddressDetails detailsOne = new SnatAddressDetails(); - detailsOne.ipAddress = new byte[] {-64, -88, 2, 1}; + detailsOne.ipAddress = new byte[]{-64, -88, 44, 1}; detailsOne.isIp4 = 1; SnatAddressDetails detailsTwo = new SnatAddressDetails(); - detailsTwo.ipAddress = new byte[] {-64, -88, 2, 2}; + detailsTwo.ipAddress = new byte[]{-64, -88, 44, 2}; detailsTwo.isIp4 = 1; SnatAddressDetails detailsThree = new SnatAddressDetails(); - detailsThree.ipAddress = new byte[] {-64, -88, 2, 3}; + detailsThree.ipAddress = new byte[]{-64, -88, 44, 3}; detailsThree.isIp4 = 1; replyDump.snatAddressDetails = Arrays.asList(detailsOne, detailsTwo, detailsThree); return replyDump; } + + private static Nat64PoolAddrDetailsReplyDump dumpReplyNat64Empty() { + return new Nat64PoolAddrDetailsReplyDump(); + } + + private static Nat64PoolAddrDetailsReplyDump dumpReplyNat64NonEmpty() { + Nat64PoolAddrDetailsReplyDump replyDump = dumpReplyNat64Empty(); + + Nat64PoolAddrDetails detailsOne = new Nat64PoolAddrDetails(); + detailsOne.address = new byte[]{-64, -88, 64, 1}; + + Nat64PoolAddrDetails detailsTwo = new Nat64PoolAddrDetails(); + detailsTwo.address = new byte[]{-64, -88, 64, 2}; + + Nat64PoolAddrDetails detailsThree = new Nat64PoolAddrDetails(); + detailsThree.address = new byte[]{-64, -88, 64, 3}; + + replyDump.nat64PoolAddrDetails = Arrays.asList(detailsOne, detailsTwo, detailsThree); + + return replyDump; + } }
\ No newline at end of file |