k8s.io/kubernetes@v1.29.3/test/integration/ipamperf/test-performance.sh (about)

     1  #!/usr/bin/env bash
     2  
     3  # Copyright 2018 The Kubernetes Authors.
     4  #
     5  # Licensed under the Apache License, Version 2.0 (the "License");
     6  # you may not use this file except in compliance with the License.
     7  # You may obtain a copy of the License at
     8  #
     9  #     http://www.apache.org/licenses/LICENSE-2.0
    10  #
    11  # Unless required by applicable law or agreed to in writing, software
    12  # distributed under the License is distributed on an "AS IS" BASIS,
    13  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14  # See the License for the specific language governing permissions and
    15  # limitations under the License.
    16  
    17  set -o errexit
    18  set -o nounset
    19  set -o pipefail
    20  
    21  TEST_ARGS=""
    22  RUN_PATTERN=".*"
    23  PROFILE_OPTS=""
    24  
    25  function usage() {
    26    echo "usage: $0 <options>"
    27    echo " -h display this help message"
    28    echo " -d enable debug logs in tests"
    29    echo " -r <pattern> regex pattern to match for tests"
    30    echo " -o <filename> file to write JSON formatted results to"
    31    echo " -p <id> enable cpu and memory profiles, output written to mem-<id>.out and cpu-<id>.out"
    32    echo " -c enable custom test configuration"
    33    echo " -a <name> allocator name, one of RangeAllocator, CloudAllocator, IPAMFromCluster, IPAMFromCloud"
    34    echo " -k <num> api server qps for allocator"
    35    echo " -n <num> number of nodes to simulate"
    36    echo " -m <num> api server qps for node creation"
    37    echo " -l <num> gce cloud endpoint qps"
    38    exit 1
    39  }
    40  
    41  while getopts ":hdr:o:p:ca:k:n:m:l:" opt; do
    42    case ${opt} in
    43      d) TEST_ARGS="${TEST_ARGS} -v=6"
    44        ;;
    45      r) RUN_PATTERN="${OPTARG}"
    46        ;;
    47      o) TEST_ARGS="${TEST_ARGS} -log ${OPTARG}"
    48        ;;
    49      p) PROFILE_OPTS="-memprofile mem-${OPTARG}.out -cpuprofile cpu-${OPTARG}.out"
    50        ;;
    51      c) TEST_ARGS="${TEST_ARGS} -custom"
    52        ;;
    53      a) TEST_ARGS="${TEST_ARGS} -allocator ${OPTARG}"
    54        ;;
    55      k) TEST_ARGS="${TEST_ARGS} -kube-qps ${OPTARG}"
    56        ;;
    57      n) TEST_ARGS="${TEST_ARGS} -num-nodes ${OPTARG}"
    58        ;;
    59      m) TEST_ARGS="${TEST_ARGS} -create-qps ${OPTARG}"
    60        ;;
    61      l) TEST_ARGS="${TEST_ARGS} -cloud-qps ${OPTARG}"
    62        ;;
    63      h) usage
    64        ;;
    65      \?) usage
    66        ;;
    67    esac
    68  done
    69  
    70  KUBE_ROOT=$(dirname "${BASH_SOURCE[0]}")/../../../
    71  source "${KUBE_ROOT}/hack/lib/init.sh"
    72  
    73  kube::golang::setup_env
    74  
    75  DIR_BASENAME=$(dirname "${BASH_SOURCE[0]}")
    76  pushd "${DIR_BASENAME}"
    77  
    78  cleanup() {
    79    popd 2> /dev/null
    80    kube::etcd::cleanup
    81    kube::log::status "performance test cleanup complete"
    82  }
    83  
    84  trap cleanup EXIT
    85  
    86  kube::etcd::start
    87  
    88  # Running IPAM tests. It might take a long time.
    89  kube::log::status "performance test (IPAM) start"
    90  go test "${PROFILE_OPTS}" -test.run="${RUN_PATTERN}" -test.timeout=60m -test.short=false -v -args "${TEST_ARGS}"
    91  kube::log::status "... IPAM tests finished."