/* $Id$ 
 * 
 * Licensed to the Apache Software Foundation (ASF) under one or more 
 * contributor license agreements. See the NOTICE file distributed with  
 * this work for additional information regarding copyright ownership. 
 * The ASF licenses this file to you under the Apache License, Version  
 * 2.0 (the "License"); you may not use this file except in compliance  
 * with the License. You may obtain a copy of the License at 
 * 
 * http://www.apache.org/licenses/LICENSE-2.0 
 * 
 * Unless required by applicable law or agreed to in writing, software 
 * distributed under the License is distributed on an "AS IS" BASIS, 
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and 
 * limitations under the License. 
 */ 

/*
 * test_threadpool.c
 */

#include "apu.h"
#include "etch_apr.h"   /* must be included first */
#include "etchthread.h" /* must be included second */
#include "etchmutex.h"

#include <tchar.h>
#include <stdio.h>
#include <conio.h>

#include "cunit.h"
#include "basic.h"
#include "automated.h"

#include "etch_global.h"
#include "etch_arraylist.h"

int init_suite(void)
{
    return etch_runtime_init_all(0); 
}

int clean_suite(void)
{
    return etch_runtime_exit(0);
}

#define IS_DEBUG_CONSOLE TRUE
int g_is_automated_test, g_bytes_allocated;

etchmutex* mutex_threadid;
int g_next_threadid;

#define IS_DUMP_AFTER_REMOVE FALSE
#define IS_TESTING_REMOVE    TRUE
#define THREAD_TESTDATA_SIG 0xfeedface
struct usertestdata { int x; int y; };

int next_threadid()
{   mutex_threadid->acquire(mutex_threadid);
    ++g_next_threadid;
    mutex_threadid->release(mutex_threadid);
    return g_next_threadid;
}

/*
 * get_threaddata()
 * get a handle to internal thread create parameters
 */
etch_apr_threaddata* get_threaddata()
{
    static etch_apr_threaddata tdata;
    memset(&tdata, 0, sizeof(etch_apr_threaddata));
    tdata.mempool = etch_apr_mempool;
    return &tdata;
}


/**
 * on_threadstart()
 * thread start callback
 */
int on_threadstart(void* param)
{
    etch_threadparams* params = (etch_threadparams*) param;
    #if (IS_DEBUG_CONSOLE)
    if  (NULL == param) 
         printf("pool thread start\n");  
    else printf("pool thread %04x start\n", params->etch_thread_id); 
    fflush(stdout);
    #endif
    return 0;
}


/**
 * on_threadexit()
 * thread stop callback
 */
int on_threadexit(void* param)
{
    etch_threadparams* params = (etch_threadparams*) param;    
    #if (IS_DEBUG_CONSOLE)
    if  (NULL == param)
         printf("pool thread exit\n");  
    else printf("pool thread %04x exit\n", params->etch_thread_id); 
    fflush(stdout);
    #endif
    return 0;
}


/**
 * createthread_threadproc
 */
void createthread_threadproc(void* data)
{
    etch_threadparams* params;
    struct usertestdata* ud; 
    #if (IS_DEBUG_CONSOLE)
    printf("in createthread_threadproc\n"); 
    #endif

    params = (etch_threadparams*) data;
    ud = (struct usertestdata*) params->data;
    CU_ASSERT_PTR_NOT_NULL_FATAL(ud);
    CU_ASSERT_EQUAL(ud->x, 1);
    CU_ASSERT_EQUAL(ud->y, 2);
}


/**
 * test_createthread
 */
void test_createthread(void)
{
    int result = 0;
    etch_threadparams tp;
    struct usertestdata  ud;
    etch_apr_threaddata* td = get_threaddata();
    memset(&tp, 0, sizeof(etch_threadparams));
    ud.x = 1; ud.y = 2;
    mutex_threadid = new_mutex(etch_apr_mempool, ETCHMUTEX_NESTED);

    etch_init_threadparams(&tp);
    tp.etch_thread_id = next_threadid();
    tp.data       = &ud;
    tp.libdata    = td;
    tp.on_start   = on_threadstart;
    tp.on_exit    = on_threadexit;
    tp.threadproc = createthread_threadproc;
    #if (IS_DEBUG_CONSOLE)
    printf("\n");
    #endif
    
    result = etch_createthread(&tp);    
    CU_ASSERT_EQUAL(result, 0);

    etch_thread_join(&tp);  /* block until thread exit */
    CU_ASSERT_EQUAL(result, 0);

    mutex_threadid->destroy(mutex_threadid);

    g_bytes_allocated = etch_showmem(0, IS_DEBUG_CONSOLE);  
    CU_ASSERT_EQUAL(g_bytes_allocated, 0);  
    memtable_clear();  /* start fresh for next test */   
}


/**
 * mutextestparams
 * thread data for mutex test
 */
struct mutextestparams
{
    unsigned signature;
    etchmutex* mutex;
    unsigned flag;
};


/**
 * mutex_threadproc_a
 */
