path: root/camel/tests/folder/test3.c

/* folder/index testing */

#include "camel-test.h"
#include "messages.h"
#include "folders.h"

#include <camel/camel-exception.h>
#include <camel/camel-service.h>
#include <camel/camel-session.h>
#include <camel/camel-store.h>

#include <camel/camel-folder.h>
#include <camel/camel-folder-summary.h>
#include <camel/camel-mime-message.h>

#define ARRAY_LEN(x) (sizeof(x)/sizeof(x[0]))


/* god, who designed this horrid interface */
static char *auth_callback(CamelAuthCallbackMode mode,
               char *data, gboolean secret,
               CamelService *service, char *item,
               CamelException *ex)
{
    return NULL;
}


static void
test_folder_search_sub(CamelFolder *folder, const char *expr, int expected)
{
    CamelException *ex = camel_exception_new();
    GPtrArray *uids;
    GHashTable *hash;
    int i;

    uids = camel_folder_search_by_expression(folder, expr, ex);
    check(uids != NULL);
    check_msg(uids->len == expected, "search %s expected %d got %d", expr, expected, uids->len);
    check_msg(!camel_exception_is_set(ex), "%s", camel_exception_get_description(ex));

    /* check the uid's are actually unique, too */
    hash = g_hash_table_new(g_str_hash, g_str_equal);
    for (i=0;i<uids->len;i++) {
        check(g_hash_table_lookup(hash, uids->pdata[i]) == NULL);
        g_hash_table_insert(hash, uids->pdata[i], uids->pdata[i]);
    }
    g_hash_table_destroy(hash);

    camel_folder_search_free(folder, uids);

    camel_exception_free(ex);
}

static void
test_folder_search(CamelFolder *folder, const char *expr, int expected)
{
    char *matchall;

#if 0
    /* FIXME: ??? */
    camel_test_nonfatal("most searches require match-all construct");
    push("Testing search: %s", expr);
    test_folder_search_sub(folder, expr, expected);
    pull();
    camel_test_fatal();
#endif

    matchall = g_strdup_printf("(match-all %s)", expr);
    push("Testing search: %s", matchall);
    test_folder_search_sub(folder, matchall, expected);
    test_free(matchall);
    pull();
}

static struct {
    int counts[3];
    char *expr;
} searches[] = {
    { { 100, 50, 0 }, "(header-contains \"subject\" \"subject\")" },
    { { 0, 0, 0 }, "(header-contains \"subject\" \"Subject\")" },

    { { 100, 50, 0 }, "(body-contains \"content\")" },
    { { 100, 50, 0 }, "(body-contains \"Content\")" },

    { { 0, 0, 0 }, "(user-flag \"every7\")" },
    { { 100/13+1, 50/13+1, 0 }, "(user-flag \"every13\")" },
    { { 1, 1, 0 }, "(= \"7tag1\" (user-tag \"every7\"))" },
    { { 100/11+1, 50/11+1, 0 }, "(= \"11tag\" (user-tag \"every11\"))" },
    
    { { 100/13 + 100/17 + 1, 50/13 + 50/17 + 2, 0 }, "(user-flag \"every13\" \"every17\")" },
    { { 100/13 + 100/17 + 1, 50/13 + 50/17 + 2, 0 }, "(or (user-flag \"every13\") (user-flag \"every17\"))" },
    { { 1, 0, 0 }, "(and (user-flag \"every13\") (user-flag \"every17\"))" },
    
    { { 0, 0, 0 }, "(and (header-contains \"subject\" \"Test1\") (header-contains \"subject\" \"Test2\"))" },
    /* we get 11 here as the header-contains is a substring match */
    { { 11, 6, 0 }, "(and (header-contains \"subject\" \"Test1\") (header-contains \"subject\" \"subject\"))" },
    { { 1, 1, 0 }, "(and (header-contains \"subject\" \"Test19\") (header-contains \"subject\" \"subject\"))" },
    { { 0, 0, 0 }, "(and (header-contains \"subject\" \"Test191\") (header-contains \"subject\" \"subject\"))" },
    { { 1, 1, 0 }, "(and (header-contains \"subject\" \"Test1\") (header-contains \"subject\" \"message99\"))" },
    
    { { 22, 11, 0 }, "(or (header-contains \"subject\" \"Test1\") (header-contains \"subject\" \"Test2\"))" },
    { { 2, 1, 0 }, "(or (header-contains \"subject\" \"Test16\") (header-contains \"subject\" \"Test99\"))" },
    { { 1, 1, 0 }, "(or (header-contains \"subject\" \"Test123\") (header-contains \"subject\" \"Test99\"))" },
    { { 100, 50, 0 }, "(or (header-contains \"subject\" \"Test1\") (header-contains \"subject\" \"subject\"))" },
    { { 11, 6, 0 }, "(or (header-contains \"subject\" \"Test1\") (header-contains \"subject\" \"message99\"))" },
    
    /* 72000 is 24*60*100 == half the 'sent date' of the messages */
    { { 100/2, 50/2, 0 }, "(> 72000 (get-sent-date))" },
    { { 100/2-1, 50/2, 0 }, "(< 72000 (get-sent-date))" },
    { { 1, 0, 0 }, "(= 72000 (get-sent-date))" },
    { { 0, 0, 0 }, "(= 72001 (get-sent-date))" },
    
    { { (100/2-1)/17+1, (50/2-1)/17+1, 0 }, "(and (user-flag \"every17\") (< 72000 (get-sent-date)))" },
    { { (100/2-1)/17+1, (50/2-1)/17, 0 }, "(and (user-flag \"every17\") (> 72000 (get-sent-date)))" },
    { { (100/2-1)/13+1, (50/2-1)/13+1, 0 }, "(and (user-flag \"every13\") (< 72000 (get-sent-date)))" },
    { { (100/2-1)/13+1, (50/2-1)/13+1, 0 }, "(and (user-flag \"every13\") (> 72000 (get-sent-date)))" },
    
    { { 100/2+100/2/17, 50/2+50/2/17, 0 }, "(or (user-flag \"every17\") (< 72000 (get-sent-date)))" },
    { { 100/2+100/2/17+1, 50/2+50/2/17+1, 0 }, "(or (user-flag \"every17\") (> 72000 (get-sent-date)))" },
    { { 100/2+100/2/13, 50/2+50/2/13+1, 0 }, "(or (user-flag \"every13\") (< 72000 (get-sent-date)))" },
    { { 100/2+100/2/13+1, 50/2+50/2/13+1, 0 }, "(or (user-flag \"every13\") (> 72000 (get-sent-date)))" },
};

