path: root/camel/tests/folder/test9.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"
#include "camel/camel-filter-driver.h"
#include "camel/camel-stream-fs.h"

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

#define CAMEL_TEST_SESSION_TYPE     (camel_test_session_get_type ())
#define CAMEL_TEST_SESSION(obj)     (CAMEL_CHECK_CAST((obj), CAMEL_TEST_SESSION_TYPE, CamelTestSession))
#define CAMEL_TEST_SESSION_CLASS(k) (CAMEL_CHECK_CLASS_CAST ((k), CAMEL_TEST_SESSION_TYPE, CamelTestSessionClass))
#define CAMEL_TEST_IS_SESSION(o)    (CAMEL_CHECK_TYPE((o), CAMEL_TEST_SESSION_TYPE))

typedef struct _CamelTestSession {
    CamelSession parent_object;
    
} CamelTestSession;

typedef struct _CamelTestSessionClass {
    CamelSessionClass parent_class;
    
} CamelTestSessionClass;

static void
init (CamelTestSession *session)
{
    ;
}

static void
class_init (CamelTestSessionClass *camel_test_session_class)
{
    CamelSessionClass *camel_session_class =
        CAMEL_SESSION_CLASS (camel_test_session_class);
    
    /* virtual method override */
}

static CamelType
camel_test_session_get_type (void)
{
    static CamelType type = CAMEL_INVALID_TYPE;
    
    if (type == CAMEL_INVALID_TYPE) {
        type = camel_type_register (
            camel_test_session_get_type (),
            "CamelTestSession",
            sizeof (CamelTestSession),
            sizeof (CamelTestSessionClass),
            (CamelObjectClassInitFunc) class_init,
            NULL,
            (CamelObjectInitFunc) init,
            NULL);
    }
    
    return type;
}

static CamelSession *
camel_test_session_new (const char *path)
{
    CamelSession *session;
    
    session = CAMEL_SESSION (camel_object_new (CAMEL_TEST_SESSION_TYPE));
    
    camel_session_construct (session, path);
    
    return session;
}

struct {
    char *name;
    CamelFolder *folder;
} mailboxes[] = {
    { "INBOX", NULL },
    { "folder1", NULL },
    { "folder2", NULL },
    { "folder3", NULL },
    { "folder4", NULL },
};

struct {
    char *name, *match, *action;
} rules[] = {
    { "empty1", "(match-all (header-contains \"Frobnitz\"))", "(copy-to \"folder1\")" },
    { "empty2", "(header-contains \"Frobnitz\")", "(copy-to \"folder2\")" },
    { "count11", "(and (header-contains \"subject\" \"Test1\") (header-contains \"subject\" \"subject\"))", "(move-to \"folder3\")" },
    { "empty3", "(and (header-contains \"subject\" \"Test1\") (header-contains \"subject\" \"subject\"))", "(move-to \"folder4\")" },
    { "count1", "(body-contains \"data50\")", "(copy-to \"folder1\")" },
    { "stop", "(body-contains \"data2\")", "(stop)" },
    { "notreached1", "(body-contains \"data2\")", "(move-to \"folder2\")" },
    { "count1", "(body-contains \"data3\")", "(move-to \"folder2\")" },
    { "ustrcasecmp", "(header-matches \"Subject\" \"Test0 message100 subject\")", "(copy-to \"folder2\")" },
};

/* broken match rules */
struct {
    char *name, *match, *action;
} brokens[] = {
    { "count1", "(body-contains data50)", "(copy-to \"folder1\")" }, /* non string argument */
    { "count1", "(body-contains-stuff \"data3\")", "(move-to-folder \"folder2\")" }, /* invalid function */
    { "count1", "(or (body-contains \"data3\") (foo))", "(move-to-folder \"folder2\")" }, /* invalid function */
    { "count1", "(or (body-contains \"data3\") (foo)", "(move-to-folder \"folder2\")" }, /* missing ) */
    { "count1", "(and body-contains \"data3\") (foo)", "(move-to-folder \"folder2\")" }, /* missing ( */
    { "count1", "body-contains \"data3\")", "(move-to-folder \"folder2\")" }, /* missing ( */
    { "count1", "body-contains \"data3\"", "(move-to-folder \"folder2\")" }, /* missing ( ) */
    { "count1", "(body-contains \"data3\" ())", "(move-to-folder \"folder2\")" }, /* extra () */
    { "count1", "()", "(move-to-folder \"folder2\")" }, /* invalid () */
    { "count1", "", "(move-to-folder \"folder2\")" }, /* empty */
};

