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