merge job rusage and pending reason support

This commit is contained in:
Rongsong Shen 2025-12-16 14:36:28 +08:00
parent 30824ae5ea
commit cb0c7fe9e7
82 changed files with 4280 additions and 2561 deletions

View file

@ -12,6 +12,9 @@ extern void test_error(void);
extern void test_utils(void);
extern void test_config(void);
extern void test_idgen(void);
extern void test_preason(void);
extern void test_ordq(void);
extern void test_delay(void);
void setUp(void) { return; }
@ -28,6 +31,9 @@ int main(int argc, char* argv[])
test_xstrings();
test_error();
test_utils();
test_preason();
test_ordq();
test_delay();
return UNITY_END();
}

View file

@ -0,0 +1,56 @@
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "delay.h"
#include "unity.h"
static int delay_work(void* arg)
{
time_t now;
int* pcount = (int*)arg;
now = time(NULL);
(*pcount)++;
printf("delay task run at %ld\n", now);
return 0;
}
static void test_delay1()
{
delaymgr_t mgr;
int count = 0;
int runcount = 15;
int delay_period = 2;
time_t t_start = time(NULL);
delaymgr_init(&mgr, 1, NULL);
delaymgr_start(&mgr);
delay_task(&mgr, delay_period, runcount, (delay_task_entry_fn_t)delay_work,
&count);
while (1) {
time_t now = time(NULL);
if (now - t_start > (runcount + 2) * delay_period) {
break;
}
sleep(5);
}
TEST_ASSERT_TRUE(count == runcount);
delaymgr_fini(&mgr);
return;
}
void test_delay(void)
{
printf("\n>> delay test\n\n");
RUN_TEST(test_delay1);
}

View file

@ -0,0 +1,82 @@
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "ordq.h"
#include "unity.h"
#include "utils.h"
static int long_cmp(ordkey_t k1, ordkey_t k2)
{
if ((long)k1 < (long)k2) {
return -1;
} else if ((long)k1 > (long)k2) {
return 1;
}
return 0;
}
static void test_ordq_get(void)
{
int rc;
ordq_t q;
ordq_item_t* it;
int keys[8] = { 9, 8, 1, 4, 7, 2, 6, 2 };
int val[8] = { 1, 2, 2, 4, 6, 7, 8, 9 };
int i;
rc = ordq_init(&q, NULL, (ordq_cmp_fn_t)long_cmp);
TEST_ASSERT_TRUE(rc >= 0);
for (i = 0; i < 8; i++) {
ordq_put(&q, mk_ordkey(keys[i]), (void*)NULL);
}
for (i = 0; i < 8; i++) {
it = ordq_get(&q);
TEST_ASSERT_TRUE((it != NULL) && (it->key == mk_ordkey(val[i])));
free(it);
}
ordq_fini(&q);
return;
}
static void test_ordq_peek(void)
{
int rc;
ordq_t q;
ordq_item_t* it;
int keys[8] = { 9, 8, 1, 4, 7, 2, 6, 2 };
int val[8] = { 1, 2, 2, 4, 6, 7, 8, 9 };
int i;
rc = ordq_init(&q, NULL, (ordq_cmp_fn_t)long_cmp);
TEST_ASSERT_TRUE(rc >= 0);
for (i = 0; i < 8; i++) {
ordq_put(&q, mk_ordkey(keys[i]), (void*)NULL);
}
for (i = 0; i < 8; i++) {
it = ordq_peek(&q); /* ordq_peek() will not remove item from q */
TEST_ASSERT_TRUE((it != NULL) && (it->key == mk_ordkey(val[0])));
}
ordq_fini(&q);
return;
}
void test_ordq(void)
{
printf("\n>> ordq test\n\n");
RUN_TEST(test_ordq_get);
RUN_TEST(test_ordq_peek);
}

View file

