mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 09:17:30 +03:00
e50c916cc3
Summary: There is redundant code in the codebase relating to fetching configuration for database for the purpose of testing. This commit refactors some code which enables removing the redundant code. Reviewed By: quark-zju Differential Revision: D8961806 fbshipit-source-id: f673bfffb80304d98987c9727f6e709fdb742813
102 lines
2.1 KiB
Bash
102 lines
2.1 KiB
Bash
TESTDIR=${TESTDIR:-.}
|
|
GETDB_PATH="$TESTDIR/${HGTEST_GETDB_PATH:-getdb.sh}"
|
|
|
|
if [[ ! -f "$GETDB_PATH" ]]; then
|
|
echo "skipped: getdb.sh missing. copy from getdb.sh.example and edit it"
|
|
exit 80
|
|
fi
|
|
|
|
if ! ${PYTHON:-python} -c "import mysql.connector" 2>/dev/null; then
|
|
echo "skipped: mysql-connector-python missing"
|
|
exit 80
|
|
fi
|
|
|
|
source "$GETDB_PATH" >/dev/null
|
|
|
|
[[ -z $DBHOST ]] && DBHOST=localhost
|
|
[[ -z $DBPORT ]] && DBPORT=3306
|
|
[[ -z $DBENGINE ]] && DBENGINE=innodb
|
|
[[ -z $DBPASS && -n $PASSWORD ]] && DBPASS="$PASSWORD"
|
|
[[ -z $DBUSER && -n $USER ]] && DBUSER="$USER"
|
|
[[ -z $DBNAME ]] && DBNAME="testdb_hgsql_$$_$(date +%s)" && DBAUTODROP=1
|
|
[[ -n $DBPASS ]] && DBPASSOPT="-p$DBPASS"
|
|
|
|
MYSQLLOG="${MYSQLLOG:-/dev/null}"
|
|
|
|
# skip if DBENGINE is not supported
|
|
( mysql -h "$DBHOST" -P "$DBPORT" -u "$DBUSER" "$DBPASSOPT" \
|
|
--execute='SHOW ENGINES' --silent --skip-column-names | \
|
|
egrep -iq "^$DBENGINE[[:space:]](default|yes)[[:space:]]" )
|
|
|
|
if [[ $? != 0 ]]; then
|
|
echo "skipped: $DBENGINE unsupported"
|
|
exit 80
|
|
fi
|
|
|
|
mysql -h "$DBHOST" -P "$DBPORT" -u "$DBUSER" "$DBPASSOPT" &>> "$MYSQLLOG" <<EOF
|
|
CREATE DATABASE IF NOT EXISTS $DBNAME;
|
|
USE $DBNAME;
|
|
DROP TABLE IF EXISTS revisions;
|
|
DROP TABLE IF EXISTS revision_references;
|
|
$(cat $TESTDIR/hgsql/schema.$DBENGINE.sql)
|
|
EOF
|
|
|
|
if [[ $? != 0 ]]; then
|
|
echo "skipped: unable to initialize the database. check your getdb.sh"
|
|
exit 80
|
|
fi
|
|
|
|
function droptestdb() {
|
|
mysql -h "$DBHOST" -P "$DBPORT" -u "$DBUSER" "$DBPASSOPT" &>> "$MYSQLLOG" <<EOF
|
|
DROP DATABASE $DBNAME;
|
|
EOF
|
|
}
|
|
|
|
[[ $DBAUTODROP == 1 ]] && trap droptestdb EXIT
|
|
|
|
function initserver() {
|
|
hg init --config extensions.hgsql= $1
|
|
configureserver $1 $2
|
|
}
|
|
|
|
configureserver() {
|
|
cat >> $1/.hg/hgrc <<EOF
|
|
[extensions]
|
|
hgsql=
|
|
strip=
|
|
|
|
[hgsql]
|
|
enabled = True
|
|
host = $DBHOST
|
|
database = $DBNAME
|
|
user = $DBUSER
|
|
password = $DBPASS
|
|
port = $DBPORT
|
|
reponame = $2
|
|
engine = $DBENGINE
|
|
|
|
[server]
|
|
preferuncompressed=True
|
|
uncompressed=True
|
|
|
|
[ui]
|
|
ssh=python "$TESTDIR/dummyssh"
|
|
EOF
|
|
}
|
|
|
|
function initclient() {
|
|
hg init $1
|
|
configureclient $1
|
|
}
|
|
|
|
configureclient() {
|
|
cat >> $1/.hg/hgrc <<EOF
|
|
[ui]
|
|
ssh=python "$TESTDIR/dummyssh"
|
|
|
|
[extensions]
|
|
hgsql=
|
|
strip=
|
|
EOF
|
|
}
|