forked from lei.chen/myxce
299 lines
12 KiB
C
299 lines
12 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <time.h>
|
|
|
|
#include <unity/unity.h>
|
|
|
|
#include "jobj.h"
|
|
#include "kvmgr.h"
|
|
#include "utils.h"
|
|
#include "xstring.h"
|
|
|
|
static void test_startinfo_create()
|
|
{
|
|
int rc;
|
|
job_startinfo_t s1 = { .exechome = "/homes/rshen",
|
|
.execcwd = "/tmp",
|
|
.execuser = "rshen",
|
|
.cmdline = "sleep 3000",
|
|
.exdata = "Hello",
|
|
.q_pre_exec = "sleep 1",
|
|
.q_post_exec = "sleep 10",
|
|
.job_pre_exec = "pwd",
|
|
.job_post_exec = "hello" };
|
|
job_startinfo_t* s2 = NULL;
|
|
|
|
rc = xstring_table_init(32);
|
|
TEST_ASSERT_TRUE(rc >= 0);
|
|
|
|
rc = job_obj_init(31, 31);
|
|
TEST_ASSERT_TRUE(rc >= 0);
|
|
|
|
s2 = makeJobstartinfo(&s1);
|
|
|
|
TEST_ASSERT_TRUE(s2 != NULL);
|
|
|
|
TEST_ASSERT_TRUE(EQUAL_STRING(s1.exechome, s2->exechome)
|
|
&& EQUAL_STRING(s1.execcwd, s2->execcwd)
|
|
&& EQUAL_STRING(s1.execuser, s2->execuser)
|
|
&& EQUAL_STRING(s1.cmdline, s2->cmdline)
|
|
&& EQUAL_STRING(s1.exdata, s2->exdata)
|
|
&& EQUAL_STRING(s1.q_pre_exec, s2->q_pre_exec)
|
|
&& EQUAL_STRING(s1.q_post_exec, s2->q_post_exec)
|
|
&& EQUAL_STRING(s1.job_pre_exec, s2->job_pre_exec)
|
|
&& EQUAL_STRING(s1.job_post_exec, s2->job_post_exec));
|
|
job_obj_fini();
|
|
xstring_table_fini();
|
|
}
|
|
|
|
static void test_startinfo_share()
|
|
{
|
|
int rc;
|
|
job_startinfo_t s1 = { .exechome = "/homes/rshen",
|
|
.execcwd = "/tmp",
|
|
.execuser = "rshen",
|
|
.cmdline = "sleep 3000",
|
|
.exdata = "Hello",
|
|
.q_pre_exec = "sleep 1",
|
|
.q_post_exec = "sleep 10",
|
|
.job_pre_exec = "pwd",
|
|
.job_post_exec = "hello" };
|
|
job_startinfo_t* s2 = NULL;
|
|
job_startinfo_t* s3 = NULL;
|
|
|
|
rc = xstring_table_init(32);
|
|
TEST_ASSERT_TRUE(rc >= 0);
|
|
|
|
rc = job_obj_init(31, 31);
|
|
TEST_ASSERT_TRUE(rc >= 0);
|
|
|
|
s2 = makeJobstartinfo(&s1);
|
|
|
|
TEST_ASSERT_TRUE(s2 != NULL);
|
|
|
|
TEST_ASSERT_TRUE(EQUAL_STRING(s1.exechome, s2->exechome)
|
|
&& EQUAL_STRING(s1.execcwd, s2->execcwd)
|
|
&& EQUAL_STRING(s1.execuser, s2->execuser)
|
|
&& EQUAL_STRING(s1.cmdline, s2->cmdline)
|
|
&& EQUAL_STRING(s1.exdata, s2->exdata)
|
|
&& EQUAL_STRING(s1.q_pre_exec, s2->q_pre_exec)
|
|
&& EQUAL_STRING(s1.q_post_exec, s2->q_post_exec)
|
|
&& EQUAL_STRING(s1.job_pre_exec, s2->job_pre_exec)
|
|
&& EQUAL_STRING(s1.job_post_exec, s2->job_post_exec));
|
|
|
|
s3 = makeJobstartinfo(&s1);
|
|
TEST_ASSERT_TRUE(s2 == s3);
|
|
|
|
job_obj_fini();
|
|
xstring_table_fini();
|
|
}
|
|
|
|
static void test_jobsub_create(void)
|
|
{
|
|
int rc;
|
|
job_startinfo_t s1 = { .exechome = "/homes/rshen",
|
|
.execcwd = "/tmp",
|
|
.execuser = "rshen",
|
|
.cmdline = "sleep 3000",
|
|
.exdata = "Hello",
|
|
.q_pre_exec = "sleep 1",
|
|
.q_post_exec = "sleep 10",
|
|
.job_pre_exec = "pwd",
|
|
.job_post_exec = "hello" };
|
|
job_submit_data_t jobsub1 = { .queue = "xce",
|
|
.user = "rshen",
|
|
.mintasks = 1,
|
|
.maxtasks = 5,
|
|
.start_info = &s1,
|
|
.job_name = "J1",
|
|
.project_name = "default",
|
|
.user_group = NULL,
|
|
.job_description = "no description",
|
|
.resreq = "type == any" };
|
|
job_submit_data_t* s2 = NULL;
|
|
|
|
rc = xstring_table_init(32);
|
|
TEST_ASSERT_TRUE(rc >= 0);
|
|
|
|
rc = job_obj_init(31, 31);
|
|
TEST_ASSERT_TRUE(rc >= 0);
|
|
|
|
s2 = makeJobsub(&jobsub1);
|
|
|
|
TEST_ASSERT_TRUE(s2 != NULL);
|
|
|
|
TEST_ASSERT_TRUE(
|
|
EQUAL_STRING(s1.exechome, s2->start_info->exechome)
|
|
&& EQUAL_STRING(s1.execcwd, s2->start_info->execcwd)
|
|
&& EQUAL_STRING(s1.execuser, s2->start_info->execuser)
|
|
&& EQUAL_STRING(s1.cmdline, s2->start_info->cmdline)
|
|
&& EQUAL_STRING(s1.exdata, s2->start_info->exdata)
|
|
&& EQUAL_STRING(s1.q_pre_exec, s2->start_info->q_pre_exec)
|
|
&& EQUAL_STRING(s1.q_post_exec, s2->start_info->q_post_exec)
|
|
&& EQUAL_STRING(s1.job_pre_exec, s2->start_info->job_pre_exec)
|
|
&& EQUAL_STRING(s1.job_post_exec, s2->start_info->job_post_exec)
|
|
&& EQUAL_STRING(s2->queue, "xce") && EQUAL_STRING(s2->user, "rshen")
|
|
&& EQUAL_STRING(s2->job_name, "J1")
|
|
&& EQUAL_STRING(s2->project_name, "default")
|
|
&& EQUAL_STRING(s2->job_description, "no description")
|
|
&& EQUAL_STRING(s2->resreq, "type == any") && (s2->mintasks == 1)
|
|
&& (s2->maxtasks == 5) && (s2->user_group == NULL));
|
|
|
|
job_obj_fini();
|
|
xstring_table_fini();
|
|
}
|
|
|
|
static void test_jobsub_share(void)
|
|
{
|
|
int rc;
|
|
job_startinfo_t s1 = { .exechome = "/homes/rshen",
|
|
.execcwd = "/tmp",
|
|
.execuser = "rshen",
|
|
.cmdline = "sleep 3000",
|
|
.exdata = "Hello",
|
|
.q_pre_exec = "sleep 1",
|
|
.q_post_exec = "sleep 10",
|
|
.job_pre_exec = "pwd",
|
|
.job_post_exec = "hello" };
|
|
job_submit_data_t jobsub1 = { .queue = "xce",
|
|
.user = "rshen",
|
|
.mintasks = 1,
|
|
.maxtasks = 5,
|
|
.start_info = &s1,
|
|
.job_name = "J1",
|
|
.project_name = "default",
|
|
.user_group = NULL,
|
|
.job_description = "no description",
|
|
.resreq = "type == any" };
|
|
job_submit_data_t* s2 = NULL;
|
|
job_submit_data_t* s3 = NULL;
|
|
|
|
rc = xstring_table_init(32);
|
|
TEST_ASSERT_TRUE(rc >= 0);
|
|
|
|
rc = job_obj_init(31, 31);
|
|
TEST_ASSERT_TRUE(rc >= 0);
|
|
|
|
s2 = makeJobsub(&jobsub1);
|
|
|
|
TEST_ASSERT_TRUE(s2 != NULL);
|
|
|
|
TEST_ASSERT_TRUE(
|
|
EQUAL_STRING(s1.exechome, s2->start_info->exechome)
|
|
&& EQUAL_STRING(s1.execcwd, s2->start_info->execcwd)
|
|
&& EQUAL_STRING(s1.execuser, s2->start_info->execuser)
|
|
&& EQUAL_STRING(s1.cmdline, s2->start_info->cmdline)
|
|
&& EQUAL_STRING(s1.exdata, s2->start_info->exdata)
|
|
&& EQUAL_STRING(s1.q_pre_exec, s2->start_info->q_pre_exec)
|
|
&& EQUAL_STRING(s1.q_post_exec, s2->start_info->q_post_exec)
|
|
&& EQUAL_STRING(s1.job_pre_exec, s2->start_info->job_pre_exec)
|
|
&& EQUAL_STRING(s1.job_post_exec, s2->start_info->job_post_exec)
|
|
&& EQUAL_STRING(s2->queue, "xce") && EQUAL_STRING(s2->user, "rshen")
|
|
&& EQUAL_STRING(s2->job_name, "J1")
|
|
&& EQUAL_STRING(s2->project_name, "default")
|
|
&& EQUAL_STRING(s2->job_description, "no description")
|
|
&& EQUAL_STRING(s2->resreq, "type == any") && (s2->mintasks == 1)
|
|
&& (s2->maxtasks == 5) && (s2->user_group == NULL));
|
|
|
|
s3 = makeJobsub(&jobsub1);
|
|
|
|
TEST_ASSERT_TRUE(s2 == s3);
|
|
|
|
job_obj_fini();
|
|
xstring_table_fini();
|
|
}
|
|
|
|
static void test_job_create(void)
|
|
{
|
|
int rc;
|
|
kvmgr_params_t jobmgr_params = { .key_type = KVMGR_KEY_INT,
|
|
.nslots = 32,
|
|
.delta = 32,
|
|
.datakiller = destroy };
|
|
var jobmgr;
|
|
job_startinfo_t s1 = { .exechome = "/homes/rshen",
|
|
.execcwd = "/tmp",
|
|
.execuser = "rshen",
|
|
.cmdline = "sleep 3000",
|
|
.exdata = "Hello",
|
|
.q_pre_exec = "sleep 1",
|
|
.q_post_exec = "sleep 10",
|
|
.job_pre_exec = "pwd",
|
|
.job_post_exec = "hello" };
|
|
job_submit_data_t jobsub1 = { .queue = "xce",
|
|
.user = "rshen",
|
|
.mintasks = 1,
|
|
.maxtasks = 5,
|
|
.start_info = &s1,
|
|
.job_name = "J1",
|
|
.project_name = "default",
|
|
.user_group = NULL,
|
|
.job_description = "no description",
|
|
.resreq = "type == any" };
|
|
xce_job_t job1 = { .submit = &jobsub1 };
|
|
job_t* j2 = NULL;
|
|
time_t now = time(NULL);
|
|
|
|
rc = xstring_table_init(32);
|
|
TEST_ASSERT_TRUE(rc >= 0);
|
|
|
|
rc = job_obj_init(31, 31);
|
|
TEST_ASSERT_TRUE(rc >= 0);
|
|
|
|
job1.sub_time = now;
|
|
job1.queue_time = now;
|
|
job1.start_time = 0;
|
|
job1.finish_time = 0;
|
|
job1.pidinfo = NULL;
|
|
job1.state = JOB_STAT_PEND;
|
|
job1.allocs = NULL;
|
|
job1.run_time = 0;
|
|
job1.run_usage = NULL;
|
|
job1.exit_info = NULL;
|
|
job1.ckptinfo = NULL;
|
|
job1.pending_ctrl = 0;
|
|
job1.jobid = 100;
|
|
|
|
jobmgr = makeInstance(&xkvmgr_type, &jobmgr_params);
|
|
TEST_ASSERT_TRUE(jobmgr != NULL);
|
|
|
|
j2 = makeJob(jobmgr, &job1);
|
|
|
|
TEST_ASSERT_TRUE(j2 != NULL);
|
|
|
|
TEST_ASSERT_TRUE(
|
|
EQUAL_STRING(s1.exechome, j2->job.submit->start_info->exechome)
|
|
&& EQUAL_STRING(s1.execcwd, j2->job.submit->start_info->execcwd)
|
|
&& EQUAL_STRING(s1.execuser, j2->job.submit->start_info->execuser)
|
|
&& EQUAL_STRING(s1.cmdline, j2->job.submit->start_info->cmdline)
|
|
&& EQUAL_STRING(s1.exdata, j2->job.submit->start_info->exdata)
|
|
&& EQUAL_STRING(s1.q_pre_exec, j2->job.submit->start_info->q_pre_exec)
|
|
&& EQUAL_STRING(s1.q_post_exec, j2->job.submit->start_info->q_post_exec)
|
|
&& EQUAL_STRING(s1.job_pre_exec, j2->job.submit->start_info->job_pre_exec)
|
|
&& EQUAL_STRING(s1.job_post_exec, j2->job.submit->start_info->job_post_exec)
|
|
&& EQUAL_STRING(j2->job.submit->queue, "xce")
|
|
&& EQUAL_STRING(j2->job.submit->user, "rshen")
|
|
&& EQUAL_STRING(j2->job.submit->job_name, "J1")
|
|
&& EQUAL_STRING(j2->job.submit->project_name, "default")
|
|
&& EQUAL_STRING(j2->job.submit->job_description, "no description")
|
|
&& EQUAL_STRING(j2->job.submit->resreq, "type == any")
|
|
&& (j2->job.submit->mintasks == 1) && (j2->job.submit->maxtasks == 5)
|
|
&& (j2->job.submit->user_group == NULL));
|
|
|
|
destroy(jobmgr);
|
|
|
|
job_obj_fini();
|
|
xstring_table_fini();
|
|
}
|
|
|
|
void test_jobobj(void)
|
|
{
|
|
printf("\n>> job obj test\n\n");
|
|
|
|
RUN_TEST(test_startinfo_create);
|
|
RUN_TEST(test_startinfo_share);
|
|
RUN_TEST(test_jobsub_create);
|
|
RUN_TEST(test_jobsub_share);
|
|
RUN_TEST(test_job_create);
|
|
}
|