valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-07 17:09:03 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #18954 from rupeshta/at_unit_tests

Browse Source 
created unit test case for function atq from module at
This commit is contained in:
Nicole Thomas 2014-12-15 08:55:18 -07:00
commit 2decd9de47
1 changed files with 141 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

141
tests/unit/modules/at_test.py Normal file
View File

@ -0,0 +1,141 @@
# -*- coding: utf-8 -*-
'''
:codeauthor: :email:`Rupesh Tare <rupesht@saltstack.com>`
'''
# Import Salt Testing Libs
from salttesting import TestCase, skipIf
from salttesting.mock import (
MagicMock,
patch,
NO_MOCK,
NO_MOCK_REASON
)
# Import Salt Libs
from salt.modules import at
# Import salt libs
import salt.utils
# Globals
at.__grains__ = {}
@skipIf(NO_MOCK, NO_MOCK_REASON)
class AtTestCase(TestCase):
'''
TestCase for the salt.modules.at module
'''
atq_output = {'jobs': [{'date': '2014-12-11', 'job': 101, 'queue': 'A',
'tag': '', 'time': '19:48:47', 'user': 'B'}]}
@patch('salt.modules.at._cmd', MagicMock(return_value=None))
def test_atq_not_available(self):
'''
Tests the at.atq not available for any type of os_family.
'''
with patch.dict(at.__grains__, {'os_family': 'RedHat'}):
self.assertEqual(at.atq(), '\'at.atq\' is not available.')
with patch.dict(at.__grains__, {'os_family': ''}):
self.assertEqual(at.atq(), '\'at.atq\' is not available.')
@patch('salt.modules.at._cmd', MagicMock(return_value=''))
def test_atq_no_jobs_available(self):
'''
Tests the no jobs available for any type of os_family.
'''
with patch.dict(at.__grains__, {'os_family': 'RedHat'}):
self.assertDictEqual(at.atq(), {'jobs': []})
with patch.dict(at.__grains__, {'os_family': ''}):
self.assertDictEqual(at.atq(), {'jobs': []})
@patch('salt.modules.at._cmd')
def test_atq_list(self, salt_modules_at__cmd_mock):
'''
Tests the list all queued and running jobs.
'''
salt_modules_at__cmd_mock.return_value = '101\tThu Dec 11 \
19:48:47 2014 A B'
with patch.dict(at.__grains__, {'os_family': '', 'os': ''}):
self.assertDictEqual(at.atq(), {'jobs': [{'date': '2014-12-11',
'job': 101,
'queue': 'A',
'tag': '',
'time': '19:48:00',
'user': 'B'}]})
salt_modules_at__cmd_mock.return_value = '101\t2014-12-11 \
19:48:47 A B'
with patch.dict(at.__grains__, {'os_family': 'RedHat', 'os': ''}):
self.assertDictEqual(at.atq(), {'jobs': [{'date': '2014-12-11',
'job': 101,
'queue': 'A',
'tag': '',
'time': '19:48:47',
'user': 'B'}]})
salt_modules_at__cmd_mock.return_value = 'SALT: Dec 11, \
2014 19:48 A 101 B'
with patch.dict(at.__grains__, {'os_family': '', 'os': 'OpenBSD'}):
self.assertDictEqual(at.atq(), {'jobs': [{'date': '2014-12-11',
'job': '101',
'queue': 'B',
'tag': '',
'time': '19:48:00',
'user': 'A'}]})
@patch('salt.modules.at.atq', MagicMock(return_value=atq_output))
def test_atrm(self):
"""
Tests for remove jobs from the queue.
"""
with patch.object(salt.utils, 'which', return_value=None):
self.assertEqual(at.atrm(), "'at.atrm' is not available.")
with patch.object(salt.utils, 'which', return_value=True):
self.assertDictEqual(at.atrm(), {'jobs': {'removed': [],
'tag': None}})
with patch.object(at, '_cmd', return_value=True):
with patch.object(salt.utils, 'which', return_value=True):
self.assertDictEqual(at.atrm('all'),
{'jobs': {'removed': ['101'],
'tag': None}})
with patch.object(at, '_cmd', return_value=True):
with patch.object(salt.utils, 'which', return_value=True):
self.assertDictEqual(at.atrm(101),
{'jobs': {'removed': ['101'],
'tag': None}})
with patch.object(at, '_cmd', return_value=None):
self.assertEqual(at.atrm(101), '\'at.atrm\' is not available.')
@patch('salt.modules.at.atq', MagicMock(return_value=atq_output))
def test_jobcheck(self):
"""
Tests for check the job from queue.
"""
self.assertDictEqual(at.jobcheck(),
{'error': 'You have given a condition'})
self.assertDictEqual(at.jobcheck(runas='foo'),
{'note': 'No match jobs or time format error',
'jobs': []})
self.assertDictEqual(at.jobcheck(runas='B', tag='', hour=19, minute=48,
day=11, month=12, Year=2014),
{'jobs': [{'date': '2014-12-11',
'job': 101,
'queue': 'A',
'tag': '',
'time': '19:48:47',
'user': 'B'}]})
if __name__ == '__main__':
from integration import run_tests
run_tests(AtTestCase, needs_daemon=False)