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 | |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 18 | import com.google.common.collect.Sets; |
| 19 | import org.junit.After; |
| 20 | import org.junit.Before; |
| 21 | import org.junit.Test; |
| 22 | import org.onlab.junit.TestUtils; |
| 23 | import org.onlab.packet.IpAddress; |
| 24 | import org.onosproject.cluster.ClusterServiceAdapter; |
| 25 | import org.onosproject.cluster.ControllerNode; |
| 26 | import org.onosproject.cluster.DefaultControllerNode; |
| 27 | import org.onosproject.cluster.NodeId; |
| 28 | import org.onosproject.common.event.impl.TestEventDispatcher; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 29 | import org.onosproject.incubator.store.meter.impl.DistributedMeterStore; |
| 30 | import org.onosproject.mastership.MastershipServiceAdapter; |
| 31 | import org.onosproject.net.DeviceId; |
| 32 | import org.onosproject.net.meter.Band; |
| 33 | import org.onosproject.net.meter.DefaultBand; |
| 34 | import org.onosproject.net.meter.DefaultMeter; |
Jordi Ortiz | aa8de49 | 2016-12-01 00:21:36 +0100 | [diff] [blame] | 35 | import org.onosproject.net.meter.DefaultMeterFeatures; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 36 | import org.onosproject.net.meter.DefaultMeterRequest; |
| 37 | import org.onosproject.net.meter.Meter; |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 38 | import org.onosproject.net.meter.MeterFeatures; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 39 | import org.onosproject.net.meter.MeterId; |
| 40 | import org.onosproject.net.meter.MeterOperation; |
| 41 | import org.onosproject.net.meter.MeterOperations; |
| 42 | import org.onosproject.net.meter.MeterProvider; |
| 43 | import org.onosproject.net.meter.MeterProviderRegistry; |
| 44 | import org.onosproject.net.meter.MeterProviderService; |
| 45 | import org.onosproject.net.meter.MeterRequest; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 46 | import org.onosproject.net.meter.MeterState; |
| 47 | import org.onosproject.net.provider.AbstractProvider; |
| 48 | import org.onosproject.net.provider.ProviderId; |
| 49 | import org.onosproject.store.service.TestStorageService; |
| 50 | |
| 51 | import java.util.Collections; |
Jordi Ortiz | aa8de49 | 2016-12-01 00:21:36 +0100 | [diff] [blame] | 52 | import java.util.HashSet; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 53 | import java.util.Set; |
| 54 | |
| 55 | import static org.hamcrest.Matchers.is; |
Jordi Ortiz | aa8de49 | 2016-12-01 00:21:36 +0100 | [diff] [blame] | 56 | import static org.junit.Assert.*; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 57 | import static org.onosproject.net.NetTestTools.APP_ID; |
| 58 | import static org.onosproject.net.NetTestTools.did; |
| 59 | import static org.onosproject.net.NetTestTools.injectEventDispatcher; |
| 60 | |
| 61 | /** |
| 62 | * Meter manager tests. |
| 63 | */ |
| 64 | public class MeterManagerTest { |
| 65 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 66 | // Test node id |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 67 | private static final NodeId NID_LOCAL = new NodeId("local"); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 68 | |
| 69 | // Test ip address |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 70 | private static final IpAddress LOCALHOST = IpAddress.valueOf("127.0.0.1"); |
| 71 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 72 | // Provider id used during the tests |
| 73 | private static final ProviderId PID = new ProviderId("of", "foo"); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 74 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 75 | // Test provider used during the tests |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 76 | private TestProvider provider; |
| 77 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 78 | // Meter manager |
| 79 | private MeterManager manager; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 80 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 81 | // Meter provider registry |
| 82 | private MeterProviderRegistry registry; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 83 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 84 | // Meter provider service |
| 85 | private MeterProviderService providerService; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 86 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 87 | // Store under testing |
| 88 | private DistributedMeterStore meterStore; |
| 89 | |
| 90 | // Device ids used during the tests |
| 91 | private DeviceId did1 = did("1"); |
| 92 | private DeviceId did2 = did("2"); |
| 93 | |
| 94 | // Meter ids used during the tests |
| 95 | private MeterId mid1 = MeterId.meterId(1); |
| 96 | |
| 97 | // Bands used during the tests |
| 98 | private Band b1 = DefaultBand.builder() |
| 99 | .ofType(Band.Type.DROP) |
| 100 | .withRate(500) |
| 101 | .build(); |
| 102 | |
| 103 | // Meters used during the tests |
| 104 | private Meter m1 = DefaultMeter.builder() |
| 105 | .forDevice(did1) |
| 106 | .fromApp(APP_ID) |
| 107 | .withId(mid1) |
| 108 | .withUnit(Meter.Unit.KB_PER_SEC) |
| 109 | .withBands(Collections.singletonList(b1)) |
| 110 | .build(); |
| 111 | private Meter m2 = DefaultMeter.builder() |
| 112 | .forDevice(did2) |
| 113 | .fromApp(APP_ID) |
| 114 | .withId(mid1) |
| 115 | .withUnit(Meter.Unit.KB_PER_SEC) |
| 116 | .withBands(Collections.singletonList(b1)) |
| 117 | .build(); |
| 118 | |
| 119 | // Meter requests used during the tests |
| 120 | private MeterRequest.Builder m1Request = DefaultMeterRequest.builder() |
| 121 | .forDevice(did1) |
| 122 | .fromApp(APP_ID) |
| 123 | .withUnit(Meter.Unit.KB_PER_SEC) |
| 124 | .withBands(Collections.singletonList(b1)); |
| 125 | private MeterRequest.Builder m2Request = DefaultMeterRequest.builder() |
| 126 | .forDevice(did2) |
| 127 | .fromApp(APP_ID) |
| 128 | .withUnit(Meter.Unit.KB_PER_SEC) |
| 129 | .withBands(Collections.singletonList(b1)); |
| 130 | |
| 131 | // Meter features used during the tests |
| 132 | private MeterFeatures mef1 = DefaultMeterFeatures.builder().forDevice(did1) |
| 133 | .withMaxMeters(3L) |
| 134 | .withBandTypes(new HashSet<>()) |
| 135 | .withUnits(new HashSet<>()) |
| 136 | .hasStats(false) |
| 137 | .hasBurst(false) |
| 138 | .withMaxBands((byte) 0) |
| 139 | .withMaxColors((byte) 0) |
| 140 | .build(); |
| 141 | private MeterFeatures mef2 = DefaultMeterFeatures.builder().forDevice(did2) |
| 142 | .withMaxMeters(10L) |
| 143 | .withBandTypes(new HashSet<>()) |
| 144 | .withUnits(new HashSet<>()) |
| 145 | .hasStats(false) |
| 146 | .hasBurst(false) |
| 147 | .withMaxBands((byte) 0) |
| 148 | .withMaxColors((byte) 0) |
| 149 | .build(); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 150 | |
| 151 | @Before |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 152 | public void setup() { |
| 153 | // Init step for the store |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 154 | meterStore = new DistributedMeterStore(); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 155 | // Let's initialize some internal services of the store |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 156 | TestUtils.setField(meterStore, "storageService", new TestStorageService()); |
| 157 | TestUtils.setField(meterStore, "clusterService", new TestClusterService()); |
| 158 | TestUtils.setField(meterStore, "mastershipService", new TestMastershipService()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 159 | // Activate the store |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 160 | meterStore.activate(); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 161 | // Init step for the manager |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 162 | manager = new MeterManager(); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 163 | // Let's initialize some internal services of the manager |
| 164 | TestUtils.setField(manager, "store", meterStore); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 165 | injectEventDispatcher(manager, new TestEventDispatcher()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 166 | // Init the reference of the registry |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 167 | registry = manager; |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 168 | // Activate the manager |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 169 | manager.activate(); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 170 | // Initialize the test provider |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 171 | provider = new TestProvider(PID); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 172 | // Register the provider against the manager |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 173 | providerService = registry.register(provider); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 174 | // Verify register |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 175 | assertTrue("provider should be registered", |
| 176 | registry.getProviders().contains(provider.id())); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 177 | } |
| 178 | |
| 179 | @After |
| 180 | public void tearDown() { |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 181 | // Unregister provider |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 182 | registry.unregister(provider); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 183 | // Verify unregister |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 184 | assertFalse("provider should not be registered", |
| 185 | registry.getProviders().contains(provider.id())); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 186 | // Deactivate manager |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 187 | manager.deactivate(); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 188 | // Remove event dispatcher |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 189 | injectEventDispatcher(manager, null); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 190 | // Deactivate store |
| 191 | meterStore.deactivate(); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 192 | } |
| 193 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 194 | private void initMeterStore() { |
| 195 | // Let's store feature for device 1 |
| 196 | meterStore.storeMeterFeatures(mef1); |
| 197 | // Let's store feature for device 2 |
| 198 | meterStore.storeMeterFeatures(mef2); |
| 199 | } |
| 200 | |
| 201 | // Emulate metrics coming from the dataplane |
| 202 | private void pushMetrics(MeterOperation.Type type, Meter meter) { |
| 203 | // If it is an add operation |
| 204 | if (type == MeterOperation.Type.ADD) { |
| 205 | // Update state to added |
| 206 | ((DefaultMeter) meter).setState(MeterState.ADDED); |
| 207 | // Push the update in the store |
| 208 | providerService.pushMeterMetrics(meter.deviceId(), Collections.singletonList(meter)); |
| 209 | } else { |
| 210 | providerService.pushMeterMetrics(meter.deviceId(), Collections.emptyList()); |
| 211 | } |
| 212 | } |
| 213 | |
| 214 | /** |
| 215 | * Test add meter. |
| 216 | */ |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 217 | @Test |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 218 | public void testAdd() { |
| 219 | // Init store |
| 220 | initMeterStore(); |
| 221 | // Submit meter request |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 222 | manager.submit(m1Request.add()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 223 | // Verify add |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 224 | assertTrue("The meter was not added", manager.getAllMeters().size() == 1); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 225 | assertTrue("The meter was not added", manager.getMeters(did1).size() == 1); |
| 226 | // Get Meter |
| 227 | Meter installingMeter = manager.getMeter(did1, mid1); |
| 228 | // Verify add of installingMeter and pending add state |
| 229 | assertThat(installingMeter, is(m1)); |
| 230 | // Verify pending add state |
| 231 | assertThat(installingMeter.state(), is(MeterState.PENDING_ADD)); |
| 232 | // Let's simulate a working data-plane |
| 233 | pushMetrics(MeterOperation.Type.ADD, installingMeter); |
| 234 | // Get meter |
| 235 | Meter installedMeter = manager.getMeter(did1, mid1); |
| 236 | // Verify installation |
| 237 | assertThat(installedMeter.state(), is(MeterState.ADDED)); |
| 238 | assertTrue("The meter was not installed", manager.getAllMeters().size() == 1); |
| 239 | assertTrue("The meter was not installed", manager.getMeters(did1).size() == 1); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 240 | } |
| 241 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 242 | /** |
| 243 | * Test remove meter. |
| 244 | */ |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 245 | @Test |
| 246 | public void testRemove() { |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 247 | // Init store |
| 248 | initMeterStore(); |
| 249 | // Submit meter request |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 250 | manager.submit(m1Request.add()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 251 | // Withdraw meter |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 252 | manager.withdraw(m1Request.remove(), m1.id()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 253 | // Get Meter |
| 254 | Meter withdrawingMeter = manager.getMeter(did1, mid1); |
| 255 | // Verify withdrawing |
| 256 | assertThat(withdrawingMeter.state(), is(MeterState.PENDING_REMOVE)); |
| 257 | assertTrue("The meter was not withdrawn", manager.getAllMeters().size() == 1); |
| 258 | assertTrue("The meter was not withdrawn", manager.getMeters(did1).size() == 1); |
| 259 | // Let's simulate a working data-plane |
| 260 | pushMetrics(MeterOperation.Type.REMOVE, withdrawingMeter); |
| 261 | // Verify withdrawn |
| 262 | assertNull(manager.getMeter(did1, mid1)); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 263 | assertTrue("The meter was not removed", manager.getAllMeters().size() == 0); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 264 | assertTrue("The meter was not removed", manager.getMeters(did1).size() == 0); |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 265 | } |
| 266 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 267 | /** |
| 268 | * Test add multiple device. |
| 269 | */ |
alshabib | 70aaa1b | 2015-09-25 14:30:59 -0700 | [diff] [blame] | 270 | @Test |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 271 | public void testAddMultipleDevice() { |
| 272 | // Init store |
| 273 | initMeterStore(); |
| 274 | // Submit meter 1 |
alshabib | 70aaa1b | 2015-09-25 14:30:59 -0700 | [diff] [blame] | 275 | manager.submit(m1Request.add()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 276 | // Submit meter 2 |
alshabib | 70aaa1b | 2015-09-25 14:30:59 -0700 | [diff] [blame] | 277 | manager.submit(m2Request.add()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 278 | // Verify add |
| 279 | assertTrue("The meter was not added", manager.getAllMeters().size() == 2); |
| 280 | assertTrue("The meter was not added", manager.getMeters(did1).size() == 1); |
| 281 | assertTrue("The meter was not added", manager.getMeters(did2).size() == 1); |
| 282 | // Get Meters |
| 283 | Meter installingMeter1 = manager.getMeter(did1, mid1); |
| 284 | Meter installingMeter2 = manager.getMeter(did2, mid1); |
| 285 | // Verify add of installingMeter |
| 286 | assertThat(installingMeter1, is(m1)); |
| 287 | assertThat(installingMeter2, is(m2)); |
| 288 | // Verify pending add state |
| 289 | assertThat(installingMeter1.state(), is(MeterState.PENDING_ADD)); |
| 290 | assertThat(installingMeter2.state(), is(MeterState.PENDING_ADD)); |
| 291 | // Let's simulate a working data-plane |
| 292 | pushMetrics(MeterOperation.Type.ADD, installingMeter1); |
| 293 | pushMetrics(MeterOperation.Type.ADD, installingMeter2); |
| 294 | // Get meter |
| 295 | Meter installedMeter1 = manager.getMeter(did1, mid1); |
| 296 | Meter installedMeter2 = manager.getMeter(did2, mid1); |
| 297 | // Verify installation |
| 298 | assertThat(installedMeter1.state(), is(MeterState.ADDED)); |
| 299 | assertThat(installedMeter2.state(), is(MeterState.ADDED)); |
| 300 | assertTrue("The meter was not installed", manager.getAllMeters().size() == 2); |
| 301 | assertTrue("The meter was not installed", manager.getMeters(did1).size() == 1); |
| 302 | assertTrue("The meter was not installed", manager.getMeters(did2).size() == 1); |
alshabib | 70aaa1b | 2015-09-25 14:30:59 -0700 | [diff] [blame] | 303 | } |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 304 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 305 | /** |
| 306 | * Test remove meter. |
| 307 | */ |
Jordi Ortiz | aa8de49 | 2016-12-01 00:21:36 +0100 | [diff] [blame] | 308 | @Test |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 309 | public void testRemoveMultipleDevice() { |
| 310 | // Init store |
| 311 | initMeterStore(); |
| 312 | // Submit meter 1 |
Jordi Ortiz | 6c84776 | 2017-01-30 17:13:05 +0100 | [diff] [blame] | 313 | manager.submit(m1Request.add()); |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 314 | // Submit meter 2 |
| 315 | manager.submit(m2Request.add()); |
| 316 | // Withdraw meter |
| 317 | manager.withdraw(m1Request.remove(), m1.id()); |
| 318 | // Withdraw meter |
| 319 | manager.withdraw(m2Request.remove(), m2.id()); |
| 320 | // Get Meters |
| 321 | Meter withdrawingMeter1 = manager.getMeter(did1, mid1); |
| 322 | Meter withdrawingMeter2 = manager.getMeter(did2, mid1); |
| 323 | // Verify withdrawing |
| 324 | assertThat(withdrawingMeter1.state(), is(MeterState.PENDING_REMOVE)); |
| 325 | assertThat(withdrawingMeter2.state(), is(MeterState.PENDING_REMOVE)); |
| 326 | assertTrue("The meter was not withdrawn", manager.getAllMeters().size() == 2); |
| 327 | assertTrue("The meter was not withdrawn", manager.getMeters(did1).size() == 1); |
| 328 | assertTrue("The meter was not withdrawn", manager.getMeters(did2).size() == 1); |
| 329 | // Let's simulate a working data-plane |
| 330 | pushMetrics(MeterOperation.Type.REMOVE, withdrawingMeter1); |
| 331 | pushMetrics(MeterOperation.Type.REMOVE, withdrawingMeter2); |
| 332 | // Verify withdrawn |
| 333 | assertNull(manager.getMeter(did1, mid1)); |
| 334 | assertNull(manager.getMeter(did2, mid1)); |
| 335 | assertTrue("The meter was not removed", manager.getAllMeters().size() == 0); |
| 336 | assertTrue("The meter was not removed", manager.getMeters(did1).size() == 0); |
| 337 | assertTrue("The meter was not removed", manager.getMeters(did2).size() == 0); |
Jordi Ortiz | 6c84776 | 2017-01-30 17:13:05 +0100 | [diff] [blame] | 338 | } |
| 339 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 340 | // Test cluster service |
| 341 | private final class TestClusterService extends ClusterServiceAdapter { |
Jordi Ortiz | aa8de49 | 2016-12-01 00:21:36 +0100 | [diff] [blame] | 342 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 343 | private ControllerNode local = new DefaultControllerNode(NID_LOCAL, LOCALHOST); |
Jordi Ortiz | aa8de49 | 2016-12-01 00:21:36 +0100 | [diff] [blame] | 344 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 345 | @Override |
| 346 | public ControllerNode getLocalNode() { |
| 347 | return local; |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 348 | } |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 349 | |
| 350 | @Override |
| 351 | public Set<ControllerNode> getNodes() { |
| 352 | return Sets.newHashSet(); |
| 353 | } |
| 354 | |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 355 | } |
| 356 | |
| 357 | private class TestProvider extends AbstractProvider implements MeterProvider { |
| 358 | |
| 359 | protected TestProvider(ProviderId id) { |
| 360 | super(PID); |
| 361 | } |
| 362 | |
| 363 | @Override |
| 364 | public void performMeterOperation(DeviceId deviceId, MeterOperations meterOps) { |
| 365 | //Currently unused. |
| 366 | } |
| 367 | |
| 368 | @Override |
| 369 | public void performMeterOperation(DeviceId deviceId, MeterOperation meterOp) { |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 370 | //Currently unused |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 371 | } |
| 372 | } |
| 373 | |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 374 | // Test mastership service |
| 375 | private final class TestMastershipService extends MastershipServiceAdapter { |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 376 | @Override |
| 377 | public NodeId getMasterFor(DeviceId deviceId) { |
| 378 | return NID_LOCAL; |
| 379 | } |
| 380 | } |
Pier Luigi | f094c61 | 2017-10-14 12:15:02 +0200 | [diff] [blame^] | 381 | |
alshabib | e1248b6 | 2015-08-20 17:21:55 -0700 | [diff] [blame] | 382 | } |