test_single_connection_directory_download.sh

#!/bin/bash

# This test sets up a single transmission seeder and a cratetorrent leecher and
# asserts that cratetorrent downloads a small archive of multiple files from the
# seed correctly.
#
# assets/dir-test
# |-file1.txt: 100 KiB 
# |-file2.txt: 17 KiB 
# |-subdir
#   |-file3.txt: 6923 b
#   |-subdir2
#     |-file4.txt: 234 b

set -e

source common.sh

# start the container (if it's not already running)
./start_transmission_seed.sh --name "${seed_container}"

torrent_name=single-peer-dir-test
# the seeded file
src_path="${assets_dir}/${torrent_name}"
# and its metainfo
metainfo_path="${src_path}.torrent"
metainfo_cont_path="/cratetorrent/${torrent_name}.torrent"
# where we download the torrent (the same path is used on both the host and in
# the container)
download_dir=/tmp/cratetorrent
# the final download destination on the host
download_path="${download_dir}/${torrent_name}"
listen_addr=0.0.0.0:51234

# relative paths of the download resources
file1=file1.txt
file2=file2.txt
subdir1=subdir
file3="${subdir1}/file3.txt"
subdir2="${subdir1}/nested"
file4="${subdir2}/file4.txt"

################################################################################
# 1. Env setup
################################################################################

# top level
file1_path="${src_path}/${file1}"
file2_path="${src_path}/${file2}"
# first dir
subdir1_path="${src_path}/${subdir1}"
file3_path="${src_path}/${file3}"
# nested dir
subdir2_path="${src_path}/${subdir2}"
file4_path="${src_path}/${file4}"

# start seeding the torrent, if it doesn't exist yet
if [ ! -d "${src_path}" ]; then
    echo "Generating torrent ${torrent_name} directories and files"

    # create source directory and its subdirectories
    for dir in "${src_path}" "${subdir1_path}" "${subdir2_path}"; do
        if [ ! -d "${dir}" ]; then
            echo "Creating directory ${dir}"
            mkdir -p "${dir}"
        fi
    done

    # create source files
    file1_size=$(( 100 * 1024 )) # 100 KiB
    ./create_random_file.sh --path "${file1_path}" --size "${file1_size}"

    file2_size=$(( 17 * 1024 )) # 17 KiB
    ./create_random_file.sh --path "${file2_path}" --size "${file2_size}"

    file3_size=$(( 6923 )) # 6923 bytes
    ./create_random_file.sh --path "${file3_path}" --size "${file3_size}"

    file4_size=$(( 234 )) # 234 bytes
    ./create_random_file.sh --path "${file4_path}" --size "${file4_size}"

    # then start seeding it
    echo "Starting seeding of torrent ${torrent_name} seeding"
    ./seed_new_torrent.sh \
        --name "${torrent_name}" \
        --path "${src_path}" \
        --seed "${seed_container}"
fi

################################################################################
# 2. Download
################################################################################

./test_download.sh --torrent-name "${torrent_name}" \
    --src-path "${src_path}" \
    --download-dir "${download_dir}" \
    --metainfo-path "${metainfo_path}" \
    --listen-addr "${listen_addr}" \
    --seeds "${seed_container}"