# Copyright (c) 2019 Cisco and/or its affiliates.
# 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.

*** Settings ***
| Library  | resources.tools.wrk.wrk
| Library  | resources.libraries.python.IPUtil
| Library  | resources.libraries.python.DUTSetup
| Library  | resources.libraries.python.TrafficGenerator
| Library  | resources.libraries.python.topology.Topology
| ...
| Documentation | L2 keywords to set up wrk and to measure performance
| ... | parameters using wrk.

*** Keywords ***
| Measure throughput
| | [Documentation]
| | ... | Measure throughput using wrk.
| | ...
| | ... | *Arguments:*
| | ... | - ${profile} - The name of the wrk traffic profile defining the
| | ... | traffic. Type: string
| | ...
| | ... | *Example:*
| | ...
| | ... | \| Measure throughput \| wrk-bw-1url-1core-50con \|
| | ...
| | [Arguments] | ${profile}
| | ...
| | ${tg_numa}= | Get interfaces numa node | ${tg} | ${tg_if1} | ${tg_if2}
| | ${output}= | Run wrk | ${tg} | ${profile} | ${tg_numa} | bw
| | Set test message | ${output}

| Measure requests per second
| | [Documentation]
| | ... | Measure number of requests per second using wrk.
| | ...
| | ... | *Arguments:*
| | ... | - ${profile} - The name of the wrk traffic profile defining the
| | ... | traffic. Type: string
| | ...
| | ... | *Example:*
| | ...
| | ... | \| Measure requests per second \| wrk-bw-1url-1core-50con \|
| | ...
| | [Arguments] | ${profile}
| | ...
| | ${tg_numa}= | Get interfaces numa node | ${tg} | ${tg_if1} | ${tg_if2}
| | ${output}= | Run wrk | ${tg} | ${profile} | ${tg_numa} | rps
| | Set test message | ${output}

| Measure connections per second
| | [Documentation]
| | ... | Measure number of connections per second using wrk.
| | ...
| | ... | *Arguments:*
| | ... | - ${profile} - The name of the wrk traffic profile defining the
| | ... | traffic. Type: string
| | ...
| | ... | *Example:*
| | ...
| | ... | \| Measure connections per second \| wrk-bw-1url-1core-50con \|
| | ...
| | [Arguments] | ${profile}
| | ...
| | ${tg_numa}= | Get interfaces numa node | ${tg} | ${tg_if1} | ${tg_if2}
| | ${output}= | Run wrk | ${tg} | ${profile} | ${tg_numa} | cps
| | Set test message | ${output}