add tests for reservations

Author: tazend

tests/integration/test_reservation.py

@@ -0,0 +1,60 @@
+#########################################################################
+# test_reservation.py - reservation integration 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_reservation.py - integration test reservation functionalities."""
+
+import pyslurm
+from datetime import datetime
+
+
+def test_api_calls():
+    start = datetime.now()
+    duration = "1-00:00:00"
+    resv = pyslurm.Reservation(
+        name="testing",
+        start_time=start,
+        duration=duration,
+        users=["root"],
+        node_count=1,
+    )
+    resv.create()
+
+    reservations = pyslurm.Reservations.load()
+    resv = reservations["testing"]
+    assert len(reservations) == 1
+    assert resv.name == "testing"
+    assert resv.to_dict()
+
+    assert resv.start_time == int(start.timestamp())
+    assert resv.duration == 60 * 24
+    assert resv.end_time == resv.start_time + (60 * 60 * 24)
+
+    resv.duration += 60 * 24
+    resv.modify()
+
+    resv = pyslurm.Reservation.load("testing")
+    assert resv.name == "testing"
+    assert resv.start_time == int(start.timestamp())
+    assert resv.duration == 2 * 60 * 24
+    assert resv.end_time == resv.start_time + (2 * 60 * 60 * 24)
+
+    resv.delete()
+    reservations = pyslurm.Reservations.load()
+    assert len(reservations) == 0

tests/unit/test_reservation.py

@@ -0,0 +1,64 @@
+#########################################################################
+# test_reservation.py - reservation 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_reservation.py - Unit test basic reservation functionalities."""
+
+import pyslurm
+from datetime import datetime
+
+
+def test_create_instance():
+    resv = pyslurm.Reservation("test")
+    assert resv.name == "test"
+    assert resv.accounts == []
+    assert resv.start_time == None
+    assert resv.end_time == None
+    assert resv.duration == 0
+    assert resv.is_active is False
+    assert resv.cpus_by_node == {}
+    assert resv.to_dict()
+
+    start = datetime.now()
+    resv.start_time = start
+    resv.duration = "1-00:00:00"
+
+    assert resv.start_time == int(start.timestamp())
+    assert resv.duration == 60 * 24
+    assert resv.end_time == resv.start_time + (60 * 60 * 24)
+
+    resv.duration += pyslurm.utils.timestr_to_mins("1-00:00:00")
+
+    assert resv.start_time == int(start.timestamp())
+    assert resv.duration == 2 * 60 * 24
+    assert resv.end_time == resv.start_time + (2 * 60 * 60 * 24)
+
+    start = datetime.fromisoformat("2022-04-03T06:00:00")
+    end = resv.end_time
+    resv.start_time = int(start.timestamp())
+
+    assert resv.start_time == int(start.timestamp())
+    assert resv.end_time == end
+    assert resv.duration == int((resv.end_time - resv.start_time) / 60)
+
+    duration = resv.duration
+    resv.end_time += 60 * 60 * 24
+    assert resv.start_time == int(start.timestamp())
+    assert resv.end_time == end + (60 * 60 * 24)
+    assert resv.duration == duration + (60 * 24)