VOL-2901 Unit tests and necessary updates on stats manager
in the scope of VOL-2796
Change-Id: I95d6fd0241d2c54a5a122d58917c6dc45e5abcf7
diff --git a/app/src/test/java/org/opencord/cordmcast/impl/McastTestBase.java b/app/src/test/java/org/opencord/cordmcast/impl/McastTestBase.java
index 986c175..4ae2825 100644
--- a/app/src/test/java/org/opencord/cordmcast/impl/McastTestBase.java
+++ b/app/src/test/java/org/opencord/cordmcast/impl/McastTestBase.java
@@ -54,6 +54,7 @@
import org.onosproject.net.flow.TrafficTreatment;
import org.onosproject.net.flow.criteria.Criterion;
import org.onosproject.net.flow.criteria.IPCriterion;
+import org.onosproject.net.flow.criteria.VlanIdCriterion;
import org.onosproject.net.flow.instructions.Instruction;
import org.onosproject.net.flow.instructions.Instructions.OutputInstruction;
import org.onosproject.net.flowobjective.FlowObjectiveServiceAdapter;
@@ -164,14 +165,22 @@
* Mocks the McastConfig class to return vlan id value.
*/
static class MockMcastConfig extends McastConfig {
+ private VlanId egressVlan;
+ private VlanId egressInnerVlan;
+
+ public MockMcastConfig(VlanId egressVlan, VlanId egressInnerVlan) {
+ this.egressVlan = egressVlan;
+ this.egressInnerVlan = egressInnerVlan;
+ }
+
@Override
public VlanId egressVlan() {
- return VlanId.vlanId("4000");
+ return egressVlan;
}
@Override
public VlanId egressInnerVlan() {
- return VlanId.NONE;
+ return egressInnerVlan;
}
}
@@ -181,9 +190,21 @@
@SuppressWarnings("unchecked")
static final class TestNetworkConfigRegistry
extends NetworkConfigRegistryAdapter {
+
+ private VlanId egressVlan = VlanId.vlanId("4000");
+ private VlanId egressInnerVlan = VlanId.NONE;
+
+ public void setEgressVlan(VlanId egressVlan) {
+ this.egressVlan = egressVlan;
+ }
+
+ public void setEgressInnerVlan(VlanId egressInnerVlan) {
+ this.egressInnerVlan = egressInnerVlan;
+ }
+
@Override
public <S, C extends Config<S>> C getConfig(S subject, Class<C> configClass) {
- McastConfig mcastConfig = new MockMcastConfig();
+ McastConfig mcastConfig = new MockMcastConfig(egressVlan, egressInnerVlan);
return (C) mcastConfig;
}
}
@@ -293,4 +314,16 @@
return (IPCriterion) ipCriterion;
}
+ public VlanIdCriterion vlanId(TrafficSelector trafficSelector, Criterion.Type type) {
+ Set<Criterion> criterionSet = trafficSelector.criteria();
+ Iterator<Criterion> it = criterionSet.iterator();
+ VlanIdCriterion criterion = null;
+ while (it.hasNext()) {
+ Criterion criteria = it.next();
+ if (type == criteria.type()) {
+ criterion = (VlanIdCriterion) criteria;
+ }
+ }
+ return criterion;
+ }
}