mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
adding memcached unit test case
This commit is contained in:
Jayesh Kariya
parent
3ecaee37c5
commit
697099c469
469
tests/unit/modules/memcached_test.py
Normal file
469
tests/unit/modules/memcached_test.py
Normal file
@ -0,0 +1,469 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
'''
|
||||
:codeauthor: :email:`Jayesh Kariya <jayeshk@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 memcached
|
||||
from salt.exceptions import CommandExecutionError, SaltInvocationError
|
||||
from salt.ext.six import integer_types
|
||||
|
||||
|
||||
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
||||
class MemcachedTestCase(TestCase):
|
||||
'''
|
||||
Test cases for salt.modules.memcached
|
||||
'''
|
||||
# 'status' function tests: 2
|
||||
|
||||
def test_status(self):
|
||||
'''
|
||||
Test if it gets memcached status
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return [('127.0.0.1:11211 (1)', {})]
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertDictEqual(memcached.status(),
|
||||
{'127.0.0.1:11211 (1)': {}})
|
||||
|
||||
def test_status_false(self):
|
||||
'''
|
||||
Test if it gets memcached status
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return []
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertFalse(memcached.status())
|
||||
|
||||
# 'get' function tests: 1
|
||||
|
||||
def test_get(self):
|
||||
'''
|
||||
Test if it retrieve value for a key
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return [('127.0.0.1:11211 (1)', {})]
|
||||
|
||||
@staticmethod
|
||||
def get(key):
|
||||
"""
|
||||
Mock of get method
|
||||
"""
|
||||
return key
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertEqual(memcached.get('salt'), 'salt')
|
||||
|
||||
# 'set_' function tests: 1
|
||||
|
||||
def test_set(self):
|
||||
'''
|
||||
Test if it set a key on the memcached server
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
def __init__(self):
|
||||
self.key = None
|
||||
self.value = None
|
||||
self.time = None
|
||||
self.min_compress_len = None
|
||||
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return [('127.0.0.1:11211 (1)', {})]
|
||||
|
||||
def set(self, key, value, time, min_compress_len):
|
||||
"""
|
||||
Mock of set method
|
||||
"""
|
||||
self.key = key
|
||||
self.value = value
|
||||
self.time = time
|
||||
self.min_compress_len = min_compress_len
|
||||
return True
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertTrue(memcached.set_('salt', '1111'))
|
||||
|
||||
self.assertRaises(SaltInvocationError, memcached.set_,
|
||||
'salt', '1111', time='0.1')
|
||||
|
||||
self.assertRaises(SaltInvocationError, memcached.set_,
|
||||
'salt', '1111', min_compress_len='0.1')
|
||||
|
||||
# 'delete' function tests: 1
|
||||
|
||||
def test_delete(self):
|
||||
'''
|
||||
Test if it delete a key from memcache server
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
def __init__(self):
|
||||
self.key = None
|
||||
self.time = None
|
||||
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return [('127.0.0.1:11211 (1)', {})]
|
||||
|
||||
def delete(self, key, time):
|
||||
"""
|
||||
Mock of delete method
|
||||
"""
|
||||
self.key = key
|
||||
self.time = time
|
||||
return True
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertTrue(memcached.delete('salt'))
|
||||
|
||||
self.assertRaises(SaltInvocationError, memcached.delete,
|
||||
'salt', '1111', time='0.1')
|
||||
|
||||
# 'add' function tests: 1
|
||||
|
||||
def test_add(self):
|
||||
'''
|
||||
Test if it add a key from memcache server
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
def __init__(self):
|
||||
self.key = None
|
||||
self.value = None
|
||||
self.time = None
|
||||
self.min_compress_len = None
|
||||
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return [('127.0.0.1:11211 (1)', {})]
|
||||
|
||||
def add(self, key, value, time, min_compress_len):
|
||||
"""
|
||||
Mock of add method
|
||||
"""
|
||||
self.key = key
|
||||
self.value = value
|
||||
self.time = time
|
||||
self.min_compress_len = min_compress_len
|
||||
return True
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertTrue(memcached.add('salt', '1111'))
|
||||
|
||||
self.assertRaises(SaltInvocationError, memcached.add,
|
||||
'salt', '1111', time='0.1')
|
||||
|
||||
self.assertRaises(SaltInvocationError, memcached.add,
|
||||
'salt', '1111', min_compress_len='0.1')
|
||||
|
||||
# 'replace' function tests: 1
|
||||
|
||||
def test_replace(self):
|
||||
'''
|
||||
Test if it replace a key from memcache server
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
def __init__(self):
|
||||
self.key = None
|
||||
self.value = None
|
||||
self.time = None
|
||||
self.min_compress_len = None
|
||||
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return [('127.0.0.1:11211 (1)', {})]
|
||||
|
||||
def replace(self, key, value, time, min_compress_len):
|
||||
"""
|
||||
Mock of replace method
|
||||
"""
|
||||
self.key = key
|
||||
self.value = value
|
||||
self.time = time
|
||||
self.min_compress_len = min_compress_len
|
||||
return True
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertTrue(memcached.replace('salt', '1111'))
|
||||
|
||||
self.assertRaises(SaltInvocationError, memcached.replace,
|
||||
'salt', '1111', time='0.1')
|
||||
|
||||
self.assertRaises(SaltInvocationError, memcached.replace,
|
||||
'salt', '1111', min_compress_len='0.1')
|
||||
|
||||
# 'increment' function tests: 3
|
||||
|
||||
def test_increment(self):
|
||||
'''
|
||||
Test if it increment the value of a key
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
def __init__(self):
|
||||
self.key = None
|
||||
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return [('127.0.0.1:11211 (1)', {})]
|
||||
|
||||
def get(self, key):
|
||||
"""
|
||||
Mock of get method
|
||||
"""
|
||||
self.key = key
|
||||
return 1
|
||||
|
||||
def incr(self, key, delta):
|
||||
"""
|
||||
Mock of incr method
|
||||
"""
|
||||
self.key = key
|
||||
if not isinstance(delta, integer_types):
|
||||
raise SaltInvocationError('Delta value must be an integer')
|
||||
return key
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertEqual(memcached.increment('salt'), 'salt')
|
||||
|
||||
self.assertRaises(SaltInvocationError, memcached.increment,
|
||||
'salt', delta='sa')
|
||||
|
||||
def test_increment_exist(self):
|
||||
'''
|
||||
Test if it increment the value of a key
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
def __init__(self):
|
||||
self.key = None
|
||||
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return [('127.0.0.1:11211 (1)', {})]
|
||||
|
||||
def get(self, key):
|
||||
"""
|
||||
Mock of get method
|
||||
"""
|
||||
self.key = key
|
||||
return key
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertRaises(CommandExecutionError, memcached.increment,
|
||||
'salt')
|
||||
|
||||
def test_increment_none(self):
|
||||
'''
|
||||
Test if it increment the value of a key
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
def __init__(self):
|
||||
self.key = None
|
||||
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return [('127.0.0.1:11211 (1)', {})]
|
||||
|
||||
def get(self, key):
|
||||
"""
|
||||
Mock of get method
|
||||
"""
|
||||
self.key = key
|
||||
return None
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertRaises(CommandExecutionError, memcached.increment,
|
||||
'salt')
|
||||
|
||||
# 'decrement' function tests: 3
|
||||
|
||||
def test_decrement(self):
|
||||
'''
|
||||
Test if it decrement the value of a key
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
def __init__(self):
|
||||
self.key = None
|
||||
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return [('127.0.0.1:11211 (1)', {})]
|
||||
|
||||
def get(self, key):
|
||||
"""
|
||||
Mock of get method
|
||||
"""
|
||||
self.key = key
|
||||
return 1
|
||||
|
||||
def decr(self, key, delta):
|
||||
"""
|
||||
Mock of decr method
|
||||
"""
|
||||
self.key = key
|
||||
if not isinstance(delta, integer_types):
|
||||
raise SaltInvocationError('Delta value must be an integer')
|
||||
return key
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertEqual(memcached.decrement('salt'), 'salt')
|
||||
|
||||
self.assertRaises(SaltInvocationError, memcached.decrement,
|
||||
'salt', delta='sa')
|
||||
|
||||
def test_decrement_exist(self):
|
||||
'''
|
||||
Test if it decrement the value of a key
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
def __init__(self):
|
||||
self.key = None
|
||||
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return [('127.0.0.1:11211 (1)', {})]
|
||||
|
||||
def get(self, key):
|
||||
"""
|
||||
Mock of get method
|
||||
"""
|
||||
self.key = key
|
||||
return key
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertRaises(CommandExecutionError, memcached.decrement,
|
||||
'salt')
|
||||
|
||||
def test_decrement_none(self):
|
||||
'''
|
||||
Test if it decrement the value of a key
|
||||
'''
|
||||
class MockMemcache(object):
|
||||
"""
|
||||
Mock of memcache
|
||||
"""
|
||||
def __init__(self):
|
||||
self.key = None
|
||||
|
||||
@staticmethod
|
||||
def get_stats():
|
||||
"""
|
||||
Mock of stats method
|
||||
"""
|
||||
return [('127.0.0.1:11211 (1)', {})]
|
||||
|
||||
def get(self, key):
|
||||
"""
|
||||
Mock of get method
|
||||
"""
|
||||
self.key = key
|
||||
return None
|
||||
|
||||
with patch.object(memcached, '_connect',
|
||||
MagicMock(return_value=MockMemcache())):
|
||||
self.assertRaises(CommandExecutionError, memcached.decrement,
|
||||
'salt')
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
from integration import run_tests
|
||||
run_tests(MemcachedTestCase, needs_daemon=False)
|
||||
Loading…
Reference in New Issue
Block a user