void mutex_threadproc_a(void* data)
{
    int result = 0;
    etch_threadparams* params;
    struct mutextestparams* p;
    params = (etch_threadparams*) data;
    p = (struct mutextestparams*) params->data;
    #if (IS_DEBUG_CONSOLE)
    printf("in threadproc_a\n"); fflush(stdout);
    #endif

    /* can't use a FATAL cunit macro outside the main thread 
     * since if it fails the exit handling causes an OS fault */

    CU_ASSERT_PTR_NOT_NULL(p); if (!p) return;
    CU_ASSERT_EQUAL(p->signature, THREAD_TESTDATA_SIG);
    if (p->signature != THREAD_TESTDATA_SIG) return;
    CU_ASSERT_PTR_NOT_NULL(p->mutex); if (!p->mutex) return;

    #if (IS_DEBUG_CONSOLE)
    printf("thread a waiting ...\n"); fflush(stdout);
    #endif

    while(p->flag == 0) /* wait until ready to acquire lock */   
          apr_sleep(1 * 1000 * 1000);

    CU_ASSERT_EQUAL(p->flag, 1); if (p->flag != 1) return;

    result = p->mutex->acquire(p->mutex); /* get lock */
    CU_ASSERT_EQUAL(result, 0); if (result != 0) return;

    p->flag = 2; /* signal thread b to attempt acquire */

    #if (IS_DEBUG_CONSOLE)
    printf("thread a waiting ...\n"); fflush(stdout);
    #endif

    while(p->flag == 2) /* wait for thread thread b signal */
          apr_sleep(1 * 1000 * 1000);

    result = p->mutex->release(p->mutex); /* release lock */
    CU_ASSERT_EQUAL(result, 0);

    p->flag = 4; /* signal thread b to proceed */

    #if (IS_DEBUG_CONSOLE)
    printf("thread a waiting ...\n"); fflush(stdout);
    #endif

    while(p->flag == 4) /* wait until thread b signals done */   
          apr_sleep(1 * 1000 * 1000);

    #if (IS_DEBUG_CONSOLE)
    printf("thread s exits\n"); fflush(stdout);
    #endif
}


/**
 * mutex_threadproc_b
 */
void mutex_threadproc_b(void* data)
{
    int result = 0;
    etch_threadparams* params;
    struct mutextestparams* p;
    params = (etch_threadparams*) data;
    p = (struct mutextestparams*) params->data;
    #if (IS_DEBUG_CONSOLE)
    printf("in mutex_threadproc_b\n"); fflush(stdout);
    #endif

    CU_ASSERT_PTR_NOT_NULL(p);
    CU_ASSERT_EQUAL(p->signature, THREAD_TESTDATA_SIG);
    CU_ASSERT_PTR_NOT_NULL(p->mutex);

    #if (IS_DEBUG_CONSOLE)
    printf("thread b waiting ...\n"); fflush(stdout);
    #endif

    while(p->flag != 2) /* wait until ready to attempt lock */   
          apr_sleep(1 * 1000 * 1000);
    
    result = p->mutex->tryacquire(p->mutex); /* try to get lock */
    CU_ASSERT_NOT_EQUAL(result, 0); /* should fail */

    p->flag = 3; /* signal thread a to continue */

    #if (IS_DEBUG_CONSOLE)
    printf("thread b waiting ...\n"); fflush(stdout);
    #endif

    while(p->flag == 3) /* wait for thread a to release lock */   
          apr_sleep(1 * 1000 * 1000);

    result = p->mutex->tryacquire(p->mutex); /* try to get lock */
    CU_ASSERT_EQUAL(result, 0); /* should succeed now */

    result = p->mutex->release(p->mutex); /* release lock */
    CU_ASSERT_EQUAL(result, 0);

    p->flag = 5; /* signal thread a that we are done here */

    #if (IS_DEBUG_CONSOLE)
    printf("thread b waiting 2 secs ...\n"); fflush(stdout);
    #endif

    /* ensure thread b is last to exit for simplicity */
    apr_sleep(2 * 1000 * 1000); 

    #if (IS_DEBUG_CONSOLE)
    printf("thread_b exits\n"); fflush(stdout);
    #endif
}


/**
 * test_mutex()
 * launches two threads which share a mutex and a signal flag with the main thread.
 * 1. thread a sets the lock, signals thread b, and waits.
 * 2. thread b tries to set the lock, fails as expected, signals thread a, and waits.
 * 3. thread a releases the lock, signals thread b, and waits.
 * 4. thread b tries the lock, succeeds as expected, releases the lock, signals thread a.
 * 5. thread a exits
 * 6. thread b exits
 */
void test_mutex(void)
{
    int result = 0;
    etchmutex* mutex = 0;
    etch_apr_threaddata* td = get_threaddata();
    etch_threadparams tpa, tpb;
    struct mutextestparams threadparams;
    etch_init_threadparams(&tpa);
    etch_init_threadparams(&tpb);
    mutex_threadid = new_mutex(etch_apr_mempool, ETCHMUTEX_NESTED);

    mutex = new_mutex(td->mempool, ETCHMUTEX_NESTED);
    CU_ASSERT_PTR_NOT_NULL_FATAL(mutex);

    result = mutex->acquire(mutex);
    CU_ASSERT_EQUAL_FATAL(result, 0);
    // result = mutex->tryacquire(mutex); /* mutexes are nestable */
    // CU_ASSERT_NOT_EQUAL_FATAL(result, 0);
    result = mutex->release(mutex);
    CU_ASSERT_EQUAL_FATAL(result, 0);

    threadparams.mutex = mutex;
    threadparams.flag  = 0;
    threadparams.signature = THREAD_TESTDATA_SIG;

    tpa.etch_thread_id = next_threadid();
    tpa.libdata    = td;
    tpa.on_start   = on_threadstart;
    tpa.on_exit    = on_threadexit;
    tpa.threadproc = mutex_threadproc_a;  
    tpa.data = &threadparams;

    #if (IS_DEBUG_CONSOLE)
    printf("\n");
    #endif

    result = etch_createthread(&tpa); /* create thread a */
    CU_ASSERT_EQUAL(result, 0);

    memcpy(&tpb, &tpa, sizeof(etch_threadparams));
    tpb.etch_thread_id = next_threadid();
    tpb.threadproc = mutex_threadproc_b;

    result = etch_createthread(&tpb); /* create thread b */   
    CU_ASSERT_EQUAL(result, 0);

    /* signal the new threads to commence doing something */
    threadparams.flag = 1;

    etch_thread_join(&tpa);  /* wait for thread a exit */
    etch_thread_join(&tpb);  /* wait for thread b exit */
    
    mutex->destroy(mutex);
    mutex_threadid->destroy(mutex_threadid);

    g_bytes_allocated = etch_showmem(0, IS_DEBUG_CONSOLE);  
    CU_ASSERT_EQUAL(g_bytes_allocated, 0);  
    memtable_clear();  /* start fresh for next test */   
}