static void
run_search(CamelFolder *folder, int m)
{
    int i, j = 0;

    check(m == 50 || m == 100 || m == 0);

    /* *shrug* messy, but it'll do */
    if (m==50)
        j = 1;
    else if (m==0)
        j = 2;

    push("performing searches, expected %d", m);
    for (i=0;i<ARRAY_LEN(searches);i++) {
        push("running search %d: %s", i, searches[i].expr);
        test_folder_search(folder, searches[i].expr, searches[i].counts[j]);
        pull();
    }
    pull();
}


static char *stores[] = {
    "mbox:///tmp/camel-test/mbox",
    "mh:///tmp/camel-test/mh",
    "maildir:///tmp/camel-test/maildir"
};

int main(int argc, char **argv)
{
    CamelSession *session;
    CamelStore *store;
    CamelException *ex;
    CamelFolder *folder;
    CamelMimeMessage *msg;
    int i, j;
    int indexed;
    GPtrArray *uids;

    gtk_init(&argc, &argv);
    camel_test_init(argc, argv);

    ex = camel_exception_new();

    /* clear out any camel-test data */
    system("/bin/rm -rf /tmp/camel-test");

    session = camel_session_new("/tmp/camel-test", auth_callback, NULL, NULL);

    /* todo: cross-check everything with folder_info checks as well */
    /* todo: work out how to do imap/pop/nntp tests */

    /* we iterate over all stores we want to test, with indexing or indexing turned on or off */
    for (i=0;i<ARRAY_LEN(stores);i++) {
        char *name = stores[i];
        for (indexed = 0;indexed<2;indexed++) {
            char *what = g_strdup_printf("folder search: %s (%sindexed)", name, indexed?"":"non-");
            int flags;

            camel_test_start(what);
            test_free(what);

            push("getting store");
            store = camel_session_get_store(session, stores[i], ex);
            check_msg(!camel_exception_is_set(ex), "getting store: %s", camel_exception_get_description(ex));
            check(store != NULL);
            pull();

            push("creating %sindexed folder", indexed?"":"non-");
            if (indexed)
                flags = CAMEL_STORE_FOLDER_CREATE|CAMEL_STORE_FOLDER_BODY_INDEX;
            else
                flags = CAMEL_STORE_FOLDER_CREATE;
            folder = camel_store_get_folder(store, "testbox", flags, ex);
            check_msg(!camel_exception_is_set(ex), "%s", camel_exception_get_description(ex));
            check(folder != NULL);

            /* we need an empty folder for this to work */
            test_folder_counts(folder, 0, 0);
            pull();

            /* append a bunch of messages with specific content */
            push("appending 100 test messages");
            for (j=0;j<100;j++) {
                char *content, *subject;

                push("creating test message");
                msg = test_message_create_simple();
                content = g_strdup_printf("data%d content\n", j);
                test_message_set_content_simple((CamelMimePart *)msg, 0, "text/plain",
                                content, strlen(content));
                test_free(content);
                subject = g_strdup_printf("Test%d message%d subject", j, 100-j);
                camel_mime_message_set_subject(msg, subject);

                camel_mime_message_set_date(msg, j*60*24, 0);
                pull();

                push("appending simple message %d", j);
                camel_folder_append_message(folder, msg, NULL, ex);
                check_msg(!camel_exception_is_set(ex), "%s", camel_exception_get_description(ex));
                pull();

                test_free(subject);

                check_unref(msg, 1);
            }
            pull();

            push("Setting up some flags &c");
            uids = camel_folder_get_uids(folder);
            check(uids->len == 100);
            for (j=0;j<100;j++) {
                char *uid = uids->pdata[j];

                if ((j/13)*13 == j) {
                    camel_folder_set_message_user_flag(folder, uid, "every13", TRUE);
                }
                if ((j/17)*17 == j) {
                    camel_folder_set_message_user_flag(folder, uid, "every17", TRUE);
                }
                if ((j/7)*7 == j) {
                    char *tag = g_strdup_printf("7tag%d", j/7);
                    camel_folder_set_message_user_tag(folder, uid, "every7", tag);
                    test_free(tag);
                }
                if ((j/11)*11 == j) {
                    camel_folder_set_message_user_tag(folder, uid, "every11", "11tag");
                }
            }
            camel_folder_free_uids(folder, uids);
            pull();

            push("Search before sync");
            run_search(folder, 100);
            pull();

            push("syncing folder, searching");
            camel_folder_sync(folder, FALSE, ex);
            run_search(folder, 100);
            pull();

            push("syncing wiht expunge, search");
            camel_folder_sync(folder, TRUE, ex);
            run_search(folder, 100);
            pull();

            push("deleting every 2nd message");
            uids = camel_folder_get_uids(folder);
            check(uids->len == 100);
            for (j=0;j<uids->len;j+=2) {
                camel_folder_delete_message(folder, uids->pdata[j]);
            }
            camel_folder_free_uids(folder, uids);
            run_search(folder, 100);

            push("syncing");
            camel_folder_sync(folder, FALSE, ex);
            check_msg(!camel_exception_is_set(ex), "%s", camel_exception_get_description(ex));
            run_search(folder, 100);
            pull();

            push("expunging");
            camel_folder_expunge(folder, ex);
            check_msg(!camel_exception_is_set(ex), "%s", camel_exception_get_description(ex));
            run_search(folder, 50);
            pull();

            pull();

            push("closing and re-opening folder");
            check_unref(folder, 1);
            folder = camel_store_get_folder(store, "testbox", flags&~(CAMEL_STORE_FOLDER_CREATE), ex);
            check_msg(!camel_exception_is_set(ex), "%s", camel_exception_get_description(ex));
            check(folder != NULL);

            push("deleting remaining messages");
            uids = camel_folder_get_uids(folder);
            check(uids->len == 50);
            for (j=0;j<uids->len;j++) {
                camel_folder_delete_message(folder, uids->pdata[j]);
            }
            camel_folder_free_uids(folder, uids);
            run_search(folder, 50);

            push("syncing");
            camel_folder_sync(folder, FALSE, ex);
            check_msg(!camel_exception_is_set(ex), "%s", camel_exception_get_description(ex));
            run_search(folder, 50);
            pull();

            push("expunging");
            camel_folder_expunge(folder, ex);
            check_msg(!camel_exception_is_set(ex), "%s", camel_exception_get_description(ex));
            run_search(folder, 0);
            pull();

            pull();

            check_unref(folder, 1);
            pull();

            push("deleting test folder, with no messages in it");
            camel_store_delete_folder(store, "testbox", ex);
            check_msg(!camel_exception_is_set(ex), "%s", camel_exception_get_description(ex));
            pull();

            check_unref(store, 1);
            camel_test_end();
        }
    }

    check_unref(session, 1);
    camel_exception_free(ex);

    return 0;
}