Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2016-present Open Networking Laboratory |
| 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 | */ |
Jian Li | 5e505c6 | 2016-12-05 02:44:24 +0900 | [diff] [blame] | 16 | package org.onosproject.lisp.ctl.impl; |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 17 | |
| 18 | import com.google.common.collect.ImmutableList; |
| 19 | import org.junit.Before; |
| 20 | import org.junit.Test; |
| 21 | import org.onlab.packet.IpAddress; |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 22 | import org.onosproject.lisp.msg.protocols.DefaultLispLocator.DefaultLocatorBuilder; |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 23 | import org.onosproject.lisp.msg.protocols.DefaultLispMapRecord.DefaultMapRecordBuilder; |
| 24 | import org.onosproject.lisp.msg.protocols.LispEidRecord; |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 25 | import org.onosproject.lisp.msg.protocols.LispLocator; |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 26 | import org.onosproject.lisp.msg.protocols.LispMapRecord; |
| 27 | import org.onosproject.lisp.msg.protocols.LispMapRecord.MapRecordBuilder; |
| 28 | import org.onosproject.lisp.msg.types.LispIpv4Address; |
| 29 | |
| 30 | import java.util.List; |
| 31 | |
| 32 | import static org.hamcrest.MatcherAssert.assertThat; |
| 33 | import static org.hamcrest.Matchers.is; |
| 34 | |
| 35 | /** |
| 36 | * Tests for LISP EID RLOC Map class. |
| 37 | */ |
Jian Li | 29986d8 | 2016-12-01 03:25:12 +0900 | [diff] [blame] | 38 | public class LispMappingDatabaseTest { |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 39 | |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 40 | private static final String LOCATOR_IP_1_1 = "123.1.1.1"; |
| 41 | private static final String LOCATOR_IP_1_2 = "123.1.1.2"; |
| 42 | private static final String LOCATOR_IP_1_3 = "123.1.1.3"; |
| 43 | |
| 44 | private static final String LOCATOR_IP_2_1 = "123.2.1.1"; |
| 45 | private static final String LOCATOR_IP_2_2 = "123.2.1.2"; |
| 46 | |
| 47 | private static final String LOCATOR_IP_3_1 = "123.3.1.1"; |
| 48 | |
| 49 | private static final String EID_IP_1 = "10.1.1.1"; |
| 50 | private static final String EID_IP_2 = "10.1.2.0"; |
| 51 | private static final String EID_IP_3 = "10.2.0.0"; |
| 52 | |
| 53 | private static final String EID_IP_PREFIX_1_32 = "10.2.1.1"; |
| 54 | private static final String EID_IP_PREFIX_1_24 = "10.2.1.0"; |
| 55 | private static final String EID_IP_PREFIX_1_16 = "10.2.0.0"; |
| 56 | |
| 57 | private static final String EID_IP_PREFIX_2_32 = "10.1.2.1"; |
| 58 | private static final String EID_IP_PREFIX_2_24 = "10.1.2.0"; |
| 59 | |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 60 | private final LispMappingDatabase expireMapDb = LispExpireMapDatabase.getInstance(); |
| 61 | private final LispMappingDatabase radixTreeDb = LispRadixTreeDatabase.getInstance(); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 62 | |
| 63 | @Before |
| 64 | public void setup() { |
| 65 | |
| 66 | byte cidr1 = (byte) 32; |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 67 | LispIpv4Address eid1 = new LispIpv4Address(IpAddress.valueOf(EID_IP_1)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 68 | LispEidRecord eidRecord1 = new LispEidRecord(cidr1, eid1); |
| 69 | |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 70 | LispIpv4Address locator11 = new LispIpv4Address(IpAddress.valueOf(LOCATOR_IP_1_1)); |
| 71 | LispIpv4Address locator12 = new LispIpv4Address(IpAddress.valueOf(LOCATOR_IP_1_2)); |
| 72 | LispIpv4Address locator13 = new LispIpv4Address(IpAddress.valueOf(LOCATOR_IP_1_3)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 73 | |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 74 | LispLocator locatorRecord11 = new DefaultLocatorBuilder() |
| 75 | .withLocatorAfi(locator11).build(); |
| 76 | LispLocator locatorRecord12 = new DefaultLocatorBuilder() |
| 77 | .withLocatorAfi(locator12).build(); |
| 78 | LispLocator locatorRecord13 = new DefaultLocatorBuilder() |
| 79 | .withLocatorAfi(locator13).build(); |
| 80 | List<LispLocator> locatorRecords1 = |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 81 | ImmutableList.of(locatorRecord11, locatorRecord12, locatorRecord13); |
| 82 | |
| 83 | byte cidr2 = (byte) 24; |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 84 | LispIpv4Address eid2 = new LispIpv4Address(IpAddress.valueOf(EID_IP_2)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 85 | LispEidRecord eidRecord2 = new LispEidRecord(cidr2, eid2); |
| 86 | |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 87 | LispIpv4Address locator21 = new LispIpv4Address(IpAddress.valueOf(LOCATOR_IP_2_1)); |
| 88 | LispIpv4Address locator22 = new LispIpv4Address(IpAddress.valueOf(LOCATOR_IP_2_2)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 89 | |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 90 | LispLocator locatorRecord21 = new DefaultLocatorBuilder() |
| 91 | .withLocatorAfi(locator21).build(); |
| 92 | LispLocator locatorRecord22 = new DefaultLocatorBuilder() |
| 93 | .withLocatorAfi(locator22).build(); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 94 | |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 95 | List<LispLocator> locatorRecords2 = |
| 96 | ImmutableList.of(locatorRecord21, locatorRecord22); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 97 | |
| 98 | byte cidr3 = (byte) 16; |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 99 | LispIpv4Address eid3 = new LispIpv4Address(IpAddress.valueOf(EID_IP_3)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 100 | LispEidRecord eidRecord3 = new LispEidRecord(cidr3, eid3); |
| 101 | |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 102 | LispIpv4Address locator31 = new LispIpv4Address(IpAddress.valueOf(LOCATOR_IP_3_1)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 103 | |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 104 | LispLocator locatorRecord31 = new DefaultLocatorBuilder() |
| 105 | .withLocatorAfi(locator31).build(); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 106 | |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 107 | List<LispLocator> locatorRecords3 = ImmutableList.of(locatorRecord31); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 108 | |
| 109 | MapRecordBuilder builder1 = new DefaultMapRecordBuilder(); |
| 110 | builder1.withMaskLength(cidr1); |
| 111 | builder1.withEidPrefixAfi(eid1); |
| 112 | builder1.withLocators(locatorRecords1); |
Jian Li | 29986d8 | 2016-12-01 03:25:12 +0900 | [diff] [blame] | 113 | builder1.withRecordTtl(60); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 114 | LispMapRecord mapRecord1 = builder1.build(); |
| 115 | |
| 116 | MapRecordBuilder builder2 = new DefaultMapRecordBuilder(); |
| 117 | builder2.withMaskLength(cidr2); |
| 118 | builder2.withEidPrefixAfi(eid2); |
| 119 | builder2.withLocators(locatorRecords2); |
Jian Li | 29986d8 | 2016-12-01 03:25:12 +0900 | [diff] [blame] | 120 | builder2.withRecordTtl(60); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 121 | LispMapRecord mapRecord2 = builder2.build(); |
| 122 | |
| 123 | MapRecordBuilder builder3 = new DefaultMapRecordBuilder(); |
| 124 | builder3.withMaskLength(cidr3); |
| 125 | builder3.withEidPrefixAfi(eid3); |
| 126 | builder3.withLocators(locatorRecords3); |
Jian Li | 29986d8 | 2016-12-01 03:25:12 +0900 | [diff] [blame] | 127 | builder3.withRecordTtl(60); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 128 | LispMapRecord mapRecord3 = builder3.build(); |
| 129 | |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 130 | expireMapDb.putMapRecord(eidRecord1, mapRecord1, true); |
| 131 | expireMapDb.putMapRecord(eidRecord2, mapRecord2, true); |
| 132 | expireMapDb.putMapRecord(eidRecord3, mapRecord3, true); |
| 133 | |
| 134 | radixTreeDb.putMapRecord(eidRecord1, mapRecord1, true); |
| 135 | radixTreeDb.putMapRecord(eidRecord2, mapRecord2, true); |
| 136 | radixTreeDb.putMapRecord(eidRecord3, mapRecord3, true); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 137 | } |
| 138 | |
| 139 | @Test |
| 140 | public void test32MaskRange() { |
| 141 | byte cidr32 = (byte) 32; |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 142 | LispIpv4Address eid = new LispIpv4Address(IpAddress.valueOf(EID_IP_1)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 143 | LispEidRecord record = new LispEidRecord(cidr32, eid); |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 144 | LispMapRecord mapRecord1 = expireMapDb.getMapRecordByEidRecord(record, true); |
| 145 | LispMapRecord mapRecord2 = radixTreeDb.getMapRecordByEidRecord(record, true); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 146 | |
| 147 | assertThat("Failed to fetch the RLOCs with /32 EID record", |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 148 | mapRecord1.getLocatorCount(), is(3)); |
| 149 | |
| 150 | assertThat("Failed to fetch the RLOCs with /32 EID record", |
| 151 | mapRecord2.getLocatorCount(), is(3)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 152 | } |
| 153 | |
| 154 | @Test |
| 155 | public void test24MaskRange() { |
| 156 | byte cidr32 = (byte) 32; |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 157 | LispIpv4Address eid = new LispIpv4Address(IpAddress.valueOf(EID_IP_PREFIX_2_32)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 158 | LispEidRecord record32 = new LispEidRecord(cidr32, eid); |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 159 | LispMapRecord mapRecordExpireMap32 = expireMapDb.getMapRecordByEidRecord(record32, true); |
| 160 | LispMapRecord mapRecordRadixTree32 = radixTreeDb.getMapRecordByEidRecord(record32, true); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 161 | |
| 162 | byte cidr24 = (byte) 24; |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 163 | LispIpv4Address eid24 = new LispIpv4Address(IpAddress.valueOf(EID_IP_PREFIX_2_24)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 164 | LispEidRecord record24 = new LispEidRecord(cidr24, eid24); |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 165 | LispMapRecord mapRecordExpireMap24 = expireMapDb.getMapRecordByEidRecord(record24, true); |
| 166 | LispMapRecord mapRecordRadixTree24 = radixTreeDb.getMapRecordByEidRecord(record32, true); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 167 | |
| 168 | assertThat("Failed to fetch the RLOCs with /32 EID record", |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 169 | mapRecordExpireMap32.getLocatorCount(), is(2)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 170 | assertThat("Failed to fetch the RLOCs with /24 EID record", |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 171 | mapRecordExpireMap24.getLocatorCount(), is(2)); |
| 172 | |
| 173 | assertThat("Failed to fetch the RLOCs with /32 EID record", |
| 174 | mapRecordRadixTree32.getLocatorCount(), is(2)); |
| 175 | assertThat("Failed to fetch the RLOCs with /24 EID record", |
| 176 | mapRecordRadixTree24.getLocatorCount(), is(2)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 177 | } |
| 178 | |
| 179 | @Test |
| 180 | public void test16MaskRange() { |
| 181 | byte cidr32 = (byte) 32; |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 182 | LispIpv4Address eid = new LispIpv4Address(IpAddress.valueOf(EID_IP_PREFIX_1_32)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 183 | LispEidRecord record32 = new LispEidRecord(cidr32, eid); |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 184 | LispMapRecord mapRecordExpireMap32 = expireMapDb.getMapRecordByEidRecord(record32, true); |
| 185 | LispMapRecord mapRecordRadixTree32 = radixTreeDb.getMapRecordByEidRecord(record32, true); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 186 | |
| 187 | byte cidr24 = (byte) 24; |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 188 | LispIpv4Address eid24 = new LispIpv4Address(IpAddress.valueOf(EID_IP_PREFIX_1_24)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 189 | LispEidRecord record24 = new LispEidRecord(cidr24, eid24); |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 190 | LispMapRecord mapRecordExpireMap24 = expireMapDb.getMapRecordByEidRecord(record24, true); |
| 191 | LispMapRecord mapRecordRadixTree24 = radixTreeDb.getMapRecordByEidRecord(record32, true); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 192 | |
| 193 | byte cidr16 = (byte) 16; |
Jian Li | 672ebda | 2017-02-06 20:21:04 +0900 | [diff] [blame] | 194 | LispIpv4Address eid16 = new LispIpv4Address(IpAddress.valueOf(EID_IP_PREFIX_1_16)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 195 | LispEidRecord record16 = new LispEidRecord(cidr16, eid16); |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 196 | LispMapRecord mapRecordExpireMap16 = expireMapDb.getMapRecordByEidRecord(record16, true); |
| 197 | LispMapRecord mapRecordRadixTree16 = radixTreeDb.getMapRecordByEidRecord(record16, true); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 198 | |
| 199 | assertThat("Failed to fetch the RLOCs with /32 EID record", |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 200 | mapRecordExpireMap32.getLocatorCount(), is(1)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 201 | assertThat("Failed to fetch the RLOCs with /24 EID record", |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 202 | mapRecordExpireMap24.getLocatorCount(), is(1)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 203 | assertThat("Failed to fetch the RLOCs with /16 EID record", |
Jian Li | 882e190 | 2017-06-08 04:06:01 +0900 | [diff] [blame^] | 204 | mapRecordExpireMap16.getLocatorCount(), is(1)); |
| 205 | |
| 206 | assertThat("Failed to fetch the RLOCs with /32 EID record", |
| 207 | mapRecordRadixTree32.getLocatorCount(), is(1)); |
| 208 | assertThat("Failed to fetch the RLOCs with /24 EID record", |
| 209 | mapRecordRadixTree24.getLocatorCount(), is(1)); |
| 210 | assertThat("Failed to fetch the RLOCs with /16 EID record", |
| 211 | mapRecordRadixTree16.getLocatorCount(), is(1)); |
Jian Li | cdd276b | 2016-11-15 15:53:21 +0900 | [diff] [blame] | 212 | } |
| 213 | } |