/**
 * synchedlist_testparams
 * thread data for synched list test
 */
struct synchedlist_testparams
{
    unsigned signature;
    etch_arraylist* list;
    etchmutex* mutex;
    unsigned flag;
};


void dumplist(etch_arraylist* list)
{
    int i=0; char c=0, result;
    const int n = list->count;
    result = arraylist_trylock(list);
    if (result != 0) return;
    // printf("\ndumping list ...\n"); fflush(stdout);
    for(; i < n; i++)
    {   etch_int32* item = list->base[i];
        printf("\n list[%d] %d\n", i, item->value); fflush(stdout);
    }

    result = arraylist_rellock(list);
    // printf("\nany key ..."); while(!c) c = _getch(); printf("\n"); 
}


/* 
 * synched_arraylist_comparator()
 * comparator to compare an integer constant with an etch_int32.
 */
int synched_arraylist_comparator(void* testvalue, void* listcontent)
{
    const int ivalue  = (int) (size_t) testvalue;
    etch_int32*  listobj = (etch_int32*) listcontent;
    int jvalue = listobj->value;
    return ivalue < jvalue? -1: ivalue > jvalue? 1: 0;
} 


/**
 * synchedlist_threadproc_a
 */
void synchedlist_threadproc_a(void* data)
{
    int result = 0, i;
    etch_iterator iterator;
    etch_threadparams* params;
    struct synchedlist_testparams* p; 
    params = (etch_threadparams*) data;
    p = (struct synchedlist_testparams*) params->data;
    #if (IS_DEBUG_CONSOLE)
    printf("in threadproc_a\n"); fflush(stdout);
    #endif

    /* can't use a FATAL cunit macro outside the main thread 
     * since if it fails the exit handling causes an OS fault */

    CU_ASSERT_PTR_NOT_NULL(p); if (!p) return;
    CU_ASSERT_EQUAL(p->signature, THREAD_TESTDATA_SIG);
    if (p->signature != THREAD_TESTDATA_SIG) return;
    CU_ASSERT_PTR_NOT_NULL(p->list); if (!p->list) return;

    #if (IS_DEBUG_CONSOLE)
    printf("thread a waiting ...\n"); fflush(stdout);
    #endif

    while(p->flag == 0) /* wait until main signals start */   
          apr_sleep(1 * 500 * 1000);

    CU_ASSERT_EQUAL(p->flag, 1); if (p->flag != 1) return;

    /* both threads should be contending for the list now */
    for(i = 1; i <= 20; i++)
    {
        #if (IS_DEBUG_CONSOLE)
        printf("thread a arraylist_add ...\n"); fflush(stdout);
        #endif
        arraylist_add(p->list, new_int32(i));
        etch_thread_yield();        
    }

    p->flag++;
    while(p->flag < 3) /* wait until both threads are done adding to list */   
          apr_sleep(1 * 1000 * 500);

    CU_ASSERT_EQUAL(p->list->count, 40); 
    p->flag++;
    while(p->flag < 5) /* wait until both threads are done checking list */   
          apr_sleep(1 * 1000 * 500);

    #if IS_TESTING_REMOVE

    /* both threads should be contending for the list now */
    for(i = 4; i <= 20; i += 4)
    {
        /* get the list index for value i */
        const int index = arraylist_indexof(p->list, (void*)(size_t) i, 0, synched_arraylist_comparator);
        CU_ASSERT_NOT_EQUAL(index, -1);

        if (index >= 0)
        {
            result = arraylist_remove(p->list, index, TRUE);
            CU_ASSERT_EQUAL(result, 0);

            #if (IS_DEBUG_CONSOLE)
            if  (result == 0)
                 printf("thread a arraylist_remove index %d value %d\n", index, i);
            else printf("thread a could not remove index %d value %d\n", index, i);
            fflush(stdout);
            #endif 
        }

        etch_thread_yield();        
    }

    #endif /* IS_TESTING_REMOVE */

    p->flag++;
    while(p->flag < 7) /* wait until both threads are done removing from list */   
          apr_sleep(1 * 1000 * 500);

    #if IS_TESTING_REMOVE
    CU_ASSERT_EQUAL(p->list->count, 30); /* each thread removed 5 items */
    #endif

    p->flag++;
    while(p->flag < 9) /* wait until both threads are done checking list */   
          apr_sleep(1 * 1000 * 500);

    set_iterator(&iterator, p->list, &p->list->iterable);

    result = arraylist_getlock(p->list);
    CU_ASSERT_EQUAL(result, 0); 

    #if (IS_DEBUG_CONSOLE)
    printf("thread a iterating list ...\n"); fflush(stdout);
    #endif  

    while(iterator.has_next(&iterator))
    {
        etch_int32* item = (etch_int32*) iterator.current_value;
        CU_ASSERT_PTR_NOT_NULL(item); 

        if (item)  
        {   result = item->value;
            #if (IS_DEBUG_CONSOLE)
            printf("thread a list value %d\n", result); fflush(stdout);
            #endif  
            #if IS_TESTING_REMOVE
            CU_ASSERT_NOT_EQUAL(result % 4, 0); /* we removed all modulo 4 values */
            #endif
        }
        
        result = iterator.next(&iterator);
    }

    etch_sleep(1000);
    result = arraylist_rellock(p->list);

    p->flag++;
    #if (IS_DEBUG_CONSOLE)
         printf("thread a waiting to exit ...\n"); fflush(stdout);
    #endif 

    while(p->flag < 11) /* wait until both threads are done iterating list */  
          apr_sleep(1 * 1000 * 500);

    #if (IS_DEBUG_CONSOLE)
    printf("thread a exits\n"); fflush(stdout);
    #endif
}


