1*8da2a883SGabriele Monaco // SPDX-License-Identifier: GPL-2.0 2*8da2a883SGabriele Monaco #include <linux/kernel.h> 3*8da2a883SGabriele Monaco #include <linux/rv.h> 4*8da2a883SGabriele Monaco #include <rv/kunit.h> 5*8da2a883SGabriele Monaco #include <trace/events/sched.h> 6*8da2a883SGabriele Monaco #include "sts_kunit.h" 7*8da2a883SGabriele Monaco 8*8da2a883SGabriele Monaco #if IS_REACHABLE(CONFIG_RV_MON_STS) 9*8da2a883SGabriele Monaco 10*8da2a883SGabriele Monaco static void rv_test_sts(struct kunit *test) 11*8da2a883SGabriele Monaco { 12*8da2a883SGabriele Monaco struct task_struct *target = rv_kunit_alloc_mock_task(test); 13*8da2a883SGabriele Monaco struct task_struct *other = rv_kunit_alloc_mock_task(test); 14*8da2a883SGabriele Monaco struct rv_kunit_ctx *ctx = test->priv; 15*8da2a883SGabriele Monaco 16*8da2a883SGabriele Monaco prepare_test(test, &rv_sts_ops.mon); 17*8da2a883SGabriele Monaco /* Per-CPU monitor, make sure we don't change CPU mid-test */ 18*8da2a883SGabriele Monaco guard(migrate)(); 19*8da2a883SGabriele Monaco 20*8da2a883SGabriele Monaco /* Switch without disabling interrupts */ 21*8da2a883SGabriele Monaco rv_sts_ops.handle_schedule_exit(NULL, false); 22*8da2a883SGabriele Monaco rv_sts_ops.handle_schedule_entry(NULL, false); 23*8da2a883SGabriele Monaco RV_KUNIT_EXPECT_REACTION_HERE(test, ctx) 24*8da2a883SGabriele Monaco rv_sts_ops.handle_sched_switch(NULL, 0, target, other, TASK_RUNNING); 25*8da2a883SGabriele Monaco 26*8da2a883SGabriele Monaco rv_sts_ops.handle_schedule_exit(NULL, false); 27*8da2a883SGabriele Monaco 28*8da2a883SGabriele Monaco /* Schedule from interrupt context */ 29*8da2a883SGabriele Monaco rv_sts_ops.handle_schedule_entry(NULL, false); 30*8da2a883SGabriele Monaco rv_sts_ops.handle_irq_disable(NULL, 0, 0); 31*8da2a883SGabriele Monaco rv_sts_ops.handle_irq_entry(NULL, 0, NULL); 32*8da2a883SGabriele Monaco RV_KUNIT_EXPECT_REACTION_HERE(test, ctx) 33*8da2a883SGabriele Monaco rv_sts_ops.handle_sched_switch(NULL, 0, target, other, TASK_RUNNING); 34*8da2a883SGabriele Monaco rv_sts_ops.handle_irq_enable(NULL, 0, 0); 35*8da2a883SGabriele Monaco } 36*8da2a883SGabriele Monaco 37*8da2a883SGabriele Monaco #else 38*8da2a883SGabriele Monaco #define rv_test_sts rv_test_stub 39*8da2a883SGabriele Monaco #endif 40