[VOL-2778] Support for TT workflow in the scale test
Change-Id: I76b850593e17ffb2781fbc169f20ecb611937811
diff --git a/libraries/onos.robot b/libraries/onos.robot
index 2121a50..11aa85d 100755
--- a/libraries/onos.robot
+++ b/libraries/onos.robot
@@ -380,11 +380,14 @@
Should Be Equal As Integers ${aaa_users} ${expected_onus}
Validate DHCP Allocations
- [Arguments] ${onos_ssh_connection} ${expected_onus}
+ [Arguments] ${onos_ssh_connection} ${count} ${workflow}
[Documentation] Matches for number of dhcpacks based on number of onus
${allocations}= Execute ONOS CLI Command on open connection ${onos_ssh_connection}
... dhcpl2relay-allocations | grep DHCPACK | wc -l
- Should Be Equal As Integers ${allocations} ${expected_onus}
+ # if the workflow is TT we'll have 2 allocations for each ONU
+ ${ttAllocations}= Evaluate (${count} * 2)
+ ${count}= Set Variable If $workflow=='tt' ${ttAllocations} ${count}
+ Should Be Equal As Integers ${allocations} ${count}
Validate Subscriber DHCP Allocation
[Arguments] ${ip} ${port} ${onu_port}
@@ -437,9 +440,10 @@
Wait Until Keyword Succeeds 10m 5s Assert Number of AAA-Users ${onos_ssh_connection} ${count}
Wait for DHCP Ack
- [Arguments] ${onos_ssh_connection} ${count}
+ [Arguments] ${onos_ssh_connection} ${count} ${workflow}
[Documentation] Waits untill a certain number of subscribers have received a DHCP_ACK
- Wait Until Keyword Succeeds 10m 5s Validate DHCP Allocations ${onos_ssh_connection} ${count}
+ Wait Until Keyword Succeeds 10m 5s Validate DHCP Allocations
+ ... ${onos_ssh_connection} ${count} ${workflow}
Provision subscriber
[Documentation] Calls volt-add-subscriber-access in ONOS