/**
 * synchedlist_threadproc_b
 */
void synchedlist_threadproc_b(void* data)
{
    int result = 0, i;
    etch_iterator iterator;
    etch_threadparams* params;
    struct synchedlist_testparams* p; 
    params = (etch_threadparams*) data;
    p = (struct synchedlist_testparams*) params->data;
    #if (IS_DEBUG_CONSOLE)
    printf("in threadproc_b\n"); fflush(stdout);
    #endif

    /* can't use a FATAL cunit macro outside the main thread 
     * since if it fails the exit handling causes an OS fault */

    CU_ASSERT_PTR_NOT_NULL(p); if (!p) return;
    CU_ASSERT_EQUAL(p->signature, THREAD_TESTDATA_SIG);
    if (p->signature != THREAD_TESTDATA_SIG) return;
    CU_ASSERT_PTR_NOT_NULL(p->list); if (!p->list) return;

    #if (IS_DEBUG_CONSOLE)
    printf("thread b waiting ...\n"); fflush(stdout);
    #endif

    while(p->flag == 0) /* wait until main signals start */   
          apr_sleep(1 * 500 * 1000);

    CU_ASSERT_EQUAL(p->flag, 1); if (p->flag != 1) return;

    /* both threads should be contending for the list now */
    for(i = 21; i <= 40; i++)
    {
        #if (IS_DEBUG_CONSOLE)
        printf("thread b arraylist_add ...\n"); fflush(stdout);
        #endif
        arraylist_add(p->list, new_int32(i));
        etch_thread_yield();        
    }

    p->flag++;
    while(p->flag < 3) /* wait until both threads are done adding to list */   
          apr_sleep(1 * 1000 * 500);

    CU_ASSERT_EQUAL(p->list->count, 40); 
    p->flag++;
    while(p->flag < 5) /* wait until both threads are done checking list */   
          apr_sleep(1 * 1000 * 500);

    #if IS_TESTING_REMOVE

    /* both threads should be contending for the list now */
    for(i = 24; i <= 40; i += 4)
    {
        const int index = arraylist_indexof(p->list, (void*)(size_t) i, 0, synched_arraylist_comparator);
        CU_ASSERT_NOT_EQUAL(index, -1);

        if (index >= 0)
        {
            result = arraylist_remove(p->list, index, TRUE);
            CU_ASSERT_EQUAL(result, 0);

            #if (IS_DEBUG_CONSOLE)
            if  (result == 0)
                 printf("thread b arraylist_remove index %d value %d\n", index, i);
            else printf("thread b could not remove index %d value %d\n", index, i);
            fflush(stdout);
            #endif 
        }
            
        etch_thread_yield();        
    }

    #endif /*  #if IS_TESTING_REMOVE */

    p->flag++;
    while(p->flag < 7) /* wait until both threads are done removing from list */   
          apr_sleep(1 * 1000 * 500);

    #if IS_TESTING_REMOVE
    CU_ASSERT_EQUAL(p->list->count, 30); /* each thread removed 5 items */

    #if (IS_DEBUG_CONSOLE)
    printf("thread b list count after remove is %d\n", p->list->count); fflush(stdout);
    #endif /* IS_DEBUG_CONSOLE */
    #endif /* IS_TESTING_REMOVE */

    #if (IS_DEBUG_CONSOLE)
    printf("thread b waiting for list lock ...\n"); fflush(stdout);
    #endif /* IS_DEBUG_CONSOLE */
    p->flag++;
    while(p->flag < 9) /* wait until both threads are done checking list */   
          apr_sleep(1 * 1000 * 500);

    set_iterator(&iterator, p->list, &p->list->iterable);  

    result = arraylist_getlock(p->list);
    CU_ASSERT_EQUAL(result, 0); 

    #if (IS_DEBUG_CONSOLE)
    printf("thread b iterating list ...\n"); fflush(stdout);
    #endif  /* IS_DEBUG_CONSOLE */ 

    while(iterator.has_next(&iterator))
    {   
        etch_int32* item = (etch_int32*) iterator.current_value;
        CU_ASSERT_PTR_NOT_NULL(item); 

        if (item)  
        {   result = item->value;
            #if (IS_DEBUG_CONSOLE)
            printf("thread b list value %d\n", result); fflush(stdout);
            #endif  
            #if IS_TESTING_REMOVE
            CU_ASSERT_NOT_EQUAL(result % 4, 0); /* we removed all modulo 4 values */
            #endif
        }
        
        iterator.next(&iterator);
    }

    result = arraylist_rellock(p->list);

    #if (IS_DEBUG_CONSOLE)
         printf("thread b waiting to exit ...\n"); fflush(stdout);
    #endif 

    p->flag++;
    while(p->flag < 11) /* wait until both threads are done iterating list */   
          apr_sleep(1 * 1000 * 500);

    /* ensure thread b is last to exit for simplicity */
    apr_sleep(1 * 1000 * 1000); 

    #if (IS_DEBUG_CONSOLE)
    printf("thread b exits\n"); fflush(stdout);
    #endif
}