@ -0,0 +1,168 @@
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "preason.h"
#include "unity.h"
static void test_main_reason()
{
xce_job_preason_t reason;
memset(&reason, 0x0, sizeof(reason));
add_job_preason(&reason, 1, NULL, NULL);
TEST_ASSERT_TRUE(reason.main_reason == 1);
TEST_ASSERT_TRUE((reason.nhost_reasons == 0)
&& (reason.host_based_reasons == NULL));
TEST_ASSERT_TRUE((reason.nres_reasons == 0)
&& (reason.res_based_reasons == NULL));
}
static void test_add_host_reason(void)
{
xce_job_preason_t reason;
memset(&reason, 0x0, sizeof(reason));
add_job_preason(&reason, 1, "h1", NULL);
TEST_ASSERT_TRUE(reason.main_reason == 0);
TEST_ASSERT_TRUE((reason.nhost_reasons == 1)
&& (reason.host_based_reasons != NULL)
&& (reason.host_based_reasons[0] != NULL));
TEST_ASSERT_TRUE(
(reason.host_based_reasons[0]->reason == 1)
&& (strcmp(reason.host_based_reasons[0]->host, "h1") == 0));
TEST_ASSERT_TRUE((reason.nres_reasons == 0)
&& (reason.res_based_reasons == NULL));
erase_job_preason(&reason);
}
static void test_add_res_reason(void)
{
xce_job_preason_t reason;
memset(&reason, 0x0, sizeof(reason));
add_job_preason(&reason, 1, "h1", "res1");
TEST_ASSERT_TRUE(reason.main_reason == 0);
TEST_ASSERT_TRUE((reason.nres_reasons == 1)
&& (reason.res_based_reasons != NULL)
&& (reason.res_based_reasons[0] != NULL));
TEST_ASSERT_TRUE(
(reason.res_based_reasons[0]->reason == 1)
&& (strcmp(reason.res_based_reasons[0]->host, "h1") == 0)
&& (strcmp(reason.res_based_reasons[0]->resource, "res1") == 0));
TEST_ASSERT_TRUE((reason.nhost_reasons == 0)
&& (reason.host_based_reasons == NULL));
erase_job_preason(&reason);
}
static void test_add_res_host_reason(void)
{
xce_job_preason_t reason;
memset(&reason, 0x0, sizeof(reason));
add_job_preason(&reason, 1, "h1", "res1");
add_job_preason(&reason, 2, "h1", NULL);
TEST_ASSERT_TRUE(reason.main_reason == 0);
TEST_ASSERT_TRUE((reason.nres_reasons == 1)
&& (reason.res_based_reasons != NULL)
&& (reason.res_based_reasons[0] != NULL));
TEST_ASSERT_TRUE(
(reason.res_based_reasons[0]->reason == 1)
&& (strcmp(reason.res_based_reasons[0]->host, "h1") == 0)
&& (strcmp(reason.res_based_reasons[0]->resource, "res1") == 0));
TEST_ASSERT_TRUE((reason.nhost_reasons == 1)
&& (reason.host_based_reasons != NULL)
&& (reason.host_based_reasons[0] != NULL));
TEST_ASSERT_TRUE(
(reason.host_based_reasons[0]->reason == 2)
&& (strcmp(reason.host_based_reasons[0]->host, "h1") == 0));
erase_job_preason(&reason);
}
static void test_add_main_reset_other(void)
{
xce_job_preason_t reason;
memset(&reason, 0x0, sizeof(reason));
add_job_preason(&reason, 1, "h1", "res1");
add_job_preason(&reason, 2, "h1", NULL);
add_job_preason(&reason, 3, NULL, NULL);
TEST_ASSERT_TRUE(reason.main_reason == 3);
TEST_ASSERT_TRUE((reason.nres_reasons == 0)
&& (reason.res_based_reasons == NULL));
TEST_ASSERT_TRUE((reason.nhost_reasons == 0)
&& (reason.host_based_reasons == NULL));
erase_job_preason(&reason);
}
static void test_add_more_host_reason(void)
{
xce_job_preason_t reason;
memset(&reason, 0x0, sizeof(reason));
add_job_preason(&reason, 1, "h1", NULL);
add_job_preason(&reason, 2, "h2", NULL);
TEST_ASSERT_TRUE(reason.main_reason == 0);
TEST_ASSERT_TRUE((reason.nhost_reasons == 2)
&& (reason.host_based_reasons != NULL)
&& (reason.host_based_reasons[0] != NULL)
&& (reason.host_based_reasons[1] != NULL));
TEST_ASSERT_TRUE(
(reason.host_based_reasons[0]->reason == 1)
&& (strcmp(reason.host_based_reasons[0]->host, "h1") == 0)
&& (reason.host_based_reasons[1]->reason == 2)
&& (strcmp(reason.host_based_reasons[1]->host, "h2") == 0));
TEST_ASSERT_TRUE((reason.nres_reasons == 0)
&& (reason.res_based_reasons == NULL));
erase_job_preason(&reason);
}
static void test_add_more_res_reason(void)
{
xce_job_preason_t reason;
memset(&reason, 0x0, sizeof(reason));
add_job_preason(&reason, 1, "h1", "res1");
add_job_preason(&reason, 2, "h2", "res2");
TEST_ASSERT_TRUE(reason.main_reason == 0);
TEST_ASSERT_TRUE((reason.nres_reasons == 2)
&& (reason.res_based_reasons != NULL)
&& (reason.res_based_reasons[0] != NULL)
&& (reason.res_based_reasons[1] != NULL));
TEST_ASSERT_TRUE(
(reason.res_based_reasons[0]->reason == 1)
&& (strcmp(reason.res_based_reasons[0]->host, "h1") == 0)
&& (strcmp(reason.res_based_reasons[0]->resource, "res1") == 0));
TEST_ASSERT_TRUE(
(reason.res_based_reasons[1]->reason == 2)
&& (strcmp(reason.res_based_reasons[1]->host, "h2") == 0)
&& (strcmp(reason.res_based_reasons[1]->resource, "res2") == 0));
TEST_ASSERT_TRUE((reason.nhost_reasons == 0)
&& (reason.host_based_reasons == NULL));
erase_job_preason(&reason);
}
void test_preason(void)
{
printf("\n>> preason test\n\n");
RUN_TEST(test_main_reason);
RUN_TEST(test_add_host_reason);
RUN_TEST(test_add_res_reason);
RUN_TEST(test_add_res_host_reason);
RUN_TEST(test_add_main_reset_other);
RUN_TEST(test_add_more_host_reason);
RUN_TEST(test_add_more_res_reason);
}