/
setup-test-table.sh
executable file
·42 lines (33 loc) · 1.08 KB
/
setup-test-table.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#!/usr/bin/env bash
# Set up a table to use for integration tests.
set -e
set -u
ADMIN_HOSTS=(
"bigtableadmin.googleapis.com:443"
"test-bigtableadmin.sandbox.googleapis.com:443"
)
ZONE=us-central1-b
FAMILY=cf
# Format: projects/<project-id>/instances/<instance-id>/tables/<table-id>
TABLE_NAME=$1
if [[ ${TABLE_NAME} =~ projects\/([^/]+)\/instances\/([^/]+)\/tables\/([^/]+) ]]; then
PROJECT_ID=${BASH_REMATCH[1]}
INSTANCE_ID=${BASH_REMATCH[2]}
TABLE_ID=${BASH_REMATCH[3]}
else
echo "Invalid table name: $TABLE_NAME" 1>&2
exit 1
fi
call_cbt() {
cbt -admin-endpoint="${ADMIN_HOST}" -project "${PROJECT_ID}" -instance "${INSTANCE_ID}" "$@"
}
for ADMIN_HOST in "${ADMIN_HOSTS[@]}"; do
# Ensure that the instance exists.
if ! call_cbt listinstances | grep -q "${INSTANCE_ID}"; then
call_cbt createinstance "${INSTANCE_ID}" "${INSTANCE_ID}" "${INSTANCE_ID}-c0" "${ZONE}" 1 SSD
fi
# Ensure that the table exists
if ! call_cbt -instance "$INSTANCE_ID" ls | grep -q "^${TABLE_ID}\$"; then
call_cbt createtable "$TABLE_ID" "families=$FAMILY:maxversions=1||maxage=1h"
fi
done