/**
 * test_synched_arraylist()
 * launches two threads which share a synchronized arraylist.
 */
void test_synched_arraylist(void)
{
    int result = 0;
    etchmutex* mutex = 0;
    etch_arraylist* list = 0;
    struct synchedlist_testparams tparams;
    etch_apr_threaddata* td = get_threaddata();
    etch_threadparams tpa, tpb;
    memset(&tpa, 0, sizeof(etch_threadparams));
    memset(&tpb, 0, sizeof(etch_threadparams));
    mutex_threadid = new_mutex(etch_apr_mempool, ETCHMUTEX_NESTED);

    mutex = new_mutex(td->mempool, ETCHMUTEX_NESTED);    
    CU_ASSERT_PTR_NOT_NULL_FATAL(mutex);

    tparams.list  = new_arraylist(0,0);
    tparams.list->synclock = mutex; /* list now owns the mutex */
    tparams.list->synchook = etchmutex_hookproc;
    tparams.flag  = 0;
    tparams.signature = THREAD_TESTDATA_SIG;

    tpa.etch_thread_id = ++g_next_threadid;
    tpa.libdata    = td;
    tpa.on_start   = on_threadstart;
    tpa.on_exit    = on_threadexit;
    tpa.threadproc = synchedlist_threadproc_a;  
    tpa.data = &tparams;

    #if (IS_DEBUG_CONSOLE)
    printf("\n");
    #endif

    result = etch_createthread(&tpa); /* create thread a */
    CU_ASSERT_EQUAL(result, 0);

    memcpy(&tpb, &tpa, sizeof(etch_threadparams));
    tpb.etch_thread_id = ++g_next_threadid;
    tpb.threadproc = synchedlist_threadproc_b;

    result = etch_createthread(&tpb); /* create thread b */   
    CU_ASSERT_EQUAL(result, 0);

    /* signal both threads to start doing something */
    tparams.flag = 1;

    etch_thread_join(&tpb);  /* wait for thread b exit */
    etch_thread_join(&tpa);  /* wait for thread a exit */
    
    tparams.list->destroy(tparams.list); /* destroy list and mutex */

    mutex_threadid->destroy(mutex_threadid);

    g_bytes_allocated = etch_showmem(0, IS_DEBUG_CONSOLE);  
    CU_ASSERT_EQUAL(g_bytes_allocated, 0);  
    memtable_clear();  /* start fresh for next test */   
}


typedef struct waittestdata { int flag; int num; int is_disposable; } waittestdata;


/**
 * wait_threadproc_a
 */
void wait_threadproc_a(void* data)
{
    int result = 0;
    etch_threadparams* params;
    waittestdata* threadtestdata; 
    params = (etch_threadparams*) data;
    threadtestdata = (waittestdata*) params->data;

    #if (IS_DEBUG_CONSOLE)
    printf("\nin wait thread_a\n"); fflush(stdout);
    #endif

    threadtestdata->flag = 1; 

    etch_sleep(3000);

    if (threadtestdata->is_disposable)
        etch_free(threadtestdata);

    #if (IS_DEBUG_CONSOLE)
    printf("exit wait thread_a\n"); fflush(stdout);
    #endif
}


/**
 * test_wait()
 * test separate thread create and start   
 */
void test_wait(void)
{
    int  result = 0;
    etch_thread* thread    = 0;
    waittestdata* testdata = 0;  
    etch_threadparams* tp  = 0;

    testdata = etch_malloc(sizeof(waittestdata),0);
    testdata->flag = testdata->num = 0;
    testdata->is_disposable = FALSE;

    /* create a thread that waits to be started */
    thread = new_thread(wait_threadproc_a, testdata);
    CU_ASSERT_PTR_NOT_NULL_FATAL(thread);

    #if (IS_DEBUG_CONSOLE)
    printf("\nnew thread waiting for signal coming in ... "); fflush(stdout);
    #endif

    tp = &thread->params;
    /* testdata = tp->data; */

    #if (IS_DEBUG_CONSOLE)
    printf("3 ... "); fflush(stdout);
    #endif   
    etch_sleep(1000);

    #if (IS_DEBUG_CONSOLE)
    printf("2 ... "); fflush(stdout);
    #endif   
    etch_sleep(1000);

    #if (IS_DEBUG_CONSOLE)
    printf("1 ... "); fflush(stdout);
    #endif   
    etch_sleep(1000);

    #if (IS_DEBUG_CONSOLE)
    printf("\nmain signaling new thread to start ...\n"); fflush(stdout);
    #endif

    thread->start(thread);

    #if (IS_DEBUG_CONSOLE)
    printf("main issued newthread.start, waiting for exit ...\n"); fflush(stdout);
    #endif  

    etch_sleep(1000);
    
    etch_thread_join(tp);  /* block until thread exit */

    result = testdata->flag;
    CU_ASSERT_EQUAL(result, 1);
    etch_free(testdata);
    thread->destroy(thread);

    g_bytes_allocated = etch_showmem(0, IS_DEBUG_CONSOLE);  
    CU_ASSERT_EQUAL(g_bytes_allocated, 0);  
    memtable_clear();  /* start fresh for next test */   
}


typedef struct wait_until_testdata 
{ etchwait* waiter;   /* the condition variable waiter */
  int64 waitingfor;   /* value the waiter is waiting for */
  int   sleepstartms; 
  int   sleepexitms; 
} wait_until_testdata;


/**
 * wait_until_threadproc_a
 */
