# coding: utf-8 # Copyright (C) 1994-2021 Altair Engineering, Inc. # For more information, contact Altair at www.altair.com. # # This file is part of both the OpenPBS software ("OpenPBS") # and the PBS Professional ("PBS Pro") software. # # Open Source License Information: # # OpenPBS is free software. You can redistribute it and/or modify it under # the terms of the GNU Affero General Public License as published by the # Free Software Foundation, either version 3 of the License, or (at your # option) any later version. # # OpenPBS 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 Affero General Public # License for more details. # # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see . # # Commercial License Information: # # PBS Pro is commercially licensed software that shares a common core with # the OpenPBS software. For a copy of the commercial license terms and # conditions, go to: (http://www.pbspro.com/agreement.html) or contact the # Altair Legal Department. # # Altair's dual-license business model allows companies, individuals, and # organizations to create proprietary derivative works of OpenPBS and # distribute them - whether embedded or bundled with other software - # under a commercial license agreement. # # Use of Altair's trademarks, including but not limited to "PBS™", # "OpenPBS®", "PBS Professional®", and "PBS Pro™" and Altair's logos is # subject to Altair's trademark licensing policies. from tests.interfaces import * @tags('multisched') class TestNodePartition(TestInterfaces): """ Test suite to test partition attr for node """ def setUp(self): TestInterfaces.setUp(self) self.mom_hostname = self.mom.shortname def set_node_partition_attr(self, mgr_cmd="MGR_CMD_SET", n_name=None, partition="P1", user=ROOT_USER): """ Common function to set partition attribute to node object :param mgr_cmd: qmgr "MGR_CMD_SET/MGR_CMD_UNSET" cmd, Defaults to MGR_CMD_SET :type mgr_cmd: str :param n_name: name of the vnode, Defaults to "hostname of server" :type n_name: str :param partition: "partition" attribute of node object, Defaults to "P1" :type partition: str :param user: one of the pre-defined set of users :type user: :py:class:`~ptl.lib.pbs_testlib.PbsUser` """ attr = {'partition': partition} if mgr_cmd == "MGR_CMD_SET": self.server.manager(MGR_CMD_SET, NODE, attr, id=n_name, runas=user) elif mgr_cmd == "MGR_CMD_UNSET": self.server.manager(MGR_CMD_UNSET, NODE, "partition", id=n_name, runas=user) else: msg = ("Error: test_set_node_partition_attr function takes only " "MGR_CMD_SET/MGR_CMD_UNSET value for mgr_cmd") self.assertTrue(False, msg) def test_set_unset_partition_node_attr(self): """ Test to set/unset the partition attribute of node object """ self.set_node_partition_attr(n_name=self.mom_hostname) self.set_node_partition_attr(partition="P2", n_name=self.mom_hostname) # resetting the same partition value self.set_node_partition_attr(partition="P2", n_name=self.mom_hostname) self.set_node_partition_attr(mgr_cmd="MGR_CMD_UNSET", n_name=self.mom_hostname) def test_set_partition_node_attr_user_permissions(self): """ Test to check the user permissions for set/unset the partition attribute of node """ self.set_node_partition_attr(n_name=self.mom_hostname) msg1 = "Unauthorized Request" msg2 = "didn't receive expected error message" try: self.set_node_partition_attr(partition="P2", user=TEST_USER) except PbsManagerError as e: self.assertTrue(msg1 in e.msg[0], msg2) try: self.set_node_partition_attr(mgr_cmd="MGR_CMD_UNSET", user=TEST_USER) except PbsManagerError as e: # self.assertEqual(e.rc, 15007) # The above code has to be uncommented when the PTL framework # bug PP-881 gets fixed self.assertTrue(msg1 in e.msg[0], msg2) def test_partition_association_with_node_and_queue(self): """ Test to check the set of partition attribute and association between queue and node """ attr = {'queue_type': "execution", 'enabled': "True", 'started': "True", 'partition': "P1"} self.server.manager(MGR_CMD_CREATE, QUEUE, attr, id="Q1") self.set_node_partition_attr(n_name=self.mom_hostname) attr = {'queue_type': "execution", 'enabled': "True", 'started': "True", 'partition': "P2"} self.server.manager(MGR_CMD_CREATE, QUEUE, attr, id="Q2") self.set_node_partition_attr(mgr_cmd="MGR_CMD_UNSET", n_name=self.mom_hostname) self.server.manager(MGR_CMD_SET, NODE, { 'queue': "Q2"}, id=self.mom_hostname) self.set_node_partition_attr(partition="P2", n_name=self.mom_hostname) def test_mismatch_of_partition_on_node_and_queue(self): """ Test to check the set of partition attribute is disallowed if partition ids do not match on queue and node """ self.test_partition_association_with_node_and_queue() msg1 = "Invalid partition in queue" msg2 = "didn't receive expected error message" try: self.server.manager(MGR_CMD_SET, QUEUE, {'partition': "P1"}, id="Q2") except PbsManagerError as e: # self.assertEqual(e.rc, 15221) # The above code has to be uncommented when the PTL framework # bug PP-881 gets fixed self.assertTrue(msg1 in e.msg[0], msg2) msg1 = "Partition P2 is not part of queue for node" try: self.server.manager(MGR_CMD_SET, NODE, {'queue': "Q1"}, id=self.mom_hostname) except PbsManagerError as e: # self.assertEqual(e.rc, 15220) # The above code has to be uncommented when the PTL framework # bug PP-881 gets fixed self.assertTrue(msg1 in e.msg[0], msg2) msg1 = "Queue Q2 is not part of partition for node" try: self.set_node_partition_attr(n_name=self.mom_hostname) except PbsManagerError as e: # self.assertEqual(e.rc, 15219) # The above code has to be uncommented when the PTL framework # bug PP-881 gets fixed self.assertTrue(msg1 in e.msg[0], msg2)