Line data    Source code

       1             : /* SPDX-License-Identifier: LGPL-2.1+ */
       2             : /***
       3             :   Copyright © 2012 Holger Hans Peter Freyther
       4             : ***/
       5             : 
       6             : #include <sched.h>
       7             : 
       8             : #include "all-units.h"
       9             : #include "macro.h"
      10             : #include "manager.h"
      11             : #include "rm-rf.h"
      12             : #include "test-helper.h"
      13             : #include "tests.h"
      14             : 
      15           1 : int main(int argc, char *argv[]) {
      16           1 :         _cleanup_(rm_rf_physical_and_freep) char *runtime_dir = NULL;
      17           1 :         _cleanup_(manager_freep) Manager *m = NULL;
      18             :         Unit *idle_ok, *idle_bad, *rr_ok, *rr_bad, *rr_sched;
      19             :         Service *ser;
      20             :         int r;
      21             : 
      22           1 :         test_setup_logging(LOG_INFO);
      23             : 
      24           1 :         r = enter_cgroup_subroot();
      25           1 :         if (r == -ENOMEDIUM)
      26           0 :                 return log_tests_skipped("cgroupfs not available");
      27             : 
      28             :         /* prepare the test */
      29           1 :         assert_se(set_unit_path(get_testdata_dir()) >= 0);
      30           1 :         assert_se(runtime_dir = setup_fake_runtime_dir());
      31           1 :         r = manager_new(UNIT_FILE_USER, MANAGER_TEST_RUN_BASIC, &m);
      32           1 :         if (MANAGER_SKIP_TEST(r))
      33           0 :                 return log_tests_skipped_errno(r, "manager_new");
      34           1 :         assert_se(r >= 0);
      35           1 :         assert_se(manager_startup(m, NULL, NULL) >= 0);
      36             : 
      37             :         /* load idle ok */
      38           1 :         assert_se(manager_load_startable_unit_or_warn(m, "sched_idle_ok.service", NULL, &idle_ok) >= 0);
      39           1 :         ser = SERVICE(idle_ok);
      40           1 :         assert_se(ser->exec_context.cpu_sched_policy == SCHED_OTHER);
      41           1 :         assert_se(ser->exec_context.cpu_sched_priority == 0);
      42             : 
      43             :         /*
      44             :          * load idle bad. This should print a warning but we have no way to look at it.
      45             :          */
      46           1 :         assert_se(manager_load_startable_unit_or_warn(m, "sched_idle_bad.service", NULL, &idle_bad) >= 0);
      47           1 :         ser = SERVICE(idle_ok);
      48           1 :         assert_se(ser->exec_context.cpu_sched_policy == SCHED_OTHER);
      49           1 :         assert_se(ser->exec_context.cpu_sched_priority == 0);
      50             : 
      51             :         /*
      52             :          * load rr ok.
      53             :          * Test that the default priority is moving from 0 to 1.
      54             :          */
      55           1 :         assert_se(manager_load_startable_unit_or_warn(m, "sched_rr_ok.service", NULL, &rr_ok) >= 0);
      56           1 :         ser = SERVICE(rr_ok);
      57           1 :         assert_se(ser->exec_context.cpu_sched_policy == SCHED_RR);
      58           1 :         assert_se(ser->exec_context.cpu_sched_priority == 1);
      59             : 
      60             :         /*
      61             :          * load rr bad.
      62             :          * Test that the value of 0 and 100 is ignored.
      63             :          */
      64           1 :         assert_se(manager_load_startable_unit_or_warn(m, "sched_rr_bad.service", NULL, &rr_bad) >= 0);
      65           1 :         ser = SERVICE(rr_bad);
      66           1 :         assert_se(ser->exec_context.cpu_sched_policy == SCHED_RR);
      67           1 :         assert_se(ser->exec_context.cpu_sched_priority == 1);
      68             : 
      69             :         /*
      70             :          * load rr change.
      71             :          * Test that anything between 1 and 99 can be set.
      72             :          */
      73           1 :         assert_se(manager_load_startable_unit_or_warn(m, "sched_rr_change.service", NULL, &rr_sched) >= 0);
      74           1 :         ser = SERVICE(rr_sched);
      75           1 :         assert_se(ser->exec_context.cpu_sched_policy == SCHED_RR);
      76           1 :         assert_se(ser->exec_context.cpu_sched_priority == 99);
      77             : 
      78           1 :         return EXIT_SUCCESS;
      79             : }