void wait_until_threadproc_a(void* data)
{
    int result = 0;
    etchwait* waiter;
    etch_threadparams* params;
    wait_until_testdata* threadtestdata; 
    params = (etch_threadparams*) data;
    threadtestdata = (wait_until_testdata*) params->data;
    waiter = threadtestdata->waiter;

    #if (IS_DEBUG_CONSOLE)
    printf("\nthread: started ...\n", threadtestdata->sleepstartms); fflush(stdout);
    #endif

    etch_sleep(threadtestdata->sleepstartms);  /* wait a bit before setting condition */

    #if (IS_DEBUG_CONSOLE)
    printf("\nthread: setting unblock condition ...\n"); fflush(stdout);
    #endif

    waiter->set(waiter, threadtestdata->waitingfor);  /* set wait condition to unblock waiters */

    etch_sleep(threadtestdata->sleepexitms);  /* wait a bit before exiting thread */

    etch_free(threadtestdata);  /* free caller's memory */

    #if (IS_DEBUG_CONSOLE)
    printf("thread: exiting ...\n"); fflush(stdout);
    #endif
}


/**
 * test_wait_until()
 * test wait on condition variable -- block until the specified condition is met.  
 */
void test_wait_until(void)
{
    int   result = 0;
    int64 condvar = 0;
    etchwait* waiter;
    etch_thread* thread = 0;
    etch_threadparams* tp = 0;
    const int64 VALUE_TO_WAIT_FOR = 1;
    wait_until_testdata* testdata = 0;  

    waiter = new_wait (etch_apr_mempool);

    testdata = etch_malloc(sizeof(wait_until_testdata),0);
    memset(testdata, 0, sizeof(wait_until_testdata));
    testdata->waiter = waiter;
    testdata->waitingfor = VALUE_TO_WAIT_FOR;  
    testdata->sleepstartms = testdata->sleepexitms = 2000;
   
    /* create a thread that waits to be started */
    thread = new_thread(wait_until_threadproc_a, testdata);
    CU_ASSERT_PTR_NOT_NULL_FATAL(thread);

    #if (IS_DEBUG_CONSOLE)
    printf("\nmain: countdown to start worker thread ... "); fflush(stdout);
    #endif

    tp = &thread->params;
    testdata = tp->data; 

    #if (IS_DEBUG_CONSOLE)
    printf("3 ... "); fflush(stdout);
    #endif   
    etch_sleep(1000);

    #if (IS_DEBUG_CONSOLE)
    printf("2 ... "); fflush(stdout);
    #endif   
    etch_sleep(1000);

    #if (IS_DEBUG_CONSOLE)
    printf("1 ... "); fflush(stdout);
    #endif   
    etch_sleep(1000);

    #if (IS_DEBUG_CONSOLE)
    printf("\nmain: signaling worker thread to start ...\n"); fflush(stdout);
    #endif

    thread->start(thread);

    #if (IS_DEBUG_CONSOLE)
    printf("main: worker thread started\n");  
    printf("main: start blocking on condition variable ...\n"); fflush(stdout);
    #endif  

    /* wait for worker thread to set condition variable */
    result = waiter->timed_waitequal(waiter, &condvar, VALUE_TO_WAIT_FOR, 5000);  

    #if (IS_DEBUG_CONSOLE)
    printf("main: wait unblocked, waiting for worker thread to exit ...\n"); fflush(stdout);
    #endif  
    CU_ASSERT_NOT_EQUAL_FATAL(result, -1);
    CU_ASSERT_NOT_EQUAL(result, ETCH_TIMEOUT);

    etch_thread_join(tp);  /* block until thread exit */
    
    waiter->destroy(waiter);
    thread->destroy(thread);

    g_bytes_allocated = etch_showmem(0, IS_DEBUG_CONSOLE);  
    CU_ASSERT_EQUAL(g_bytes_allocated, 0);  
    memtable_clear();  /* start fresh for next test */   
}



/**
 * test_wait_until_preexisting()
 * test wait on condition variable when unblock condition exists prior to request.  
 */
void test_wait_until_preexisting(void)
{
    int   result = 0;
    int64 condvar = 0;
    etchwait* waiter;
    etch_thread* thread = 0;
    etch_threadparams* tp = 0;
    const int64 VALUE_TO_WAIT_FOR = 1;
    wait_until_testdata* testdata = 0;  
    clock_t tickcount1 = 0, tickcount2 = 0, tickdiff = 0;

    waiter = new_wait (etch_apr_mempool);

    testdata = etch_malloc(sizeof(wait_until_testdata),0);
    memset(testdata, 0, sizeof(wait_until_testdata));
    testdata->waiter = waiter;
    testdata->waitingfor = VALUE_TO_WAIT_FOR;  
    testdata->sleepstartms = testdata->sleepexitms = 2000;
   
    /* create a thread that waits to be started */
    thread = new_thread(wait_until_threadproc_a, testdata);
    CU_ASSERT_PTR_NOT_NULL_FATAL(thread);

    #if (IS_DEBUG_CONSOLE)
    printf("\nmain: countdown to start worker thread ... "); fflush(stdout);
    #endif

    tp = &thread->params;
    testdata = tp->data; 

    #if (IS_DEBUG_CONSOLE)
    printf("3 ... "); fflush(stdout);
    #endif   
    etch_sleep(1000);

    #if (IS_DEBUG_CONSOLE)
    printf("2 ... "); fflush(stdout);
    #endif   
    etch_sleep(1000);

    #if (IS_DEBUG_CONSOLE)
    printf("1 ... "); fflush(stdout);
    #endif   
    etch_sleep(1000);

    #if (IS_DEBUG_CONSOLE)
    printf("\nmain: signaling worker thread to start ...\n"); fflush(stdout);
    #endif

    thread->start(thread);

    #if (IS_DEBUG_CONSOLE)
    printf("main: worker thread started\n");  
    #endif  

    condvar = VALUE_TO_WAIT_FOR; /* pre-set unblock condition */
    tickcount1 = clock();    

    /* wait for worker thread to set condition variable. we expect that the 
     * thread will not be started since the wait condition pre-exists */
    result = waiter->timed_waitequal(waiter, &condvar, VALUE_TO_WAIT_FOR, 5000);
    CU_ASSERT_EQUAL(result, 0);  

    /* we expect zero wait time since the wait condition preexisted */
    tickcount2 = clock();
    tickdiff = tickcount2 - tickcount1;
    result = tickdiff < 20; /* resolution of tick timer could be as high as 20ms */ 
    CU_ASSERT_EQUAL(result, TRUE); /* if debugger stepping this assert will fail */

    /* block until thread exit. however we expect that it has already done so. */
    etch_thread_join(tp);  
    
    waiter->destroy(waiter);
    thread->destroy(thread);

    g_bytes_allocated = etch_showmem(0, IS_DEBUG_CONSOLE);  
    CU_ASSERT_EQUAL(g_bytes_allocated, 0);  
    memtable_clear();  /* start fresh for next test */   
}


