ipfs-cluster/sharness/lib/test-lib.sh
2019-08-19 16:49:27 +05:30

149 lines
3.7 KiB
Bash
Executable File

#!/bin/bash
# Sharness test framework for ipfs-cluster
#
# We are using sharness (https://github.com/chriscool/sharness)
# which was extracted from the Git test framework.
SHARNESS_TEST_SRCDIR="lib/sharness/test"
SHARNESS_LIB="lib/sharness/sharness.sh"
# Daemons output will be redirected to...
IPFS_OUTPUT="/dev/null"
#IPFS_OUTPUT="/dev/stderr" # uncomment for debugging
. "$SHARNESS_LIB" || {
echo >&2 "Cannot source: $SHARNESS_LIB"
echo >&2 "Please check Sharness installation."
exit 1
}
which jq >/dev/null 2>&1
if [ $? -eq 0 ]; then
test_set_prereq JQ
fi
# Set prereqs
test_ipfs_init() {
which docker >/dev/null 2>&1
if [ $? -ne 0 ]; then
echo "Docker not found"
exit 1
fi
if docker ps --format '{{.Names}}' | egrep -q '^ipfs$'; then
echo "ipfs container already running"
else
docker run --name ipfs -d -p 127.0.0.1:5001:5001 ipfs/go-ipfs > /dev/null 2>&1
if [ $? -ne 0 ]; then
echo "IPFS init FAIL: Error running go-ipfs in docker."
exit 1
fi
while ! curl -s "localhost:5001/api/v0/version" > /dev/null; do
sleep 0.2
done
sleep 2
fi
test_set_prereq IPFS
}
ipfsCmd() {
docker exec -i ipfs ipfs $@
}
test_ipfs_running() {
if curl -s "localhost:5001/api/v0/version" > /dev/null; then
test_set_prereq IPFS
else
echo "IPFS is not running"
exit 1
fi
}
test_cluster_init() {
custom_config_files="$1"
consensus="$2"
if [ -z "$consensus" ]; then
consensus="crdt"
fi
which ipfs-cluster-service >/dev/null 2>&1
if [ $? -ne 0 ]; then
echo "cluster init FAIL: ipfs-cluster-service not found"
exit 1
fi
which ipfs-cluster-ctl >/dev/null 2>&1
if [ $? -ne 0 ]; then
echo "cluster init FAIL: ipfs-cluster-ctl not found"
exit 1
fi
ipfs-cluster-service --config "test-config" init --force --consensus "$consensus" >"$IPFS_OUTPUT" 2>&1
if [ $? -ne 0 ]; then
echo "cluster init FAIL: error on ipfs cluster init"
exit 1
fi
rm -rf "test-config/raft"
if [ -n "$custom_config_files" ]; then
cp -f ${custom_config_files}/* "test-config"
fi
cluster_start
}
test_cluster_config() {
# export CLUSTER_CONFIG_PATH="test-config/service.json"
export CLUSTER_IDENTITY_PATH="test-config/identity.json"
export CLUSTER_IDENTITY_ID=`jq --raw-output ".id" $CLUSTER_IDENTITY_PATH`
export CLUSTER_IDENTITY_PK=`jq --raw-output ".private_key" $CLUSTER_IDENTITY_PATH`
[ "$CLUSTER_IDENTITY_ID" != "null" ] && [ "$CLUSTER_IDENTITY_PK" != "null" ]
}
cluster_id() {
jq --raw-output ".id" test-config/identity.json
}
test_confirm_v1State() {
V1_SNAP_PATH="../test_data/v1State"
V1_CRC_PATH="../test_data/v1Crc"
if [ -f $V1_SNAP_PATH ] && [ -f $V1_CRC_PATH ]; then
export V1_CRC=$(cat ../test_data/v1Crc)
cp $V1_SNAP_PATH v1State
test_set_prereq V1STATE
fi
}
test_confirm_importState() {
IMP_STATE_PATH="../test_data/importState"
if [ -f $IMP_STATE_PATH ]; then
cp $IMP_STATE_PATH importState
test_set_prereq IMPORTSTATE
fi
}
cluster_kill(){
pkill -1 -f ipfs-cluster-service
while pgrep ipfs-cluster-service >/dev/null; do
sleep 0.2
done
}
cluster_start(){
ipfs-cluster-service --config "test-config" daemon >"$IPFS_OUTPUT" 2>&1 &
while ! curl -s 'localhost:9095/api/v0/version' >/dev/null; do
sleep 0.2
done
sleep 5 # wait for leader election
test_set_prereq CLUSTER
}
# Cleanup functions
test_clean_ipfs(){
docker kill ipfs >/dev/null
docker rm ipfs >/dev/null
sleep 1
}
test_clean_cluster(){
cluster_kill
rm -rf 'test-config'
}