Ray Milkey
Committed by Gerrit Code Review

Unit tests for DefaultTrafficTreatment

Change-Id: I68b0773b69d85c1d05d58b32c1dceaca73c0a5e1
/*
* Copyright 2014 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.onlab.onos.net.flow;
import java.util.List;
import org.junit.Test;
import org.onlab.onos.net.PortNumber;
import org.onlab.onos.net.flow.instructions.Instruction;
import org.onlab.packet.IpAddress;
import org.onlab.packet.MacAddress;
import org.onlab.packet.VlanId;
import com.google.common.testing.EqualsTester;
import static org.hamcrest.CoreMatchers.equalTo;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.hasSize;
import static org.hamcrest.Matchers.is;
import static org.onlab.onos.net.flow.instructions.L0ModificationInstruction.L0SubType;
import static org.onlab.onos.net.flow.instructions.L0ModificationInstruction.ModLambdaInstruction;
/**
* Unit tests for the DefaultTrafficTreatment class.
*/
public class DefaultTrafficTreatmentTest {
// Tests for the nested Builder class
/**
* Tests that the Builder constructors return equivalent objects
* when given the same data.
*/
@Test
public void testTreatmentBuilderConstructors() {
final TrafficTreatment treatment1 =
DefaultTrafficTreatment.builder()
.add(new ModLambdaInstruction(L0SubType.LAMBDA, (short) 4))
.build();
final TrafficTreatment treatment2 =
DefaultTrafficTreatment.builder(treatment1).build();
assertThat(treatment1, is(equalTo(treatment2)));
}
/**
* Tests methods defined on the Builder.
*/
@Test
public void testBuilderMethods() {
final Instruction instruction1 =
new ModLambdaInstruction(L0SubType.LAMBDA, (short) 4);
final TrafficTreatment.Builder builder1 =
DefaultTrafficTreatment.builder()
.add(instruction1)
.setEthDst(MacAddress.BROADCAST)
.setEthSrc(MacAddress.BROADCAST)
.setIpDst(IpAddress.valueOf("1.1.1.1"))
.setIpSrc(IpAddress.valueOf("2.2.2.2"))
.setLambda((short) 4)
.setOutput(PortNumber.portNumber(2))
.setVlanId(VlanId.vlanId((short) 4))
.setVlanPcp((byte) 3);
final TrafficTreatment treatment1 = builder1.build();
final List<Instruction> instructions1 = treatment1.instructions();
assertThat(instructions1, hasSize(9));
builder1.drop();
builder1.add(instruction1);
final List<Instruction> instructions2 = builder1.build().instructions();
assertThat(instructions2, hasSize(8));
}
/**
* Tests equals(), hashCode() and toString() methods of
* DefaultTrafficTreatment.
*/
@Test
public void testEquals() {
final TrafficTreatment treatment1 =
DefaultTrafficTreatment.builder()
.add(new ModLambdaInstruction(L0SubType.LAMBDA, (short) 4))
.build();
final TrafficTreatment sameAsTreatment1 =
DefaultTrafficTreatment.builder()
.add(new ModLambdaInstruction(L0SubType.LAMBDA, (short) 4))
.build();
final TrafficTreatment treatment2 =
DefaultTrafficTreatment.builder()
.add(new ModLambdaInstruction(L0SubType.LAMBDA, (short) 2))
.build();
new EqualsTester()
.addEqualityGroup(treatment1, sameAsTreatment1)
.addEqualityGroup(treatment2)
.testEquals();
}
}