prepare some test data, to ensure parsing from a pointer works.

Author: tazend

pyslurm/core/slurmctld/stats.pxd

@@ -36,6 +36,7 @@
 # cython: c_string_type=unicode, c_string_encoding=default
 # cython: language_level=3
 
+from libc.string cimport memset
 from pyslurm cimport slurm
 from pyslurm.slurm cimport (
     stats_info_response_msg_t,
@@ -44,6 +45,7 @@ from pyslurm.slurm cimport (
     slurm_reset_statistics,
     slurm_free_stats_response_msg,
     xfree,
+    xmalloc,
 )
 from pyslurm.utils cimport cstr
 from libc.stdint cimport uint8_t, uint16_t, uint32_t, uint64_t, int64_t

pyslurm/core/slurmctld/stats.pyx

@@ -427,3 +427,68 @@ cdef parse_response(stats_info_response_msg_t *ptr):
     out.backfill_exit = BackfillExitStatistics.from_ptr(ptr)
 
     return out
+
+
+# Prepare some test data
+def _parse_test_data():
+    import datetime
+
+    cdef stats_info_response_msg_t stats
+    memset(&stats, 0, sizeof(stats))
+
+    stats.req_time = int(datetime.datetime.now().timestamp())
+    stats.req_time_start = int(datetime.datetime.now().timestamp()) - 200
+    stats.jobs_submitted = 20
+    stats.jobs_running = 3
+    stats.schedule_cycle_counter = 10
+    stats.schedule_cycle_last = 40
+    stats.schedule_cycle_sum = 45
+
+    stats.bf_cycle_counter = 100
+    stats.bf_active = 0
+    stats.bf_backfilled_jobs = 10
+    stats.bf_cycle_sum = 200
+    stats.bf_depth_try_sum = 300
+    stats.bf_queue_len_sum = 600
+    stats.bf_table_size_sum = 200
+
+    stats.rpc_type_size = 3
+    stats.rpc_type_id = <uint16_t*>xmalloc(sizeof(uint16_t) * stats.rpc_type_size)
+    stats.rpc_type_cnt = <uint32_t*>xmalloc(sizeof(uint32_t) * stats.rpc_type_size)
+    stats.rpc_type_time = <uint64_t*>xmalloc(sizeof(uint64_t) * stats.rpc_type_size)
+
+    for i in range(stats.rpc_type_size):
+        stats.rpc_type_id[i] = 2000+i
+        stats.rpc_type_cnt[i] = i+1
+        stats.rpc_type_time[i] = i+2
+
+    stats.rpc_user_size = 1
+    stats.rpc_user_id = <uint32_t*>xmalloc(sizeof(uint32_t) * stats.rpc_user_size)
+    stats.rpc_user_cnt = <uint32_t*>xmalloc(sizeof(uint32_t) * stats.rpc_user_size)
+    stats.rpc_user_time = <uint64_t*>xmalloc(sizeof(uint64_t) * stats.rpc_user_size)
+
+    for i in range(stats.rpc_user_size):
+        stats.rpc_user_id[i] = i
+        stats.rpc_user_cnt[i] = i+1
+        stats.rpc_user_time[i] = i+2
+
+    stats.bf_exit_cnt = BF_EXIT_COUNT
+    stats.bf_exit = <uint32_t*>xmalloc(sizeof(uint32_t) * BF_EXIT_COUNT)
+    for i in range(stats.bf_exit_cnt):
+        stats.bf_exit[i] = i+1
+
+    stats.schedule_exit_cnt = SCHED_EXIT_COUNT
+    stats.schedule_exit = <uint32_t*>xmalloc(sizeof(uint32_t) * SCHED_EXIT_COUNT)
+
+    for i in range(stats.schedule_exit_cnt):
+        stats.schedule_exit[i] = i+1
+
+    stats.rpc_queue_type_count = 5
+    stats.rpc_queue_count = <uint32_t*>xmalloc(sizeof(uint32_t) * stats.rpc_queue_type_count)
+    stats.rpc_queue_type_id = <uint32_t*>xmalloc(sizeof(uint32_t) * stats.rpc_queue_type_count)
+
+    for i in range(stats.rpc_queue_type_count):
+        stats.rpc_queue_count[i] = i+1
+        stats.rpc_queue_type_id[i] = 2000+i
+
+    return parse_response(&stats)

tests/unit/test_slurmctld.py

@@ -0,0 +1,69 @@
+#########################################################################
+# test_slurmctld.py - slurmctld unit tests
+#########################################################################
+# Copyright (C) 2025 Toni Harzendorf <[email protected]>
+#
+# This file is part of PySlurm
+#
+# PySlurm is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+
+# PySlurm is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with PySlurm; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+"""test_slurmctld.py - Unit test basic slurmctld functionalities."""
+
+import pyslurm
+from pyslurm import slurmctld
+from pyslurm.core.slurmctld.stats import _parse_test_data
+
+
+def test_statistics():
+    stats = _parse_test_data()
+    assert stats.to_dict()
+
+    assert len(stats.rpcs_by_type) == 3
+    for typ, val in stats.rpcs_by_type.items():
+        assert val.count > 0
+        assert val.time > 0
+        assert val.average_time > 0
+        assert typ is not None
+
+    assert len(stats.rpcs_pending) == 5
+    for typ, val in stats.rpcs_pending.items():
+        assert val.count > 0
+        assert typ is not None
+        assert isinstance(typ, str)
+
+    assert len(stats.rpcs_by_user) == 1
+    for typ, val in stats.rpcs_by_user.items():
+        assert val.user_id == 0
+        assert val.user_name == "root"
+        assert val.time > 0
+        assert val.average_time > 0
+        assert typ is not None
+
+    assert stats.schedule_exit
+    assert stats.backfill_exit
+    assert stats.jobs_submitted == 20
+    assert stats.jobs_running == 3
+    assert stats.schedule_cycle_last == 40
+    assert stats.schedule_cycle_sum == 45
+    assert stats.schedule_cycle_mean == 4
+    assert stats.schedule_cycle_counter == 10
+
+    assert stats.backfill_cycle_counter == 100
+    assert stats.backfill_active is False
+    assert stats.backfilled_jobs == 10
+    assert stats.backfill_cycle_sum == 200
+    assert stats.backfill_depth_try_sum == 300
+    assert stats.backfill_queue_length_sum == 600
+    assert stats.backfill_table_size_sum == 200
+    assert stats.backfill_cycle_mean == 2