]> git.notmuchmail.org Git - notmuch/blobdiff - performance-test/perf-test-lib.sh
perf-test: unpack tags.
[notmuch] / performance-test / perf-test-lib.sh
index e399d3f551ce6755653be4f1eb9f287bb53ac9ec..3a7ef2b35999506e7b3c47b0511ed94f9f0d4f4e 100644 (file)
@@ -35,44 +35,96 @@ then
        exit 1
 fi
 
+DB_CACHE_DIR=${TEST_DIRECTORY}/notmuch.cache.$corpus_size
+
 add_email_corpus ()
 {
     rm -rf ${MAIL_DIR}
 
-    case "$1" in
-       --small)
-           arg="mail/enron/bailey-s"
+    case "$corpus_size" in
+       small)
+           mail_subdir="mail/enron/bailey-s"
+           check_for="${TEST_DIRECTORY}/corpus/$mail_subdir"
            ;;
-       --medium)
-           arg="mail/notmuch-archive"
+       medium)
+           mail_subdir="mail/notmuch-archive"
+           check_for="${TEST_DIRECTORY}/corpus/$mail_subdir"
            ;;
        *)
-           arg=mail
+           mail_subdir=mail
+           check_for="${TEST_DIRECTORY}/corpus/$mail_subdir/enron/wolfe-j"
     esac
 
-    if command -v pixz > /dev/null; then
-       XZ=pixz
+    MAIL_CORPUS="${TEST_DIRECTORY}/corpus/$mail_subdir"
+    TAG_CORPUS="${TEST_DIRECTORY}/corpus/tags"
+
+    args=()
+    if [ ! -d "$TAG_CORPUS" ] ; then
+       args+=("notmuch-email-corpus/tags")
+    fi
+
+    if [ ! -d "$check_for" ] ; then
+       args+=("notmuch-email-corpus/$mail_subdir")
+    fi
+
+    if [[ ${#args[@]} > 0 ]]; then
+       if command -v pixz > /dev/null; then
+           XZ=pixz
+       else
+           XZ=xz
+       fi
+
+       printf "Unpacking corpus\n"
+       mkdir -p "${TEST_DIRECTORY}/corpus"
+
+       tar --checkpoint=.5000 --extract --strip-components=1 \
+           --directory ${TEST_DIRECTORY}/corpus \
+           --use-compress-program ${XZ} \
+           --file ../download/notmuch-email-corpus-${PERFTEST_VERSION}.tar.xz \
+           "${args[@]}"
+
+       printf "\n"
+
+    fi
+
+    cp -lr $TAG_CORPUS $TMP_DIRECTORY/corpus.tags
+    cp -lr $MAIL_CORPUS $MAIL_DIR
+
+}
+
+time_start () {
+
+    add_email_corpus
+
+    print_header
+
+    if [ -d $DB_CACHE_DIR ]; then
+       cp -r $DB_CACHE_DIR ${MAIL_DIR}/.notmuch
     else
-       XZ=xz
+       time_run 'Initial notmuch new' "notmuch new"
+       cache_database
     fi
+}
 
-    printf "Unpacking corpus\n"
-    tar --checkpoint=.5000 --extract --strip-components=1 \
-       --directory ${TMP_DIRECTORY} \
-       --use-compress-program ${XZ} \
-       --file ../download/notmuch-email-corpus-${PERFTEST_VERSION}.tar.xz \
-       notmuch-email-corpus/"$arg"
+cache_database () {
+    if [ -d $MAIL_DIR/.notmuch ]; then
+       cp -r $MAIL_DIR/.notmuch $DB_CACHE_DIR
+    else
+       echo "Warning: No database found to cache"
+    fi
+}
 
-    printf "\n"
+uncache_database () {
+    rm -rf $DB_CACHE_DIR
 }
 
 print_header () {
-    printf "[v%4s %6s]        Wall(s)\tUsr(s)\tSys(s)\tRes(K)\tIn/Out(512B)\n" \
+    printf "[v%4s %6s]          Wall(s)\tUsr(s)\tSys(s)\tRes(K)\tIn/Out(512B)\n" \
           ${PERFTEST_VERSION} ${corpus_size}
 }
 
 time_run () {
-    printf "%-22s" "$1"
+    printf "  %-22s" "$1"
     if test "$verbose" != "t"; then exec 4>test.output 3>&4; fi
     if ! eval >&3 "/usr/bin/time -f '%e\t%U\t%S\t%M\t%I/%O' $2" ; then
        test_failure=$(($test_failure + 1))
@@ -92,3 +144,6 @@ time_done () {
 
 cd -P "$test" || error "Cannot setup test environment"
 test_failure=0
+
+echo
+echo $(basename "$0"): "Testing ${test_description:-notmuch performance}"