Skip to content

Commit 21d0073

Browse files
committed
roachtest/vecindex: create a nightly test for vector indexing
The new vector index roachtest doest too much work on the test runner. To avoid flakiness, we're going to run this test on a separate test runner until a more roachtest-friendly version of the test can be built. These scripts implement that test. Informs: #154590 Release note: None
1 parent 81f0490 commit 21d0073

File tree

2 files changed

+52
-0
lines changed

2 files changed

+52
-0
lines changed
Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
#!/usr/bin/env bash
2+
3+
# Copyright 2024 The Cockroach Authors.
4+
#
5+
# Use of this software is governed by the CockroachDB Software License
6+
# included in the /LICENSE file.
7+
8+
9+
set -exuo pipefail
10+
11+
dir="$(dirname $(dirname $(dirname $(dirname "${0}"))))"
12+
13+
source "$dir/teamcity-support.sh" # For $root
14+
source "$dir/teamcity-bazel-support.sh" # For run_bazel
15+
16+
BAZEL_SUPPORT_EXTRA_DOCKER_ARGS="-e LITERAL_ARTIFACTS_DIR=$root/artifacts -e BUILD_VCS_NUMBER -e CLOUD -e COCKROACH_DEV_LICENSE -e TESTS -e COUNT -e GITHUB_API_TOKEN -e GITHUB_ORG -e GITHUB_REPO -e GOOGLE_EPHEMERAL_CREDENTIALS -e GOOGLE_KMS_KEY_A -e GOOGLE_KMS_KEY_B -e GOOGLE_CREDENTIALS_ASSUME_ROLE -e GOOGLE_SERVICE_ACCOUNT -e SLACK_TOKEN -e TC_BUILDTYPE_ID -e TC_BUILD_BRANCH -e TC_BUILD_ID -e TC_SERVER_URL -e SELECT_PROBABILITY -e COCKROACH_RANDOM_SEED -e ROACHTEST_ASSERTIONS_ENABLED_SEED -e ROACHTEST_FORCE_RUN_INVALID_RELEASE_BRANCH -e GRAFANA_SERVICE_ACCOUNT_JSON -e GRAFANA_SERVICE_ACCOUNT_AUDIENCE -e USE_SPOT -e SNOWFLAKE_USER -e SNOWFLAKE_PVT_KEY -e COCKROACH_EA_PROBABILITY" \
17+
run_bazel build/teamcity/cockroach/nightlies/vecindex_nightly_impl.sh
Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,35 @@
1+
#!/usr/bin/env bash
2+
3+
# Copyright 2024 The Cockroach Authors.
4+
#
5+
# Use of this software is governed by the CockroachDB Software License
6+
# included in the /LICENSE file.
7+
8+
set -exuo pipefail
9+
10+
dir="$(dirname $(dirname $(dirname $(dirname "${0}"))))"
11+
set -a
12+
source "$dir/teamcity-support.sh"
13+
set +a
14+
15+
if [[ ! -f ~/.ssh/id_rsa.pub ]]; then
16+
ssh-keygen -q -C "roachtest-nightly-bazel $(date)" -N "" -f ~/.ssh/id_rsa
17+
fi
18+
19+
source $root/build/teamcity/util/roachtest_util.sh
20+
21+
artifacts=/artifacts
22+
23+
arch=amd64
24+
$root/build/teamcity/cockroach/nightlies/roachtest_compile_bits.sh $arch
25+
26+
build/teamcity-roachtest-invoke.sh \
27+
--suite vecindex \
28+
--cloud "gce" \
29+
--cluster-id "${TC_BUILD_ID}" \
30+
--artifacts=/artifacts \
31+
--artifacts-literal="${LITERAL_ARTIFACTS_DIR:-}" \
32+
--parallelism="${PARALLELISM}" \
33+
--cpu-quota="${CPUQUOTA}" \
34+
--use-spot="${USE_SPOT:-auto}" \
35+
--slack-token="${SLACK_TOKEN}" \

0 commit comments

Comments
 (0)