#include <rtdevice.h>
#include "utest.h"
#include "utest_assert.h"
#include "smp.h"
int pass_count = 0;
int pass = 1000;
struct rt_spinlock lock;

void test_call(void *data)
{
    rt_spin_lock(&lock);
    int *i = (int *)data;
    int id = rt_hw_cpu_id();
    *i &= ~(1 << id);
    if(*i == 0)
        pass_count++;
    rt_spin_unlock(&lock);
}


void test()
{
    int cpu_mask = 0xf;
    for(int i =0 ;i < 1000 ;i++)
    {
        cpu_mask = rand()% 0xf;
        if (cpu_mask == 0) 
        pass--;
        rt_smp_call_func_cond(cpu_mask,test_call, &cpu_mask);
        if(i % 20 == 0)
            rt_kprintf("#");
        rt_thread_mdelay(1);
    }    
    rt_kprintf("\n");
}



static rt_err_t utest_tc_init(void)
{
    rt_spin_lock_init(&lock);
    return RT_EOK;
}

static rt_err_t utest_tc_cleanup(void)
{
    uassert_true(pass_count == pass);
    return RT_EOK;
}
static void testcase(void)
{
    UTEST_UNIT_RUN(test);
}

UTEST_TC_EXPORT(testcase, "testcase.smp.smp", utest_tc_init, utest_tc_cleanup, 10);