use of org.openkilda.testing.service.traffexam.model.Bandwidth in project open-kilda by telstra.
the class FlowTrafficExamBuilder method buildExam.
/**
* Build Exam in one direction.
*/
public Exam buildExam(FlowPayload flow, int bandwidth, Long duration) throws FlowNotApplicableException {
Optional<TraffGen> source = Optional.ofNullable(endpointToTraffGen.get(makeComparableEndpoint(flow.getSource())));
Optional<TraffGen> dest = Optional.ofNullable(endpointToTraffGen.get(makeComparableEndpoint(flow.getDestination())));
// do not allow traffic exam on OF version <1.3. We are not installing meters on OF 1.2 intentionally
String srcOfVersion = topology.getSwitches().stream().filter(sw -> sw.getDpId().equals(flow.getSource().getDatapath())).findFirst().map(Switch::getOfVersion).orElseThrow(() -> new IllegalStateException(format("Switch %s not found", flow.getSource().getDatapath())));
String dstOfVersion = topology.getSwitches().stream().filter(sw -> sw.getDpId().equals(flow.getDestination().getDatapath())).findFirst().map(Switch::getOfVersion).orElseThrow(() -> new IllegalStateException(format("Switch %s not found", flow.getDestination().getDatapath())));
checkIsFlowApplicable(flow.getId(), source.isPresent() && !"OF_12".equals(srcOfVersion), dest.isPresent() && !"OF_12".equals(dstOfVersion));
List<Vlan> srcVlanIds = new ArrayList<Vlan>();
srcVlanIds.add(new Vlan(flow.getSource().getVlanId()));
srcVlanIds.add(new Vlan(flow.getSource().getInnerVlanId()));
List<Vlan> dstVlanIds = new ArrayList<Vlan>();
dstVlanIds.add(new Vlan(flow.getDestination().getVlanId()));
dstVlanIds.add(new Vlan(flow.getDestination().getInnerVlanId()));
// noinspection ConstantConditions
Host sourceHost = traffExam.hostByName(source.get().getName());
// noinspection ConstantConditions
Host destHost = traffExam.hostByName(dest.get().getName());
// 1024 * 1024 / 8 / 1500 = 87.3...
return Exam.builder().flow(flow).source(sourceHost).sourceVlans(srcVlanIds).dest(destHost).destVlans(dstVlanIds).bandwidthLimit(new Bandwidth(bandwidth)).burstPkt(100).timeLimitSeconds(duration != null ? new TimeLimit(duration) : null).build();
}
use of org.openkilda.testing.service.traffexam.model.Bandwidth in project open-kilda by telstra.
the class StubServiceFactory method getTraffExamStub.
/**
* Get a stub for {@link TraffExamService}. The instance is tied to the factory state.
*/
public TraffExamService getTraffExamStub() {
TraffExamService serviceMock = mock(TraffExamService.class);
when(serviceMock.hostByName(any())).thenAnswer(invocation -> {
String hostName = (String) invocation.getArguments()[0];
Host host = mock(Host.class);
when(host.getName()).thenReturn(hostName);
return host;
});
when(serviceMock.waitExam(any())).thenAnswer(invocation -> {
Exam exam = (Exam) invocation.getArguments()[0];
ExamReport report = mock(ExamReport.class);
when(report.hasError()).thenReturn(false);
when(report.hasTraffic()).thenReturn(flows.containsKey(exam.getFlow().getId()));
when(report.getBandwidth()).thenReturn(new Bandwidth(exam.getFlow().getMaximumBandwidth()));
return report;
});
return serviceMock;
}
use of org.openkilda.testing.service.traffexam.model.Bandwidth in project open-kilda by telstra.
the class FlowTrafficExamBuilder method buildYFlowExam.
/**
* Build traff exam object for both 'y-flow' subflows in both directions.
*/
public YFlowBidirectionalExam buildYFlowExam(YFlow flow, long bandwidth, Long duration) throws FlowNotApplicableException {
SubFlow subFlow1 = flow.getSubFlows().get(0);
SubFlow subFlow2 = flow.getSubFlows().get(1);
Optional<TraffGen> source = Optional.ofNullable(endpointToTraffGen.get(makeComparableEndpoint(flow.getSharedEndpoint())));
Optional<TraffGen> dest1 = Optional.ofNullable(endpointToTraffGen.get(makeComparableEndpoint(subFlow1.getEndpoint())));
Optional<TraffGen> dest2 = Optional.ofNullable(endpointToTraffGen.get(makeComparableEndpoint(subFlow2.getEndpoint())));
checkIsFlowApplicable(flow.getYFlowId(), source.isPresent(), dest1.isPresent() && dest2.isPresent());
List<Vlan> srcVlanIds1 = ImmutableList.of(new Vlan(subFlow1.getSharedEndpoint().getVlanId()), new Vlan(subFlow1.getSharedEndpoint().getInnerVlanId()));
List<Vlan> srcVlanIds2 = ImmutableList.of(new Vlan(subFlow2.getSharedEndpoint().getVlanId()), new Vlan(subFlow2.getSharedEndpoint().getInnerVlanId()));
List<Vlan> dstVlanIds1 = ImmutableList.of(new Vlan(subFlow1.getEndpoint().getVlanId()), new Vlan(subFlow1.getEndpoint().getInnerVlanId()));
List<Vlan> dstVlanIds2 = ImmutableList.of(new Vlan(subFlow2.getEndpoint().getVlanId()), new Vlan(subFlow2.getEndpoint().getInnerVlanId()));
// noinspection ConstantConditions
Host sourceHost = traffExam.hostByName(source.get().getName());
// noinspection ConstantConditions
Host destHost1 = traffExam.hostByName(dest1.get().getName());
Host destHost2 = traffExam.hostByName(dest2.get().getName());
Exam forward1 = Exam.builder().flow(null).source(sourceHost).sourceVlans(srcVlanIds1).dest(destHost1).destVlans(dstVlanIds1).bandwidthLimit(new Bandwidth(bandwidth)).burstPkt(200).timeLimitSeconds(duration != null ? new TimeLimit(duration) : null).build();
Exam forward2 = Exam.builder().flow(null).source(sourceHost).sourceVlans(srcVlanIds2).dest(destHost2).destVlans(dstVlanIds2).bandwidthLimit(new Bandwidth(bandwidth)).burstPkt(200).timeLimitSeconds(duration != null ? new TimeLimit(duration) : null).build();
Exam reverse1 = Exam.builder().flow(null).source(destHost1).sourceVlans(dstVlanIds1).dest(sourceHost).destVlans(srcVlanIds1).bandwidthLimit(new Bandwidth(bandwidth)).burstPkt(200).timeLimitSeconds(duration != null ? new TimeLimit(duration) : null).build();
Exam reverse2 = Exam.builder().flow(null).source(destHost2).sourceVlans(dstVlanIds2).dest(sourceHost).destVlans(srcVlanIds2).bandwidthLimit(new Bandwidth(bandwidth)).burstPkt(200).timeLimitSeconds(duration != null ? new TimeLimit(duration) : null).build();
return new YFlowBidirectionalExam(forward1, reverse1, forward2, reverse2);
}
use of org.openkilda.testing.service.traffexam.model.Bandwidth in project open-kilda by telstra.
the class FlowTrafficExamBuilder method buildBidirectionalExam.
/**
* Builds bidirectional exam.
*/
public FlowBidirectionalExam buildBidirectionalExam(FlowPayload flow, long bandwidth, Long duration) throws FlowNotApplicableException {
Optional<TraffGen> source = Optional.ofNullable(endpointToTraffGen.get(makeComparableEndpoint(flow.getSource())));
Optional<TraffGen> dest = Optional.ofNullable(endpointToTraffGen.get(makeComparableEndpoint(flow.getDestination())));
checkIsFlowApplicable(flow.getId(), source.isPresent(), dest.isPresent());
List<Vlan> srcVlanIds = new ArrayList<Vlan>();
srcVlanIds.add(new Vlan(flow.getSource().getVlanId()));
srcVlanIds.add(new Vlan(flow.getSource().getInnerVlanId()));
List<Vlan> dstVlanIds = new ArrayList<Vlan>();
dstVlanIds.add(new Vlan(flow.getDestination().getVlanId()));
dstVlanIds.add(new Vlan(flow.getDestination().getInnerVlanId()));
// noinspection ConstantConditions
Host sourceHost = traffExam.hostByName(source.get().getName());
// noinspection ConstantConditions
Host destHost = traffExam.hostByName(dest.get().getName());
Exam forward = Exam.builder().flow(flow).source(sourceHost).sourceVlans(srcVlanIds).dest(destHost).destVlans(dstVlanIds).bandwidthLimit(new Bandwidth(bandwidth)).burstPkt(0).timeLimitSeconds(duration != null ? new TimeLimit(duration) : null).build();
Exam reverse = Exam.builder().flow(flow).source(destHost).sourceVlans(dstVlanIds).dest(sourceHost).destVlans(srcVlanIds).bandwidthLimit(new Bandwidth(bandwidth)).burstPkt(0).timeLimitSeconds(duration != null ? new TimeLimit(duration) : null).build();
return new FlowBidirectionalExam(forward, reverse);
}
Aggregations