/* broken action rules */
struct {
    char *name, *match, *action;
} brokena[] = {
    { "a", "(body-contains \"data2\")", "(body-contains \"help\")" }, /* rule in action */
    { "a", "(body-contains \"data2\")", "(move-to-folder-name \"folder2\")" }, /* unknown function */
    { "a", "(body-contains \"data2\")", "(or (move-to-folder \"folder2\")" }, /* missing ) */
    { "a", "(body-contains \"data2\")", "(or move-to-folder \"folder2\"))" }, /* missing ( */
    { "a", "(body-contains \"data2\")", "move-to-folder \"folder2\")" }, /* missing ( */
    { "a", "(body-contains \"data2\")", "(move-to-folder \"folder2\" ())" }, /* invalid () */
    { "a", "(body-contains \"data2\")", "()" }, /* invalid () */
    { "a", "(body-contains \"data2\")", "" }, /* empty */
};

static CamelFolder *get_folder(CamelFilterDriver *d, const char *uri, void *data, CamelException *ex)
{
    int i;

    for (i=0;i<ARRAY_LEN(mailboxes);i++)
        if (!strcmp(mailboxes[i].name, uri)) {
            camel_object_ref((CamelObject *)mailboxes[i].folder);
            return mailboxes[i].folder;
        }
    return NULL;
}

int main(int argc, char **argv)
{
    CamelSession *session;
    CamelStore *store;
    CamelException *ex;
    CamelFolder *folder;
    CamelMimeMessage *msg;
    int i, j;
    CamelStream *mbox;
    CamelFilterDriver *driver;

    /*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");

    camel_test_start("Simple filtering of mbox");

    session = camel_test_session_new ("/tmp/camel-test");

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

    push("getting store");
    store = camel_session_get_store(session, "mbox:///tmp/camel-test/mbox", ex);
    check_msg(!camel_exception_is_set(ex), "getting store: %s", camel_exception_get_description(ex));
    check(store != NULL);
    pull();

    push("Creating output folders");
    for (i=0;i<ARRAY_LEN(mailboxes);i++) {
        push("creating %s", mailboxes[i].name);
        mailboxes[i].folder = folder = camel_store_get_folder(store, mailboxes[i].name, CAMEL_STORE_FOLDER_CREATE, 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();
    }
    pull();

    /* append a bunch of messages with specific content */
    push("creating 100 test message mbox");
    mbox = camel_stream_fs_new_with_name("/tmp/camel-test/inbox", O_WRONLY|O_CREAT|O_EXCL, 0600);
    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();
        
        camel_stream_printf(mbox, "From \n");
        check(camel_data_wrapper_write_to_stream((CamelDataWrapper *)msg, mbox) != -1);
#if 0       
        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();
#endif              
        test_free(subject);
        
        check_unref(msg, 1);
    }
    check(camel_stream_close(mbox) != -1);
    check_unref(mbox, 1);
    pull();

    push("Building filters");
    driver = camel_filter_driver_new(get_folder, NULL);
    for (i=0;i<ARRAY_LEN(rules);i++) {
        camel_filter_driver_add_rule(driver, rules[i].name, rules[i].match, rules[i].action);
    }
    pull();

    push("Executing filters");
    camel_filter_driver_set_default_folder(driver, mailboxes[0].folder);
    camel_filter_driver_filter_mbox(driver, "/tmp/camel-test/inbox", ex);
    check_msg(!camel_exception_is_set(ex), "%s", camel_exception_get_description(ex));

    /* now need to check the folder counts/etc */

    check_unref(driver, 1);
    pull();

    /* this tests that invalid rules are caught */
    push("Testing broken match rules");
    for (i=0;i<ARRAY_LEN(brokens);i++) {
        push("rule %s", brokens[i].match);
        driver = camel_filter_driver_new(get_folder, NULL);
        camel_filter_driver_add_rule(driver, brokens[i].name, brokens[i].match, brokens[i].action);
        camel_filter_driver_filter_mbox(driver, "/tmp/camel-test/inbox", ex);
        check(camel_exception_is_set(ex));
        camel_exception_clear(ex);
        check_unref(driver, 1);
        pull();
    }
    pull();

    push("Testing broken action rules");
    for (i=0;i<ARRAY_LEN(brokena);i++) {
        push("rule %s", brokena[i].action);
        driver = camel_filter_driver_new(get_folder, NULL);
        camel_filter_driver_add_rule(driver, brokena[i].name, brokena[i].match, brokena[i].action);
        camel_filter_driver_filter_mbox(driver, "/tmp/camel-test/inbox", ex);
        check(camel_exception_is_set(ex));
        camel_exception_clear(ex);
        check_unref(driver, 1);
        pull();
    }
    pull();


    for (i=0;i<ARRAY_LEN(mailboxes);i++) {
        check_unref(mailboxes[i].folder, 1);
    }

    check_unref(store, 1);

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

    camel_test_end();

    return 0;
}