/**
 * test_wait_until_negative_test()
 * test wait on condition variable  
 */
void test_wait_until_negative_test(void)
{
    int   result = 0;
    int64 condvar = 0;
    etchwait* waiter;
    etch_thread* thread = 0;
    etch_threadparams* tp = 0;
    const int VALUE_TO_WAIT_FOR = 1, VALUE_TO_SET = 2, THREAD_PAUSE_MS = 1000;
    wait_until_testdata* testdata = 0;  

    waiter = new_wait (etch_apr_mempool);

    testdata = etch_malloc(sizeof(wait_until_testdata),0);
    memset(testdata, 0, sizeof(wait_until_testdata));
    testdata->waiter = waiter;
    testdata->waitingfor = VALUE_TO_SET;  /* not the value we're going to wait for */ 
    testdata->sleepstartms = testdata->sleepexitms = THREAD_PAUSE_MS;
   
    /* create a thread that waits to be started */
    thread = new_thread(wait_until_threadproc_a, testdata);
    CU_ASSERT_PTR_NOT_NULL_FATAL(thread);

    #if (IS_DEBUG_CONSOLE)
    printf("\nmain: countdown to start worker thread ... "); fflush(stdout);
    #endif

    tp = &thread->params;
    testdata = tp->data; 

    #if (IS_DEBUG_CONSOLE)
    printf("3 ... "); fflush(stdout);
    #endif   
    etch_sleep(1000);

    #if (IS_DEBUG_CONSOLE)
    printf("2 ... "); fflush(stdout);
    #endif   
    etch_sleep(1000);

    #if (IS_DEBUG_CONSOLE)
    printf("1 ... "); fflush(stdout);
    #endif   
    etch_sleep(1000);

    #if (IS_DEBUG_CONSOLE)
    printf("\nmain: signaling worker thread to start ...\n"); fflush(stdout);
    #endif

    thread->start(thread);

    #if (IS_DEBUG_CONSOLE)
    printf("main: worker thread started\n");  
    printf("main: start blocking on condition variable ...\n"); fflush(stdout);
    #endif  

    /* wait for worker thread to set condition variable. in this test, we have told
     * the thread to set some other value than what we are waiting for, so we want
     * to see a wait timeout here. in other words, we are testing that setting the
     * waiter to some value other than what we are waiting for, does not unblock. 
     */
    result = waiter->timed_waitequal(waiter, &condvar, VALUE_TO_WAIT_FOR, THREAD_PAUSE_MS + 1000 );  

    CU_ASSERT_EQUAL(result, ETCH_TIMEOUT);

    #if (IS_DEBUG_CONSOLE)
    printf("main: wait unblocked, waiting for worker thread to exit ...\n"); fflush(stdout);
    #endif  

    etch_thread_join(tp);  /* block until thread exit */
    
    waiter->destroy(waiter);
    thread->destroy(thread);

    g_bytes_allocated = etch_showmem(0, IS_DEBUG_CONSOLE);  
    CU_ASSERT_EQUAL(g_bytes_allocated, 0);  
    memtable_clear();  /* start fresh for next test */   
}


/**
 * wait_multiple_threadproc
 */
void wait_multiple_threadproc(void* data)
{
    int threadnum, waitms, remainingms;
    etch_threadparams* params = (etch_threadparams*) data;
    waittestdata* tdata = (waittestdata*) params->data; 
    threadnum = tdata->num;
    waitms = tdata->flag;
    remainingms = waitms;

    #if (IS_DEBUG_CONSOLE)
    printf("\nwait thread %d sleeping %d ms ...\n", threadnum, waitms); fflush(stdout);
    #endif 

    while(remainingms > 0)
    {
        if (params->threadstate > ETCH_THREADSTATE_STARTED) break; 
        etch_sleep(remainingms > 500? 500: remainingms);
        remainingms -= 500;
    }

    #if (IS_DEBUG_CONSOLE)
    printf("exit wait thread %d\n", threadnum); fflush(stdout);
    #endif
}


/**
 * test_wait_for_multiple()
 * launch multiple threads and wait for them all to complete   
 */
