alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 1 | /* |
Brian O'Connor | a09fe5b | 2017-08-03 21:12:30 -0700 | [diff] [blame] | 2 | * Copyright 2015-present Open Networking Foundation |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 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 | */ |
Ray Milkey | ebdd485 | 2017-10-18 16:19:28 -0700 | [diff] [blame] | 16 | package org.onosproject.net.meter.impl; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 17 | |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 18 | import com.google.common.collect.ImmutableList; |
| 19 | import com.google.common.collect.ImmutableMap; |
Charles Chan | 593acf9 | 2017-11-22 13:55:41 -0800 | [diff] [blame] | 20 | import com.google.common.collect.Lists; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 21 | import com.google.common.collect.Sets; |
| 22 | import org.junit.After; |
| 23 | import org.junit.Before; |
| 24 | import org.junit.Test; |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 25 | import org.onlab.junit.TestTools; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 26 | import org.onlab.junit.TestUtils; |
| 27 | import org.onlab.packet.IpAddress; |
Charles Chan | 593acf9 | 2017-11-22 13:55:41 -0800 | [diff] [blame] | 28 | import org.onlab.util.KryoNamespace; |
| 29 | import org.onosproject.TestApplicationId; |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 30 | import org.onosproject.cfg.ComponentConfigAdapter; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 31 | import org.onosproject.cluster.ClusterServiceAdapter; |
| 32 | import org.onosproject.cluster.ControllerNode; |
| 33 | import org.onosproject.cluster.DefaultControllerNode; |
| 34 | import org.onosproject.cluster.NodeId; |
| 35 | import org.onosproject.common.event.impl.TestEventDispatcher; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 36 | import org.onosproject.mastership.MastershipServiceAdapter; |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 37 | import org.onosproject.net.AnnotationKeys; |
| 38 | import org.onosproject.net.DefaultAnnotations; |
| 39 | import org.onosproject.net.DefaultDevice; |
| 40 | import org.onosproject.net.Device; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 41 | import org.onosproject.net.DeviceId; |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 42 | import org.onosproject.net.MastershipRole; |
| 43 | import org.onosproject.net.behaviour.MeterQuery; |
Thomas Vachuska | 164ecf6 | 2018-05-08 17:29:55 -0700 | [diff] [blame] | 44 | import org.onosproject.net.config.NetworkConfigService; |
| 45 | import org.onosproject.net.config.NetworkConfigServiceAdapter; |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 46 | import org.onosproject.net.device.DeviceService; |
| 47 | import org.onosproject.net.device.DeviceServiceAdapter; |
| 48 | import org.onosproject.net.driver.AbstractHandlerBehaviour; |
| 49 | import org.onosproject.net.driver.DefaultDriver; |
| 50 | import org.onosproject.net.driver.DriverRegistry; |
| 51 | import org.onosproject.net.driver.impl.DriverManager; |
| 52 | import org.onosproject.net.driver.impl.DriverRegistryManager; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 53 | import org.onosproject.net.meter.Band; |
| 54 | import org.onosproject.net.meter.DefaultBand; |
| 55 | import org.onosproject.net.meter.DefaultMeter; |
Jordi Ortiz | aa8de49 | 2016-12-01 00:21:36 +0100 | [diff] [blame] | 56 | import org.onosproject.net.meter.DefaultMeterFeatures; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 57 | import org.onosproject.net.meter.DefaultMeterRequest; |
| 58 | import org.onosproject.net.meter.Meter; |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 59 | import org.onosproject.net.meter.MeterFeatures; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 60 | import org.onosproject.net.meter.MeterId; |
| 61 | import org.onosproject.net.meter.MeterOperation; |
| 62 | import org.onosproject.net.meter.MeterOperations; |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 63 | import org.onosproject.net.meter.MeterProgrammable; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 64 | import org.onosproject.net.meter.MeterProvider; |
| 65 | import org.onosproject.net.meter.MeterProviderRegistry; |
| 66 | import org.onosproject.net.meter.MeterProviderService; |
| 67 | import org.onosproject.net.meter.MeterRequest; |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 68 | import org.onosproject.net.meter.MeterService; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 69 | import org.onosproject.net.meter.MeterState; |
Carmelo Cascone | 0761cd3 | 2018-08-29 19:22:50 -0700 | [diff] [blame] | 70 | import org.onosproject.net.pi.PiPipeconfServiceAdapter; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 71 | import org.onosproject.net.provider.AbstractProvider; |
| 72 | import org.onosproject.net.provider.ProviderId; |
Thomas Vachuska | 52f2cd1 | 2018-11-08 21:20:04 -0800 | [diff] [blame] | 73 | import org.onosproject.store.meter.impl.DistributedMeterStore; |
Charles Chan | 593acf9 | 2017-11-22 13:55:41 -0800 | [diff] [blame] | 74 | import org.onosproject.store.service.Serializer; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 75 | import org.onosproject.store.service.TestStorageService; |
| 76 | |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 77 | import java.util.ArrayList; |
| 78 | import java.util.Collection; |
Carmelo Cascone | 0761cd3 | 2018-08-29 19:22:50 -0700 | [diff] [blame] | 79 | import java.util.Collections; |
| 80 | import java.util.HashSet; |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 81 | import java.util.List; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 82 | import java.util.Set; |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 83 | import java.util.concurrent.CompletableFuture; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 84 | |
| 85 | import static org.hamcrest.Matchers.is; |
Carmelo Cascone | 0761cd3 | 2018-08-29 19:22:50 -0700 | [diff] [blame] | 86 | import static org.junit.Assert.assertEquals; |
| 87 | import static org.junit.Assert.assertFalse; |
| 88 | import static org.junit.Assert.assertNull; |
| 89 | import static org.junit.Assert.assertThat; |
| 90 | import static org.junit.Assert.assertTrue; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 91 | import static org.onosproject.net.NetTestTools.APP_ID; |
| 92 | import static org.onosproject.net.NetTestTools.did; |
| 93 | import static org.onosproject.net.NetTestTools.injectEventDispatcher; |
| 94 | |
| 95 | /** |
| 96 | * Meter manager tests. |
| 97 | */ |
| 98 | public class MeterManagerTest { |
| 99 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 100 | // Test node id |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 101 | private static final NodeId NID_LOCAL = new NodeId("local"); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 102 | |
| 103 | // Test ip address |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 104 | private static final IpAddress LOCALHOST = IpAddress.valueOf("127.0.0.1"); |
| 105 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 106 | private static final ProviderId PID = new ProviderId("of", "foo"); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 107 | |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 108 | private static final ProviderId PROGRAMMABLE_PROVIDER = new ProviderId("foo", "foo"); |
| 109 | private static final DeviceId PROGRAMMABLE_DID = DeviceId.deviceId("test:002"); |
| 110 | |
| 111 | private static final DefaultAnnotations ANNOTATIONS = |
| 112 | DefaultAnnotations.builder().set(AnnotationKeys.DRIVER, "foo").build(); |
| 113 | |
| 114 | private static final Device PROGRAMMABLE_DEV = |
| 115 | new DefaultDevice(PROGRAMMABLE_PROVIDER, PROGRAMMABLE_DID, Device.Type.SWITCH, |
| 116 | "", "", "", "", null, ANNOTATIONS); |
| 117 | |
| 118 | |
| 119 | private MeterService service; |
| 120 | |
| 121 | // Test Driver service used during the tests |
| 122 | private DriverManager driverService; |
| 123 | |
| 124 | // Test device service used during the tests |
| 125 | private DeviceService deviceService; |
| 126 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 127 | // Test provider used during the tests |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 128 | private TestProvider provider; |
| 129 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 130 | // Meter manager |
| 131 | private MeterManager manager; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 132 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 133 | // Meter provider registry |
| 134 | private MeterProviderRegistry registry; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 135 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 136 | // Meter provider service |
| 137 | private MeterProviderService providerService; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 138 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 139 | // Store under testing |
| 140 | private DistributedMeterStore meterStore; |
| 141 | |
| 142 | // Device ids used during the tests |
| 143 | private DeviceId did1 = did("1"); |
| 144 | private DeviceId did2 = did("2"); |
| 145 | |
| 146 | // Meter ids used during the tests |
| 147 | private MeterId mid1 = MeterId.meterId(1); |
| 148 | |
| 149 | // Bands used during the tests |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 150 | private static Band b1 = DefaultBand.builder() |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 151 | .ofType(Band.Type.DROP) |
| 152 | .withRate(500) |
| 153 | .build(); |
| 154 | |
| 155 | // Meters used during the tests |
| 156 | private Meter m1 = DefaultMeter.builder() |
| 157 | .forDevice(did1) |
| 158 | .fromApp(APP_ID) |
| 159 | .withId(mid1) |
| 160 | .withUnit(Meter.Unit.KB_PER_SEC) |
| 161 | .withBands(Collections.singletonList(b1)) |
| 162 | .build(); |
| 163 | private Meter m2 = DefaultMeter.builder() |
| 164 | .forDevice(did2) |
| 165 | .fromApp(APP_ID) |
| 166 | .withId(mid1) |
| 167 | .withUnit(Meter.Unit.KB_PER_SEC) |
| 168 | .withBands(Collections.singletonList(b1)) |
| 169 | .build(); |
| 170 | |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 171 | private static Meter mProgrammable = DefaultMeter.builder() |
| 172 | .forDevice(PROGRAMMABLE_DID) |
| 173 | .fromApp(APP_ID) |
| 174 | .withId(MeterId.meterId(1)) |
| 175 | .withUnit(Meter.Unit.KB_PER_SEC) |
| 176 | .withBands(Collections.singletonList(b1)) |
| 177 | .build(); |
| 178 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 179 | // Meter requests used during the tests |
| 180 | private MeterRequest.Builder m1Request = DefaultMeterRequest.builder() |
| 181 | .forDevice(did1) |
| 182 | .fromApp(APP_ID) |
| 183 | .withUnit(Meter.Unit.KB_PER_SEC) |
| 184 | .withBands(Collections.singletonList(b1)); |
| 185 | private MeterRequest.Builder m2Request = DefaultMeterRequest.builder() |
| 186 | .forDevice(did2) |
| 187 | .fromApp(APP_ID) |
| 188 | .withUnit(Meter.Unit.KB_PER_SEC) |
| 189 | .withBands(Collections.singletonList(b1)); |
| 190 | |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 191 | private MeterRequest.Builder mProgrammableRequest = DefaultMeterRequest.builder() |
| 192 | .forDevice(PROGRAMMABLE_DID) |
| 193 | .fromApp(APP_ID) |
| 194 | .withUnit(Meter.Unit.KB_PER_SEC) |
| 195 | .withBands(Collections.singletonList(b1)); |
| 196 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 197 | // Meter features used during the tests |
| 198 | private MeterFeatures mef1 = DefaultMeterFeatures.builder().forDevice(did1) |
| 199 | .withMaxMeters(3L) |
| 200 | .withBandTypes(new HashSet<>()) |
| 201 | .withUnits(new HashSet<>()) |
| 202 | .hasStats(false) |
| 203 | .hasBurst(false) |
| 204 | .withMaxBands((byte) 0) |
| 205 | .withMaxColors((byte) 0) |
| 206 | .build(); |
| 207 | private MeterFeatures mef2 = DefaultMeterFeatures.builder().forDevice(did2) |
| 208 | .withMaxMeters(10L) |
| 209 | .withBandTypes(new HashSet<>()) |
| 210 | .withUnits(new HashSet<>()) |
| 211 | .hasStats(false) |
| 212 | .hasBurst(false) |
| 213 | .withMaxBands((byte) 0) |
| 214 | .withMaxColors((byte) 0) |
| 215 | .build(); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 216 | |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 217 | |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 218 | @Before |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 219 | public void setup() { |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 220 | //Init step for the deviceService |
| 221 | deviceService = new TestDeviceService(); |
| 222 | //Init step for the driver registry and driver service. |
| 223 | DriverRegistryManager driverRegistry = new DriverRegistryManager(); |
Thomas Vachuska | 164ecf6 | 2018-05-08 17:29:55 -0700 | [diff] [blame] | 224 | driverService = new TestDriverManager(driverRegistry, deviceService, new NetworkConfigServiceAdapter()); |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 225 | driverRegistry.addDriver(new DefaultDriver("foo", ImmutableList.of(), "", |
| 226 | "", "", |
| 227 | ImmutableMap.of(MeterProgrammable.class, |
| 228 | TestMeterProgrammable.class, MeterQuery.class, TestMeterQuery.class), |
| 229 | ImmutableMap.of())); |
| 230 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 231 | // Init step for the store |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 232 | meterStore = new DistributedMeterStore(); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 233 | // Let's initialize some internal services of the store |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 234 | TestUtils.setField(meterStore, "storageService", new TestStorageService()); |
| 235 | TestUtils.setField(meterStore, "clusterService", new TestClusterService()); |
| 236 | TestUtils.setField(meterStore, "mastershipService", new TestMastershipService()); |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 237 | TestUtils.setField(meterStore, "driverService", driverService); |
Charles Chan | 593acf9 | 2017-11-22 13:55:41 -0800 | [diff] [blame] | 238 | |
| 239 | // Inject TestApplicationId into the DistributedMeterStore serializer |
| 240 | KryoNamespace.Builder testKryoBuilder = TestUtils.getField(meterStore, "APP_KRYO_BUILDER"); |
| 241 | testKryoBuilder.register(TestApplicationId.class); |
| 242 | Serializer testSerializer = Serializer.using(Lists.newArrayList(testKryoBuilder.build())); |
| 243 | TestUtils.setField(meterStore, "serializer", testSerializer); |
| 244 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 245 | // Activate the store |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 246 | meterStore.activate(); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 247 | // Init step for the manager |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 248 | manager = new MeterManager(); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 249 | // Let's initialize some internal services of the manager |
| 250 | TestUtils.setField(manager, "store", meterStore); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 251 | injectEventDispatcher(manager, new TestEventDispatcher()); |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 252 | manager.deviceService = deviceService; |
| 253 | manager.mastershipService = new TestMastershipService(); |
| 254 | manager.cfgService = new ComponentConfigAdapter(); |
| 255 | TestUtils.setField(manager, "storageService", new TestStorageService()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 256 | // Init the reference of the registry |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 257 | registry = manager; |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 258 | |
| 259 | manager.driverService = driverService; |
| 260 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 261 | // Activate the manager |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 262 | manager.activate(null); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 263 | // Initialize the test provider |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 264 | |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 265 | provider = new TestProvider(PID); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 266 | // Register the provider against the manager |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 267 | providerService = registry.register(provider); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 268 | // Verify register |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 269 | assertTrue("provider should be registered", |
| 270 | registry.getProviders().contains(provider.id())); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 271 | } |
| 272 | |
| 273 | @After |
| 274 | public void tearDown() { |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 275 | // Unregister provider |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 276 | registry.unregister(provider); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 277 | // Verify unregister |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 278 | assertFalse("provider should not be registered", |
| 279 | registry.getProviders().contains(provider.id())); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 280 | // Deactivate manager |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 281 | manager.deactivate(); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 282 | // Remove event dispatcher |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 283 | injectEventDispatcher(manager, null); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 284 | // Deactivate store |
| 285 | meterStore.deactivate(); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 286 | } |
| 287 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 288 | private void initMeterStore() { |
| 289 | // Let's store feature for device 1 |
| 290 | meterStore.storeMeterFeatures(mef1); |
| 291 | // Let's store feature for device 2 |
| 292 | meterStore.storeMeterFeatures(mef2); |
| 293 | } |
| 294 | |
| 295 | // Emulate metrics coming from the dataplane |
| 296 | private void pushMetrics(MeterOperation.Type type, Meter meter) { |
| 297 | // If it is an add operation |
| 298 | if (type == MeterOperation.Type.ADD) { |
| 299 | // Update state to added |
| 300 | ((DefaultMeter) meter).setState(MeterState.ADDED); |
| 301 | // Push the update in the store |
| 302 | providerService.pushMeterMetrics(meter.deviceId(), Collections.singletonList(meter)); |
| 303 | } else { |
| 304 | providerService.pushMeterMetrics(meter.deviceId(), Collections.emptyList()); |
| 305 | } |
| 306 | } |
| 307 | |
| 308 | /** |
| 309 | * Test add meter. |
| 310 | */ |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 311 | @Test |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 312 | public void testAdd() { |
| 313 | // Init store |
| 314 | initMeterStore(); |
| 315 | // Submit meter request |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 316 | manager.submit(m1Request.add()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 317 | // Verify add |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 318 | assertTrue("The meter was not added", manager.getAllMeters().size() == 1); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 319 | assertTrue("The meter was not added", manager.getMeters(did1).size() == 1); |
| 320 | // Get Meter |
| 321 | Meter installingMeter = manager.getMeter(did1, mid1); |
| 322 | // Verify add of installingMeter and pending add state |
| 323 | assertThat(installingMeter, is(m1)); |
| 324 | // Verify pending add state |
| 325 | assertThat(installingMeter.state(), is(MeterState.PENDING_ADD)); |
| 326 | // Let's simulate a working data-plane |
| 327 | pushMetrics(MeterOperation.Type.ADD, installingMeter); |
| 328 | // Get meter |
| 329 | Meter installedMeter = manager.getMeter(did1, mid1); |
| 330 | // Verify installation |
| 331 | assertThat(installedMeter.state(), is(MeterState.ADDED)); |
| 332 | assertTrue("The meter was not installed", manager.getAllMeters().size() == 1); |
| 333 | assertTrue("The meter was not installed", manager.getMeters(did1).size() == 1); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 334 | } |
| 335 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 336 | /** |
| 337 | * Test remove meter. |
| 338 | */ |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 339 | @Test |
| 340 | public void testRemove() { |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 341 | // Init store |
| 342 | initMeterStore(); |
| 343 | // Submit meter request |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 344 | manager.submit(m1Request.add()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 345 | // Withdraw meter |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 346 | manager.withdraw(m1Request.remove(), m1.id()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 347 | // Get Meter |
| 348 | Meter withdrawingMeter = manager.getMeter(did1, mid1); |
| 349 | // Verify withdrawing |
| 350 | assertThat(withdrawingMeter.state(), is(MeterState.PENDING_REMOVE)); |
| 351 | assertTrue("The meter was not withdrawn", manager.getAllMeters().size() == 1); |
| 352 | assertTrue("The meter was not withdrawn", manager.getMeters(did1).size() == 1); |
| 353 | // Let's simulate a working data-plane |
| 354 | pushMetrics(MeterOperation.Type.REMOVE, withdrawingMeter); |
| 355 | // Verify withdrawn |
| 356 | assertNull(manager.getMeter(did1, mid1)); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 357 | assertTrue("The meter was not removed", manager.getAllMeters().size() == 0); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 358 | assertTrue("The meter was not removed", manager.getMeters(did1).size() == 0); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 359 | } |
| 360 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 361 | /** |
| 362 | * Test add multiple device. |
| 363 | */ |
alshabib | 70aaa1b | 2015-09-25 14:30:59 -0700 | [diff] [blame] | 364 | @Test |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 365 | public void testAddMultipleDevice() { |
| 366 | // Init store |
| 367 | initMeterStore(); |
| 368 | // Submit meter 1 |
alshabib | 70aaa1b | 2015-09-25 14:30:59 -0700 | [diff] [blame] | 369 | manager.submit(m1Request.add()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 370 | // Submit meter 2 |
alshabib | 70aaa1b | 2015-09-25 14:30:59 -0700 | [diff] [blame] | 371 | manager.submit(m2Request.add()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 372 | // Verify add |
| 373 | assertTrue("The meter was not added", manager.getAllMeters().size() == 2); |
| 374 | assertTrue("The meter was not added", manager.getMeters(did1).size() == 1); |
| 375 | assertTrue("The meter was not added", manager.getMeters(did2).size() == 1); |
| 376 | // Get Meters |
| 377 | Meter installingMeter1 = manager.getMeter(did1, mid1); |
| 378 | Meter installingMeter2 = manager.getMeter(did2, mid1); |
| 379 | // Verify add of installingMeter |
| 380 | assertThat(installingMeter1, is(m1)); |
| 381 | assertThat(installingMeter2, is(m2)); |
| 382 | // Verify pending add state |
| 383 | assertThat(installingMeter1.state(), is(MeterState.PENDING_ADD)); |
| 384 | assertThat(installingMeter2.state(), is(MeterState.PENDING_ADD)); |
| 385 | // Let's simulate a working data-plane |
| 386 | pushMetrics(MeterOperation.Type.ADD, installingMeter1); |
| 387 | pushMetrics(MeterOperation.Type.ADD, installingMeter2); |
| 388 | // Get meter |
| 389 | Meter installedMeter1 = manager.getMeter(did1, mid1); |
| 390 | Meter installedMeter2 = manager.getMeter(did2, mid1); |
| 391 | // Verify installation |
| 392 | assertThat(installedMeter1.state(), is(MeterState.ADDED)); |
| 393 | assertThat(installedMeter2.state(), is(MeterState.ADDED)); |
| 394 | assertTrue("The meter was not installed", manager.getAllMeters().size() == 2); |
| 395 | assertTrue("The meter was not installed", manager.getMeters(did1).size() == 1); |
| 396 | assertTrue("The meter was not installed", manager.getMeters(did2).size() == 1); |
alshabib | 70aaa1b | 2015-09-25 14:30:59 -0700 | [diff] [blame] | 397 | } |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 398 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 399 | /** |
| 400 | * Test remove meter. |
| 401 | */ |
Jordi Ortiz | aa8de49 | 2016-12-01 00:21:36 +0100 | [diff] [blame] | 402 | @Test |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 403 | public void testRemoveMultipleDevice() { |
| 404 | // Init store |
| 405 | initMeterStore(); |
| 406 | // Submit meter 1 |
Jordi Ortiz | 6c84776 | 2017-01-30 17:13:05 +0100 | [diff] [blame] | 407 | manager.submit(m1Request.add()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 408 | // Submit meter 2 |
| 409 | manager.submit(m2Request.add()); |
| 410 | // Withdraw meter |
| 411 | manager.withdraw(m1Request.remove(), m1.id()); |
| 412 | // Withdraw meter |
| 413 | manager.withdraw(m2Request.remove(), m2.id()); |
| 414 | // Get Meters |
| 415 | Meter withdrawingMeter1 = manager.getMeter(did1, mid1); |
| 416 | Meter withdrawingMeter2 = manager.getMeter(did2, mid1); |
| 417 | // Verify withdrawing |
| 418 | assertThat(withdrawingMeter1.state(), is(MeterState.PENDING_REMOVE)); |
| 419 | assertThat(withdrawingMeter2.state(), is(MeterState.PENDING_REMOVE)); |
| 420 | assertTrue("The meter was not withdrawn", manager.getAllMeters().size() == 2); |
| 421 | assertTrue("The meter was not withdrawn", manager.getMeters(did1).size() == 1); |
| 422 | assertTrue("The meter was not withdrawn", manager.getMeters(did2).size() == 1); |
| 423 | // Let's simulate a working data-plane |
| 424 | pushMetrics(MeterOperation.Type.REMOVE, withdrawingMeter1); |
| 425 | pushMetrics(MeterOperation.Type.REMOVE, withdrawingMeter2); |
| 426 | // Verify withdrawn |
| 427 | assertNull(manager.getMeter(did1, mid1)); |
| 428 | assertNull(manager.getMeter(did2, mid1)); |
| 429 | assertTrue("The meter was not removed", manager.getAllMeters().size() == 0); |
| 430 | assertTrue("The meter was not removed", manager.getMeters(did1).size() == 0); |
| 431 | assertTrue("The meter was not removed", manager.getMeters(did2).size() == 0); |
Jordi Ortiz | 6c84776 | 2017-01-30 17:13:05 +0100 | [diff] [blame] | 432 | } |
| 433 | |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 434 | @Test |
| 435 | public void testAddFromMeterProgrammable() { |
| 436 | |
| 437 | // Init store |
| 438 | initMeterStore(); |
| 439 | |
| 440 | manager.submit(mProgrammableRequest.add()); |
| 441 | |
| 442 | TestTools.assertAfter(500, () -> { |
| 443 | |
| 444 | assertTrue("The meter was not added", manager.getAllMeters().size() == 1); |
| 445 | |
| 446 | assertThat(manager.getMeter(PROGRAMMABLE_DID, MeterId.meterId(1)), is(mProgrammable)); |
| 447 | }); |
| 448 | |
| 449 | } |
| 450 | |
| 451 | @Test |
| 452 | public void testAddBatchFromMeterProgrammable() { |
| 453 | |
| 454 | // Init store |
| 455 | initMeterStore(); |
| 456 | |
| 457 | List<MeterOperation> operations = ImmutableList.of(new MeterOperation(mProgrammable, MeterOperation.Type.ADD)); |
| 458 | manager.defaultProvider().performMeterOperation(PROGRAMMABLE_DID, new MeterOperations(operations)); |
| 459 | |
| 460 | TestTools.assertAfter(500, () -> { |
| 461 | |
| 462 | assertTrue("The meter was not added", meterOperations.size() == 1); |
| 463 | |
| 464 | assertTrue("Wrong Meter Operation", meterOperations.get(0).meter().id().equals(mProgrammable.id())); |
| 465 | }); |
| 466 | |
| 467 | } |
| 468 | |
| 469 | @Test |
| 470 | public void testGetFromMeterProgrammable() { |
| 471 | |
| 472 | // Init store |
| 473 | initMeterStore(); |
| 474 | |
| 475 | MeterDriverProvider fallback = (MeterDriverProvider) manager.defaultProvider(); |
| 476 | |
| 477 | testAddFromMeterProgrammable(); |
| 478 | |
| 479 | fallback.init(manager.deviceService, fallback.meterProviderService, manager.mastershipService, 1); |
| 480 | |
| 481 | TestTools.assertAfter(2000, () -> { |
| 482 | assertTrue("The meter was not added", manager.getAllMeters().size() == 1); |
| 483 | Meter m = manager.getMeters(PROGRAMMABLE_DID).iterator().next(); |
| 484 | assertEquals("incorrect state", MeterState.ADDED, m.state()); |
| 485 | }); |
| 486 | |
| 487 | } |
| 488 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 489 | // Test cluster service |
| 490 | private final class TestClusterService extends ClusterServiceAdapter { |
Jordi Ortiz | aa8de49 | 2016-12-01 00:21:36 +0100 | [diff] [blame] | 491 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 492 | private ControllerNode local = new DefaultControllerNode(NID_LOCAL, LOCALHOST); |
Jordi Ortiz | aa8de49 | 2016-12-01 00:21:36 +0100 | [diff] [blame] | 493 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 494 | @Override |
| 495 | public ControllerNode getLocalNode() { |
| 496 | return local; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 497 | } |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 498 | |
| 499 | @Override |
| 500 | public Set<ControllerNode> getNodes() { |
| 501 | return Sets.newHashSet(); |
| 502 | } |
| 503 | |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 504 | } |
| 505 | |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 506 | private static class TestDeviceService extends DeviceServiceAdapter { |
| 507 | @Override |
| 508 | public int getDeviceCount() { |
| 509 | return 1; |
| 510 | } |
| 511 | |
| 512 | @Override |
| 513 | public Iterable<Device> getDevices() { |
| 514 | return ImmutableList.of(PROGRAMMABLE_DEV); |
| 515 | } |
| 516 | |
| 517 | @Override |
| 518 | public Iterable<Device> getAvailableDevices() { |
| 519 | return getDevices(); |
| 520 | } |
| 521 | |
| 522 | @Override |
| 523 | public Device getDevice(DeviceId deviceId) { |
| 524 | return PROGRAMMABLE_DEV; |
| 525 | } |
| 526 | } |
| 527 | |
| 528 | private class TestDriverManager extends DriverManager { |
Thomas Vachuska | 164ecf6 | 2018-05-08 17:29:55 -0700 | [diff] [blame] | 529 | TestDriverManager(DriverRegistry registry, DeviceService deviceService, |
| 530 | NetworkConfigService networkConfigService) { |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 531 | this.registry = registry; |
| 532 | this.deviceService = deviceService; |
Thomas Vachuska | 164ecf6 | 2018-05-08 17:29:55 -0700 | [diff] [blame] | 533 | this.networkConfigService = networkConfigService; |
Carmelo Cascone | 0761cd3 | 2018-08-29 19:22:50 -0700 | [diff] [blame] | 534 | this.pipeconfService = new PiPipeconfServiceAdapter(); |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 535 | activate(); |
| 536 | } |
| 537 | } |
| 538 | |
| 539 | public static class TestMeterQuery extends AbstractHandlerBehaviour |
| 540 | implements MeterQuery { |
| 541 | private static final long MAX_METER = 0x00000FFF; |
| 542 | |
| 543 | @Override |
| 544 | public long getMaxMeters() { |
| 545 | return MAX_METER; |
| 546 | } |
| 547 | } |
| 548 | |
| 549 | private static List<MeterOperation> meterOperations = new ArrayList<>(); |
| 550 | |
| 551 | public static class TestMeterProgrammable extends AbstractHandlerBehaviour |
| 552 | implements MeterProgrammable { |
| 553 | |
| 554 | @Override |
| 555 | public CompletableFuture<Boolean> performMeterOperation(MeterOperation meterOp) { |
| 556 | return CompletableFuture.completedFuture(meterOperations.add(meterOp)); |
| 557 | } |
| 558 | |
| 559 | @Override |
| 560 | public CompletableFuture<Collection<Meter>> getMeters() { |
| 561 | //ADD METER |
| 562 | DefaultMeter mProgrammableAdded = (DefaultMeter) mProgrammable; |
| 563 | mProgrammableAdded.setState(MeterState.ADDED); |
| 564 | return CompletableFuture.completedFuture(ImmutableList.of(mProgrammableAdded)); |
| 565 | } |
| 566 | } |
| 567 | |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 568 | private class TestProvider extends AbstractProvider implements MeterProvider { |
| 569 | |
| 570 | protected TestProvider(ProviderId id) { |
| 571 | super(PID); |
| 572 | } |
| 573 | |
| 574 | @Override |
| 575 | public void performMeterOperation(DeviceId deviceId, MeterOperations meterOps) { |
| 576 | //Currently unused. |
| 577 | } |
| 578 | |
| 579 | @Override |
| 580 | public void performMeterOperation(DeviceId deviceId, MeterOperation meterOp) { |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 581 | //Currently unused |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 582 | } |
| 583 | } |
| 584 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 585 | // Test mastership service |
| 586 | private final class TestMastershipService extends MastershipServiceAdapter { |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 587 | @Override |
| 588 | public NodeId getMasterFor(DeviceId deviceId) { |
| 589 | return NID_LOCAL; |
| 590 | } |
Andrea Campanella | e370878 | 2017-10-16 16:00:21 +0200 | [diff] [blame] | 591 | |
| 592 | @Override |
| 593 | public MastershipRole getLocalRole(DeviceId deviceId) { |
| 594 | return MastershipRole.MASTER; |
| 595 | } |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 596 | } |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame] | 597 | |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 598 | } |