def test_at(self):
"""
Tests for add a job to the queue.
"""
with patch("salt.modules.at.atq",
MagicMock(return_value=self.atq_output)):
self.assertDictEqual(at.at(), {"jobs": []})
with patch.object(salt.utils.path, "which", return_value=None):
self.assertEqual(
at.at("12:05am", "/sbin/reboot", tag="reboot"),
"'at.at' is not available.",
)
with patch.object(salt.utils.path, "which", return_value=True):
with patch.dict(at.__grains__, {"os_family": "RedHat"}):
mock = MagicMock(return_value=None)
with patch.dict(at.__salt__, {"cmd.run": mock}):
self.assertEqual(
at.at("12:05am", "/sbin/reboot", tag="reboot"),
"'at.at' is not available.",
)
mock = MagicMock(return_value="Garbled time")
with patch.dict(at.__salt__, {"cmd.run": mock}):
self.assertDictEqual(
at.at("12:05am", "/sbin/reboot", tag="reboot"),
{
"jobs": [],
"error": "invalid timespec"
},
)
mock = MagicMock(return_value="warning: commands\nA B")
with patch.dict(at.__salt__, {"cmd.run": mock}):
with patch.dict(at.__grains__, {"os": "OpenBSD"}):
self.assertDictEqual(
at.at("12:05am", "/sbin/reboot", tag="reboot"),
{
"jobs": [{
"date": "2014-12-11",
"job": 101,
"queue": "A",
"tag": "",
"time": "19:48:47",
"user": "******",
}]
},
)
with patch.dict(at.__grains__, {"os_family": ""}):
mock = MagicMock(return_value=None)
with patch.dict(at.__salt__, {"cmd.run": mock}):
self.assertEqual(
at.at("12:05am", "/sbin/reboot", tag="reboot"),
"'at.at' is not available.",
)
def test_at(self):
"""
Tests for add a job to the queue.
"""
with patch('salt.modules.at.atq',
MagicMock(return_value=self.atq_output)):
self.assertDictEqual(at.at(), {'jobs': []})
with patch.object(salt.utils.path, 'which', return_value=None):
self.assertEqual(
at.at('12:05am', '/sbin/reboot', tag='reboot'),
"'at.at' is not available.")
with patch.object(salt.utils.path, 'which', return_value=True):
with patch.dict(at.__grains__, {'os_family': 'RedHat'}):
mock = MagicMock(return_value=None)
with patch.dict(at.__salt__, {'cmd.run': mock}):
self.assertEqual(
at.at('12:05am', '/sbin/reboot', tag='reboot'),
"'at.at' is not available.")
mock = MagicMock(return_value='Garbled time')
with patch.dict(at.__salt__, {'cmd.run': mock}):
self.assertDictEqual(
at.at('12:05am', '/sbin/reboot', tag='reboot'), {
'jobs': [],
'error': 'invalid timespec'
})
mock = MagicMock(return_value='warning: commands\nA B')
with patch.dict(at.__salt__, {'cmd.run': mock}):
with patch.dict(at.__grains__, {'os': 'OpenBSD'}):
self.assertDictEqual(
at.at('12:05am', '/sbin/reboot', tag='reboot'),
{
'jobs': [{
'date': '2014-12-11',
'job': 101,
'queue': 'A',
'tag': '',
'time': '19:48:47',
'user': '******'
}]
})
with patch.dict(at.__grains__, {'os_family': ''}):
mock = MagicMock(return_value=None)
with patch.dict(at.__salt__, {'cmd.run': mock}):
self.assertEqual(
at.at('12:05am', '/sbin/reboot', tag='reboot'),
"'at.at' is not available.")
def test_at(self):
"""
Tests for add a job to the queue.
"""
self.assertDictEqual(at.at(), {'jobs': []})
with patch.object(salt.utils, 'which', return_value=None):
self.assertEqual(at.at('12:05am', '/sbin/reboot', tag='reboot'),
"'at.at' is not available.")
with patch.object(salt.utils, 'which', return_value=True):
with patch.dict(at.__grains__, {'os_family': 'RedHat'}):
mock = MagicMock(return_value=None)
with patch.dict(at.__salt__, {'cmd.run': mock}):
self.assertEqual(at.at('12:05am', '/sbin/reboot',
tag='reboot'),
"'at.at' is not available.")
mock = MagicMock(return_value='Garbled time')
with patch.dict(at.__salt__, {'cmd.run': mock}):
self.assertDictEqual(at.at('12:05am', '/sbin/reboot',
tag='reboot'),
{'jobs': [],
'error': 'invalid timespec'})
mock = MagicMock(return_value='warning: commands\nA B')
with patch.dict(at.__salt__, {'cmd.run': mock}):
with patch.dict(at.__grains__, {'os': 'OpenBSD'}):
self.assertDictEqual(at.at('12:05am', '/sbin/reboot',
tag='reboot'),
{'jobs': [{'date': '2014-12-11',
'job': 101,
'queue': 'A',
'tag': '',
'time': '19:48:47',
'user': '******'}]})
with patch.dict(at.__grains__, {'os_family': ''}):
mock = MagicMock(return_value=None)
with patch.dict(at.__salt__, {'cmd.run': mock}):
self.assertEqual(at.at('12:05am', '/sbin/reboot',
tag='reboot'),
"'at.at' is not available.")