void test_wait_for_multiple(void)
{
    #define TEST_WAIT_MULT_NUMTHREADS 4

    etch_thread* thread[TEST_WAIT_MULT_NUMTHREADS];
    int waitms[TEST_WAIT_MULT_NUMTHREADS], i, priori = 1;

    for(i=0; i < TEST_WAIT_MULT_NUMTHREADS; i++)
    {   /* initialize wait times for each thread */
        const int newi = i & 1? priori - 1: priori + 2;
        waitms[i] = ((priori = newi) * 1000);        
    }

    #if (IS_DEBUG_CONSOLE)
    printf("main creating all threads ...\n"); fflush(stdout);
    #endif 
  
    for(i=0; i < TEST_WAIT_MULT_NUMTHREADS; i++) /* create all threads */
    {   
        etch_thread* newthread;
        waittestdata* thread_data = etch_malloc(sizeof(waittestdata),0);
        thread_data->num = i; thread_data->flag = waitms[i];

        newthread = new_thread(wait_multiple_threadproc, thread_data);
        CU_ASSERT_PTR_NOT_NULL_FATAL(newthread);

        /* configure thread to free memory for the waittestdata we passed it */
        newthread->params.is_own_data = TRUE;  
        thread[i] = newthread;
    }

    #if (IS_DEBUG_CONSOLE)
    printf("main starting all threads ...\n"); fflush(stdout);
    #endif 

    for(i=0; i < TEST_WAIT_MULT_NUMTHREADS; i++) /* start all threads */
        thread[i]->start(thread[i]);
         
    #if (IS_DEBUG_CONSOLE)
    printf("main waiting for all threads to exit ...\n"); fflush(stdout);
    #endif 

    for(i=0; i < TEST_WAIT_MULT_NUMTHREADS; i++) /* wait for all threads to exit */
        etch_join(thread[i]); 

    #if (IS_DEBUG_CONSOLE)
    printf("main all threads accounted for\n"); fflush(stdout);
    #endif 

    for(i=0; i < TEST_WAIT_MULT_NUMTHREADS; i++) /* free thread objects */
        thread[i]->destroy(thread[i]);
       
    g_bytes_allocated = etch_showmem(0, IS_DEBUG_CONSOLE);  
    CU_ASSERT_EQUAL(g_bytes_allocated, 0);  
    memtable_clear();  /* start fresh for next test */   
}


/**
 * test_free_threadpool()
 * test thread "pool" (ad hoc threads)   
 */
void test_free_threadpool(void)
{
    #define TEST_FREEPOOL_NUMTHREADS 4
    etch_threadpool* threadpool = NULL;
    void* runthread = NULL;
    int waitms[TEST_FREEPOOL_NUMTHREADS], i, priori = 1;

    for(i=0; i < TEST_FREEPOOL_NUMTHREADS; i++)
    {   /* initialize wait times for each thread */
        const int newi = i & 1? priori - 1: priori + 2;
        waitms[i] = ((priori = newi) * 1000);        
    }

    threadpool = new_threadpool(ETCH_THREADPOOLTYPE_FREE, 0);
    CU_ASSERT_PTR_NOT_NULL_FATAL(threadpool);
    threadpool->is_free_threads = TRUE;

    #if (IS_DEBUG_CONSOLE)
    printf("main creating %d pool threads ...\n", TEST_FREEPOOL_NUMTHREADS); fflush(stdout);
    #endif 
  
    for(i=0; i < TEST_FREEPOOL_NUMTHREADS; i++) /* launch all threads */
    {   
        waittestdata* thread_data = etch_malloc(sizeof(waittestdata),0);
        thread_data->num = i; thread_data->flag = waitms[i];

        runthread = threadpool->run (threadpool, wait_multiple_threadproc, thread_data);

        CU_ASSERT_PTR_NOT_NULL(runthread);
    }
         
    #if (IS_DEBUG_CONSOLE)
    printf("main all pool threads launched\n"); fflush(stdout);
    printf("main destroying thread pool ...\n"); fflush(stdout);
    #endif 

    threadpool->destroy(threadpool);

    #if (IS_DEBUG_CONSOLE)
    printf("main all threads accounted for\n"); fflush(stdout);
    #endif 
       
    g_bytes_allocated = etch_showmem(0, IS_DEBUG_CONSOLE);  
    CU_ASSERT_EQUAL(g_bytes_allocated, 0);  
    memtable_clear();  /* start fresh for next test */   
}


/**
 * main   
 */
int _tmain(int argc, _TCHAR* argv[])
{    
    CU_pSuite ps = NULL; char c=0;
    g_is_automated_test = argc > 1 && 0 != wcscmp(argv[1], L"-a");
    if (CUE_SUCCESS != CU_initialize_registry()) return 0;
    CU_set_output_filename("../test_threadpool");
    ps = CU_add_suite("suite_threadpool", init_suite, clean_suite);
    etch_watch_id = 0;  
        
    CU_add_test(ps, "test create thread", test_createthread);
    CU_add_test(ps, "test etch mutex", test_mutex);
    CU_add_test(ps, "test arraylist synchronization", test_synched_arraylist);
    CU_add_test(ps, "test thread wait/start", test_wait);

    CU_add_test(ps, "test wait on condition", test_wait_until); 
    CU_add_test(ps, "test wait on pre-existing condition", test_wait_until_preexisting);
    CU_add_test(ps, "test wait on condition - negative", test_wait_until_negative_test); 

    CU_add_test(ps, "test wait all threads exit", test_wait_for_multiple);
    CU_add_test(ps, "test free threadpool", test_free_threadpool);

    if (g_is_automated_test)    
        CU_automated_run_tests();    
    else
    {   CU_basic_set_mode(CU_BRM_VERBOSE);
        CU_basic_run_tests();
    }

    if (!g_is_automated_test) { printf("any key ..."); while(!c) c = _getch(); printf("\n"); }     
    CU_cleanup_registry();
    return CU_get_error();
}