#!/bin/bash

# Simple jobs that never fails
# Each should be taking 30-100s and be possible to run in parallel
# I.e.: No race conditions, no logins

par_sigterm() {
    echo '### Test SIGTERM'
    parallel -k -j5 sleep 15';' echo ::: {1..99} >/tmp/parallel$$ 2>&1 &
    A=$!
    sleep 29; kill -TERM $A
    wait
    sort /tmp/parallel$$
    rm /tmp/parallel$$
}

par_race_condition1() {
    echo '### Test race condition on 8 CPU (my laptop)'
    seq 1 5000000 > /tmp/parallel_race_cond
    seq 1 10 | parallel -k "cat /tmp/parallel_race_cond | parallel --pipe --recend '' -k gzip >/dev/null; echo {}"
    rm /tmp/parallel_race_cond
}

par_tmp_full() {
    # Assume /tmp/shm is easy to fill up
    export SHM=/tmp/shm/parallel
    mkdir -p $SHM
    sudo umount -l $SHM 2>/dev/null
    sudo mount -t tmpfs -o size=10% none $SHM

    echo "### Test --tmpdir running full. bug #40733 was caused by this"
    stdout parallel -j1 --tmpdir $SHM cat /dev/zero ::: dummy
}

par_memory_leak() {
    a_run() {
	seq $1 |time -v parallel true 2>&1 |
	grep 'Maximum resident' |
	field 6;
    }
    export -f a_run
    echo "### Test for memory leaks"
    echo "Of 100 runs of 1 job at least one should be bigger than a 3000 job run"
    small_max=$(seq 100 | parallel a_run 1 | jq -s max)
    big=$(a_run 3000)
    if [ $small_max -lt $big ] ; then
	echo "Bad: Memleak likely."
    else
	echo "Good: No memleak detected."	
    fi
}

par_linebuffer_matters_compress_tag() {
    echo "### (--linebuffer) --compress --tag should give different output"
    nolbfile=$(mktemp)
    lbfile=$(mktemp)
    controlfile=$(mktemp)
    randomfile=$(mktemp)
    # Random data because it does not compress well
    # forcing the compress tool to spit out compressed blocks
    head -c 10000000 /dev/urandom > $randomfile 

    testfunc() {
	linebuffer="$1"
	# Sleep 2 sec to give time to linebuffer-print the first part
	parallel -j0 $linebuffer --compress --tag \
		 "shuf $randomfile; sleep 2; shuf $randomfile; true" ::: {0..10} |
	    perl -ne '/^(\S+)\t/ and print "$1\n"' | uniq | sort
    }

    testfunc > $nolbfile &
    testfunc > $controlfile &
    testfunc --linebuffer > $lbfile &
    wait

    nolb="$(cat $nolbfile)"
    control="$(cat $controlfile)"
    lb="$(cat $lbfile)"
    rm $nolbfile $lbfile $controlfile $randomfile

    if [ "$nolb" == "$control" ] ; then
	if [ "$lb" == "$nolb" ] ; then
	    echo "BAD: --linebuffer makes no difference"
	else
	    echo "OK: --linebuffer makes a difference"
	fi
    else
	echo "BAD: control and nolb are not the same"
    fi
}

par_linebuffer_matters_compress() {
    echo "### (--linebuffer) --compress should give different output"
    random_data_with_id_prepended() {
	perl -pe 's/^/'$1'/' /dev/urandom |
	  pv -qL 300000 | head -c 1000000
    }
    export -f random_data_with_id_prepended

    nolb=$(seq 10 |
      parallel -j0 --compress random_data_with_id_prepended {#} |
      field 1 | uniq)
    lb=$(seq 10 |
      parallel -j0 --linebuffer --compress random_data_with_id_prepended {#} |
      field 1 | uniq)
    if [ "$lb" == "$nolb" ] ; then
	echo "BAD: --linebuffer makes no difference"
    else
	echo "OK: --linebuffer makes a difference"
    fi
}

par_memfree() {
    echo '### test memfree'
    parallel --memfree 1k echo Free mem: ::: 1k
    stdout parallel --timeout 20 --argsep II parallel --memfree 1t echo Free mem: ::: II 1t
}

export -f $(compgen -A function | grep par_)
compgen -A function | grep par_ | sort |
    parallel -j0 --tag -k --joblog /tmp/jl-`basename $0` '{} 2>&1'