def test_setvcpus(self): ''' Test to changes the amount of vcpus allocated to VM. ''' with patch.object(xapi, "_get_xapi_session", MagicMock()): mock = MagicMock(side_effect=[False, ["a", "b", "c"]]) with patch.object(xapi, "_get_label_uuid", mock): self.assertFalse(xapi.setvcpus("salt", "1")) self.assertTrue(xapi.setvcpus("salt", "1")) with patch.object(xapi, "_check_xenapi", MagicMock(return_value=Mockxapi)): mock = MagicMock(return_value=True) with patch.dict(xapi.__salt__, {'config.option': mock}): with patch.object(xapi, "_get_label_uuid", mock): self.assertFalse(xapi.setvcpus("salt", "1"))