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