-
Notifications
You must be signed in to change notification settings - Fork 10
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
t: add tests for tracking resources across jobs
Problem: There exists no tests for tracking the number of resources used across an association's running jobs and ensuring that the counts are as expected. Add some tests.
- Loading branch information
Showing
2 changed files
with
95 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,94 @@ | ||
#!/bin/bash | ||
|
||
test_description='track resources across running jobs per-association in priority plugin' | ||
|
||
. `dirname $0`/sharness.sh | ||
|
||
mkdir -p conf.d | ||
|
||
MULTI_FACTOR_PRIORITY=${FLUX_BUILD_DIR}/src/plugins/.libs/mf_priority.so | ||
SUBMIT_AS=${SHARNESS_TEST_SRCDIR}/scripts/submit_as.py | ||
DB_PATH=$(pwd)/FluxAccountingTest.db | ||
|
||
export TEST_UNDER_FLUX_SCHED_SIMPLE_MODE="limited=1" | ||
test_under_flux 4 job -o,--config-path=$(pwd)/conf.d | ||
|
||
flux setattr log-stderr-level 1 | ||
|
||
test_expect_success 'allow guest access to testexec' ' | ||
flux config load <<-EOF | ||
[exec.testexec] | ||
allow-guests = true | ||
EOF | ||
' | ||
|
||
test_expect_success 'load multi-factor priority plugin' ' | ||
flux jobtap load -r .priority-default ${MULTI_FACTOR_PRIORITY} | ||
' | ||
|
||
test_expect_success 'check that mf_priority plugin is loaded' ' | ||
flux jobtap list | grep mf_priority | ||
' | ||
|
||
test_expect_success 'create flux-accounting DB' ' | ||
flux account -p ${DB_PATH} create-db | ||
' | ||
|
||
test_expect_success 'start flux-accounting service' ' | ||
flux account-service -p ${DB_PATH} -t | ||
' | ||
|
||
test_expect_success 'add banks' ' | ||
flux account add-bank root 1 && | ||
flux account add-bank --parent-bank=root A 1 | ||
' | ||
|
||
test_expect_success 'add an association' ' | ||
flux account add-user --username=user1 --userid=5001 --bank=A | ||
' | ||
|
||
test_expect_success 'send flux-accounting DB information to the plugin' ' | ||
flux account-priority-update -p ${DB_PATH} | ||
' | ||
|
||
test_expect_success 'submit 2 jobs that take up 1 node each; check resource counts' ' | ||
job1=$(flux python ${SUBMIT_AS} 5001 -N1 sleep 60) && | ||
flux job wait-event -f json ${job1} priority && | ||
job2=$(flux python ${SUBMIT_AS} 5001 -N1 sleep 60) && | ||
flux job wait-event -f json ${job2} priority && | ||
flux jobtap query mf_priority.so > query.json && | ||
test_debug "jq -S . <query.json" && | ||
jq -e ".mf_priority_map[] | select(.userid == 5001) | .banks[0].cur_nodes == 2" <query.json && | ||
jq -e ".mf_priority_map[] | select(.userid == 5001) | .banks[0].cur_cores == 2" <query.json | ||
' | ||
|
||
test_expect_success 'cancel jobs; check resource counts' ' | ||
flux cancel ${job1} && | ||
flux job wait-event -f json ${job1} clean && | ||
flux cancel ${job2} && | ||
flux job wait-event -f json ${job2} clean && | ||
flux jobtap query mf_priority.so > query.json && | ||
test_debug "jq -S . <query.json" && | ||
jq -e ".mf_priority_map[] | select(.userid == 5001) | .banks[0].cur_nodes == 0" <query.json && | ||
jq -e ".mf_priority_map[] | select(.userid == 5001) | .banks[0].cur_cores == 0" <query.json | ||
' | ||
|
||
test_expect_success 'submit a job that takes up one core' ' | ||
job3=$(flux python ${SUBMIT_AS} 5001 -n1 sleep 60) && | ||
flux job wait-event -f json ${job3} priority && | ||
flux jobtap query mf_priority.so > query.json && | ||
test_debug "jq -S . <query.json" && | ||
jq -e ".mf_priority_map[] | select(.userid == 5001) | .banks[0].cur_nodes == 0" <query.json && | ||
jq -e ".mf_priority_map[] | select(.userid == 5001) | .banks[0].cur_cores == 1" <query.json | ||
' | ||
|
||
test_expect_success 'cancel job; check resource counts' ' | ||
flux cancel ${job3} && | ||
flux job wait-event -f json ${job3} clean && | ||
flux jobtap query mf_priority.so > query.json && | ||
test_debug "jq -S . <query.json" && | ||
jq -e ".mf_priority_map[] | select(.userid == 5001) | .banks[0].cur_nodes == 0" <query.json && | ||
jq -e ".mf_priority_map[] | select(.userid == 5001) | .banks[0].cur_cores == 0" <query.json | ||
' | ||
|
||
test_done |