TT scale job needs to account for IGMP flow
Change-Id: I467b00c65b3ecb5147f3bfe677873b95e6d32465
diff --git a/libraries/flows.robot b/libraries/flows.robot
index e5f28b5..ff8b673 100644
--- a/libraries/flows.robot
+++ b/libraries/flows.robot
@@ -109,8 +109,8 @@
[Documentation] Calculate the flow for the TT workflow
[Arguments] ${uni_count} ${olt_count} ${provisioned} ${withDhcp} ${withIgmp} ${withLldp}
# TODO account for withDhcp, withIgmp, see Calculate Att flows for examples
- # (1 LLDP + 1 DHCP * OLTS) before provisioning
- # (1 DHCP, 1 IGMP, 4 DP * ONUs) * (1 LLDP + 1 DHCP * OLTS) after provisioning
+ # (1 LLDP + 1 DHCP + 1 IGMP * OLTS) before provisioning
+ # (1 DHCP, 1 IGMP, 4 DP * ONUs) * (1 LLDP + 1 DHCP + 1 IGMP * OLTS) after provisioning
${dhcpFlowsCount}= Run Keyword If $withDhcp=='true'
... Evaluate 1
... ELSE
@@ -119,8 +119,15 @@
... Evaluate 1
... ELSE
... Evaluate 0
- ${flow_count}= Run Keyword If $provisioned=='false'
- ... Evaluate (${olt_count} * ${dhcpFlowsCount}) + (${olt_count} * ${lldpFlowsCount})
+ ${igmpFlowsCount}= Run Keyword If $withIgmp=='true'
+ ... Evaluate 1
... ELSE
- ... Evaluate (${uni_count} * 15) + (${olt_count} * ${dhcpFlowsCount}) + (${olt_count} * ${lldpFlowsCount})
+ ... Evaluate 0
+ ${totalDhcpFlows}= Evaluate ${olt_count} * ${dhcpFlowsCount}
+ ${totalLldpFlows}= Evaluate ${olt_count} * ${lldpFlowsCount}
+ ${totalIgmpFlows}= Evaluate ${olt_count} * ${igmpFlowsCount}
+ ${flow_count}= Run Keyword If $provisioned=='false'
+ ... Evaluate ${totalDhcpFlows} + ${totalLldpFlows} + ${totalIgmpFlows}
+ ... ELSE
+ ... Evaluate (${uni_count} * 15) + ${totalDhcpFlows} + ${totalLldpFlows} + ${totalIgmpFlows}
Return From Keyword ${flow_count}
\ No newline at end of file