Ray Milkey

Add unit test for link collection intent installer

Change-Id: Ifd89b9868b320546bc55ba78e207993e35aaf50c
/*
* Copyright 2015 Open Networking Laboratory
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.onosproject.net.intent.impl;
import java.util.Collection;
import java.util.List;
import java.util.Set;
import org.junit.Before;
import org.junit.Test;
import org.onosproject.net.DefaultLink;
import org.onosproject.net.DeviceId;
import org.onosproject.net.Link;
import org.onosproject.net.flow.FlowRuleOperation;
import org.onosproject.net.intent.LinkCollectionIntent;
import com.google.common.collect.ImmutableSet;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.hasSize;
import static org.hamcrest.Matchers.notNullValue;
import static org.onosproject.net.Link.Type.DIRECT;
import static org.onosproject.net.NetTestTools.APP_ID;
import static org.onosproject.net.NetTestTools.PID;
public class LinkCollectionIntentInstallerTest extends IntentInstallerTest {
LinkCollectionIntentInstaller installer;
private final Set<Link> links = ImmutableSet.of(
new DefaultLink(PID, d1p1, d2p0, DIRECT),
new DefaultLink(PID, d2p1, d3p1, DIRECT),
new DefaultLink(PID, d1p1, d3p1, DIRECT));
private LinkCollectionIntent intent;
/**
* Configures objects used in all the test cases.
*/
@Before
public void localSetUp() {
installer = new LinkCollectionIntentInstaller();
installer.coreService = testCoreService;
installer.intentManager =
new IntentInstallerTest.MockIntentManager(LinkCollectionIntent.class);
intent = new LinkCollectionIntent(APP_ID, selector, treatment, links, d1p1, d3p1);
}
private FlowRuleOperation findOperation(Collection<FlowRuleOperation> ops,
DeviceId deviceId) {
for (FlowRuleOperation op : ops) {
if (op.rule().deviceId().equals(deviceId)) {
return op;
}
}
return null;
}
/**
* Tests activation and deactivation of the installer.
*/
@Test
public void activateDeactivate() {
installer.activate();
installer.deactivate();
}
/**
* Tests installation operation of the path intent installer.
*/
@Test
public void install() {
installer.activate();
List<Collection<FlowRuleOperation>> operations =
installer.install(intent);
assertThat(operations, notNullValue());
assertThat(operations, hasSize(1));
Collection<FlowRuleOperation> flowRuleOpsCollection = operations.get(0);
assertThat(flowRuleOpsCollection, hasSize(links.size()));
FlowRuleOperation op0 = findOperation(flowRuleOpsCollection,
d1p0.deviceId());
checkFlowOperation(op0, FlowRuleOperation.Type.ADD, d1p0.deviceId());
FlowRuleOperation op1 = findOperation(flowRuleOpsCollection,
d2p0.deviceId());
checkFlowOperation(op1, FlowRuleOperation.Type.ADD, d2p0.deviceId());
FlowRuleOperation op2 = findOperation(flowRuleOpsCollection,
d3p0.deviceId());
checkFlowOperation(op2, FlowRuleOperation.Type.ADD, d3p0.deviceId());
installer.deactivate();
}
/**
* Checks the uninstall operation of the path intent installer.
*/
@Test
public void uninstall() {
installer.activate();
List<Collection<FlowRuleOperation>> operations =
installer.uninstall(intent);
assertThat(operations, notNullValue());
assertThat(operations, hasSize(1));
Collection<FlowRuleOperation> flowRuleOpsCollection = operations.get(0);
assertThat(flowRuleOpsCollection, hasSize(links.size()));
FlowRuleOperation op0 = findOperation(flowRuleOpsCollection,
d1p0.deviceId());
checkFlowOperation(op0, FlowRuleOperation.Type.REMOVE, d1p0.deviceId());
FlowRuleOperation op1 = findOperation(flowRuleOpsCollection,
d2p0.deviceId());
checkFlowOperation(op1, FlowRuleOperation.Type.REMOVE, d2p0.deviceId());
FlowRuleOperation op2 = findOperation(flowRuleOpsCollection,
d3p0.deviceId());
checkFlowOperation(op2, FlowRuleOperation.Type.REMOVE, d3p0.deviceId());
installer.deactivate();
}
}