[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]
[RFC PATCH 13/15] jobs: use job locks and helpers also in the unit tests
|
From: |
Emanuele Giuseppe Esposito |
|
Subject: |
[RFC PATCH 13/15] jobs: use job locks and helpers also in the unit tests |
|
Date: |
Fri, 29 Oct 2021 12:39:12 -0400 |
Add missing job synchronization in the unit tests, with
both explicit locks and helpers.
Signed-off-by: Emanuele Giuseppe Esposito <eesposit@redhat.com>
---
tests/unit/test-bdrv-drain.c | 40 +++++++++++-----------
tests/unit/test-block-iothread.c | 4 +++
tests/unit/test-blockjob-txn.c | 10 ++++++
tests/unit/test-blockjob.c | 57 +++++++++++++++++++++-----------
4 files changed, 72 insertions(+), 39 deletions(-)
diff --git a/tests/unit/test-bdrv-drain.c b/tests/unit/test-bdrv-drain.c
index 2d3c17e566..535c39b5a8 100644
--- a/tests/unit/test-bdrv-drain.c
+++ b/tests/unit/test-bdrv-drain.c
@@ -941,61 +941,63 @@ static void test_blockjob_common_drain_node(enum
drain_type drain_type,
}
}
- g_assert_cmpint(job->job.pause_count, ==, 0);
- g_assert_false(job->job.paused);
+ g_assert_cmpint(job_get_pause_count(&job->job), ==, 0);
+ g_assert_false(job_get_paused(&job->job));
g_assert_true(tjob->running);
- g_assert_true(job->job.busy); /* We're in qemu_co_sleep_ns() */
+ g_assert_true(job_get_busy(&job->job)); /* We're in qemu_co_sleep_ns() */
do_drain_begin_unlocked(drain_type, drain_bs);
if (drain_type == BDRV_DRAIN_ALL) {
/* bdrv_drain_all() drains both src and target */
- g_assert_cmpint(job->job.pause_count, ==, 2);
+ g_assert_cmpint(job_get_pause_count(&job->job), ==, 2);
} else {
- g_assert_cmpint(job->job.pause_count, ==, 1);
+ g_assert_cmpint(job_get_pause_count(&job->job), ==, 1);
}
- g_assert_true(job->job.paused);
- g_assert_false(job->job.busy); /* The job is paused */
+ g_assert_true(job_get_paused(&job->job));
+ g_assert_false(job_get_busy(&job->job)); /* The job is paused */
do_drain_end_unlocked(drain_type, drain_bs);
if (use_iothread) {
/* paused is reset in the I/O thread, wait for it */
- while (job->job.paused) {
+ while (job_get_paused(&job->job)) {
aio_poll(qemu_get_aio_context(), false);
}
}
- g_assert_cmpint(job->job.pause_count, ==, 0);
- g_assert_false(job->job.paused);
- g_assert_true(job->job.busy); /* We're in qemu_co_sleep_ns() */
+ g_assert_cmpint(job_get_pause_count(&job->job), ==, 0);
+ g_assert_false(job_get_paused(&job->job));
+ g_assert_true(job_get_busy(&job->job)); /* We're in qemu_co_sleep_ns() */
do_drain_begin_unlocked(drain_type, target);
if (drain_type == BDRV_DRAIN_ALL) {
/* bdrv_drain_all() drains both src and target */
- g_assert_cmpint(job->job.pause_count, ==, 2);
+ g_assert_cmpint(job_get_pause_count(&job->job), ==, 2);
} else {
- g_assert_cmpint(job->job.pause_count, ==, 1);
+ g_assert_cmpint(job_get_pause_count(&job->job), ==, 1);
}
- g_assert_true(job->job.paused);
- g_assert_false(job->job.busy); /* The job is paused */
+ g_assert_true(job_get_paused(&job->job));
+ g_assert_false(job_get_busy(&job->job)); /* The job is paused */
do_drain_end_unlocked(drain_type, target);
if (use_iothread) {
/* paused is reset in the I/O thread, wait for it */
- while (job->job.paused) {
+ while (job_get_paused(&job->job)) {
aio_poll(qemu_get_aio_context(), false);
}
}
- g_assert_cmpint(job->job.pause_count, ==, 0);
- g_assert_false(job->job.paused);
- g_assert_true(job->job.busy); /* We're in qemu_co_sleep_ns() */
+ g_assert_cmpint(job_get_pause_count(&job->job), ==, 0);
+ g_assert_false(job_get_paused(&job->job));
+ g_assert_true(job_get_busy(&job->job)); /* We're in qemu_co_sleep_ns() */
aio_context_acquire(ctx);
+ job_lock();
ret = job_complete_sync(&job->job, &error_abort);
+ job_unlock();
g_assert_cmpint(ret, ==, (result == TEST_JOB_SUCCESS ? 0 : -EIO));
if (use_iothread) {
diff --git a/tests/unit/test-block-iothread.c b/tests/unit/test-block-iothread.c
index aea660aeed..f39cb8b7ef 100644
--- a/tests/unit/test-block-iothread.c
+++ b/tests/unit/test-block-iothread.c
@@ -456,7 +456,9 @@ static void test_attach_blockjob(void)
}
aio_context_acquire(ctx);
+ job_lock();
job_complete_sync(&tjob->common.job, &error_abort);
+ job_unlock();
blk_set_aio_context(blk, qemu_get_aio_context(), &error_abort);
aio_context_release(ctx);
@@ -630,7 +632,9 @@ static void test_propagate_mirror(void)
BLOCKDEV_ON_ERROR_REPORT, BLOCKDEV_ON_ERROR_REPORT,
false, "filter_node", MIRROR_COPY_MODE_BACKGROUND,
&error_abort);
+ job_lock();
job = job_get("job0");
+ job_unlock();
filter = bdrv_find_node("filter_node");
/* Change the AioContext of src */
diff --git a/tests/unit/test-blockjob-txn.c b/tests/unit/test-blockjob-txn.c
index 8bd13b9949..1ae3a9d443 100644
--- a/tests/unit/test-blockjob-txn.c
+++ b/tests/unit/test-blockjob-txn.c
@@ -124,16 +124,20 @@ static void test_single_job(int expected)
job = test_block_job_start(1, true, expected, &result, txn);
job_start(&job->job);
+ job_lock();
if (expected == -ECANCELED) {
job_cancel(&job->job, false);
}
+ job_unlock();
while (result == -EINPROGRESS) {
aio_poll(qemu_get_aio_context(), true);
}
g_assert_cmpint(result, ==, expected);
+ job_lock();
job_txn_unref(txn);
+ job_unlock();
}
static void test_single_job_success(void)
@@ -168,6 +172,7 @@ static void test_pair_jobs(int expected1, int expected2)
/* Release our reference now to trigger as many nice
* use-after-free bugs as possible.
*/
+ job_lock();
job_txn_unref(txn);
if (expected1 == -ECANCELED) {
@@ -176,6 +181,7 @@ static void test_pair_jobs(int expected1, int expected2)
if (expected2 == -ECANCELED) {
job_cancel(&job2->job, false);
}
+ job_unlock();
while (result1 == -EINPROGRESS || result2 == -EINPROGRESS) {
aio_poll(qemu_get_aio_context(), true);
@@ -227,7 +233,9 @@ static void test_pair_jobs_fail_cancel_race(void)
job_start(&job1->job);
job_start(&job2->job);
+ job_lock();
job_cancel(&job1->job, false);
+ job_unlock();
/* Now make job2 finish before the main loop kicks jobs. This simulates
* the race between a pending kick and another job completing.
@@ -242,7 +250,9 @@ static void test_pair_jobs_fail_cancel_race(void)
g_assert_cmpint(result1, ==, -ECANCELED);
g_assert_cmpint(result2, ==, -ECANCELED);
+ job_lock();
job_txn_unref(txn);
+ job_unlock();
}
int main(int argc, char **argv)
diff --git a/tests/unit/test-blockjob.c b/tests/unit/test-blockjob.c
index 4c9e1bf1e5..b94e1510c9 100644
--- a/tests/unit/test-blockjob.c
+++ b/tests/unit/test-blockjob.c
@@ -211,8 +211,11 @@ static CancelJob *create_common(Job **pjob)
bjob = mk_job(blk, "Steve", &test_cancel_driver, true,
JOB_MANUAL_FINALIZE | JOB_MANUAL_DISMISS);
job = &bjob->job;
+ job_lock();
job_ref(job);
assert(job->status == JOB_STATUS_CREATED);
+ job_unlock();
+
s = container_of(bjob, CancelJob, common);
s->blk = blk;
@@ -230,6 +233,7 @@ static void cancel_common(CancelJob *s)
ctx = job->job.aio_context;
aio_context_acquire(ctx);
+ job_lock();
job_cancel_sync(&job->job, true);
if (sts != JOB_STATUS_CREATED && sts != JOB_STATUS_CONCLUDED) {
Job *dummy = &job->job;
@@ -237,6 +241,7 @@ static void cancel_common(CancelJob *s)
}
assert(job->job.status == JOB_STATUS_NULL);
job_unref(&job->job);
+ job_unlock();
destroy_blk(blk);
aio_context_release(ctx);
@@ -259,7 +264,7 @@ static void test_cancel_running(void)
s = create_common(&job);
job_start(job);
- assert(job->status == JOB_STATUS_RUNNING);
+ assert(job_get_status(job) == JOB_STATUS_RUNNING);
cancel_common(s);
}
@@ -272,11 +277,13 @@ static void test_cancel_paused(void)
s = create_common(&job);
job_start(job);
- assert(job->status == JOB_STATUS_RUNNING);
+ assert(job_get_status(job) == JOB_STATUS_RUNNING);
+ job_lock();
job_user_pause(job, &error_abort);
+ job_unlock();
job_enter(job);
- assert(job->status == JOB_STATUS_PAUSED);
+ assert(job_get_status(job) == JOB_STATUS_PAUSED);
cancel_common(s);
}
@@ -289,11 +296,11 @@ static void test_cancel_ready(void)
s = create_common(&job);
job_start(job);
- assert(job->status == JOB_STATUS_RUNNING);
+ assert(job_get_status(job) == JOB_STATUS_RUNNING);
s->should_converge = true;
job_enter(job);
- assert(job->status == JOB_STATUS_READY);
+ assert(job_get_status(job) == JOB_STATUS_READY);
cancel_common(s);
}
@@ -306,15 +313,17 @@ static void test_cancel_standby(void)
s = create_common(&job);
job_start(job);
- assert(job->status == JOB_STATUS_RUNNING);
+ assert(job_get_status(job) == JOB_STATUS_RUNNING);
s->should_converge = true;
job_enter(job);
- assert(job->status == JOB_STATUS_READY);
+ assert(job_get_status(job) == JOB_STATUS_READY);
+ job_lock();
job_user_pause(job, &error_abort);
+ job_unlock();
job_enter(job);
- assert(job->status == JOB_STATUS_STANDBY);
+ assert(job_get_status(job) == JOB_STATUS_STANDBY);
cancel_common(s);
}
@@ -327,20 +336,22 @@ static void test_cancel_pending(void)
s = create_common(&job);
job_start(job);
- assert(job->status == JOB_STATUS_RUNNING);
+ assert(job_get_status(job) == JOB_STATUS_RUNNING);
s->should_converge = true;
job_enter(job);
- assert(job->status == JOB_STATUS_READY);
+ assert(job_get_status(job) == JOB_STATUS_READY);
+ job_lock();
job_complete(job, &error_abort);
+ job_unlock();
job_enter(job);
while (!job->deferred_to_main_loop) {
aio_poll(qemu_get_aio_context(), true);
}
- assert(job->status == JOB_STATUS_READY);
+ assert(job_get_status(job) == JOB_STATUS_READY);
aio_poll(qemu_get_aio_context(), true);
- assert(job->status == JOB_STATUS_PENDING);
+ assert(job_get_status(job) == JOB_STATUS_PENDING);
cancel_common(s);
}
@@ -353,25 +364,29 @@ static void test_cancel_concluded(void)
s = create_common(&job);
job_start(job);
- assert(job->status == JOB_STATUS_RUNNING);
+ assert(job_get_status(job) == JOB_STATUS_RUNNING);
s->should_converge = true;
job_enter(job);
- assert(job->status == JOB_STATUS_READY);
+ assert(job_get_status(job) == JOB_STATUS_READY);
+ job_lock();
job_complete(job, &error_abort);
+ job_unlock();
job_enter(job);
while (!job->deferred_to_main_loop) {
aio_poll(qemu_get_aio_context(), true);
}
- assert(job->status == JOB_STATUS_READY);
+ assert(job_get_status(job) == JOB_STATUS_READY);
aio_poll(qemu_get_aio_context(), true);
- assert(job->status == JOB_STATUS_PENDING);
+ assert(job_get_status(job) == JOB_STATUS_PENDING);
aio_context_acquire(job->aio_context);
+ job_lock();
job_finalize(job, &error_abort);
+ job_unlock();
aio_context_release(job->aio_context);
- assert(job->status == JOB_STATUS_CONCLUDED);
+ assert(job_get_status(job) == JOB_STATUS_CONCLUDED);
cancel_common(s);
}
@@ -459,22 +474,23 @@ static void test_complete_in_standby(void)
bjob = mk_job(blk, "job", &test_yielding_driver, true,
JOB_MANUAL_FINALIZE | JOB_MANUAL_DISMISS);
job = &bjob->job;
- assert(job->status == JOB_STATUS_CREATED);
+ assert(job_get_status(job) == JOB_STATUS_CREATED);
/* Wait for the job to become READY */
job_start(job);
aio_context_acquire(ctx);
- AIO_WAIT_WHILE(ctx, job->status != JOB_STATUS_READY);
+ AIO_WAIT_WHILE(ctx, job_get_status(job) != JOB_STATUS_READY);
aio_context_release(ctx);
/* Begin the drained section, pausing the job */
bdrv_drain_all_begin();
- assert(job->status == JOB_STATUS_STANDBY);
+ assert(job_get_status(job) == JOB_STATUS_STANDBY);
/* Lock the IO thread to prevent the job from being run */
aio_context_acquire(ctx);
/* This will schedule the job to resume it */
bdrv_drain_all_end();
+ job_lock();
/* But the job cannot run, so it will remain on standby */
assert(job->status == JOB_STATUS_STANDBY);
@@ -489,6 +505,7 @@ static void test_complete_in_standby(void)
assert(job->status == JOB_STATUS_CONCLUDED);
job_dismiss(&job, &error_abort);
+ job_unlock();
destroy_blk(blk);
aio_context_release(ctx);
--
2.27.0
- [RFC PATCH 02/15] job.c: make job_lock/unlock public, (continued)
- [RFC PATCH 02/15] job.c: make job_lock/unlock public, Emanuele Giuseppe Esposito, 2021/10/29
- [RFC PATCH 07/15] job-driver.h: add helper functions, Emanuele Giuseppe Esposito, 2021/10/29
- [RFC PATCH 09/15] job.c: move inner aiocontext lock in callbacks, Emanuele Giuseppe Esposito, 2021/10/29
- [RFC PATCH 05/15] job-monitor.h: define the job monitor API, Emanuele Giuseppe Esposito, 2021/10/29
- [RFC PATCH 08/15] job.c: minor adjustments in preparation to job-driver, Emanuele Giuseppe Esposito, 2021/10/29
- [RFC PATCH 15/15] jobs: remove all unnecessary AioContext locks, Emanuele Giuseppe Esposito, 2021/10/29
- [RFC PATCH 10/15] aio-wait.h: introduce AIO_WAIT_WHILE_UNLOCKED, Emanuele Giuseppe Esposito, 2021/10/29
- [RFC PATCH 12/15] jobs: protect jobs with job_lock/unlock, Emanuele Giuseppe Esposito, 2021/10/29
- [RFC PATCH 11/15] jobs: remove aiocontext locks since the functions are under BQL, Emanuele Giuseppe Esposito, 2021/10/29
- [RFC PATCH 06/15] jobs: add job-driver.h, Emanuele Giuseppe Esposito, 2021/10/29
- [RFC PATCH 13/15] jobs: use job locks and helpers also in the unit tests,
Emanuele Giuseppe Esposito <=
- [RFC PATCH 14/15] jobs: add missing job locks to replace aiocontext lock, Emanuele Giuseppe Esposito, 2021/10/29