mirror of
https://github.com/sprockets/sprockets.clients.memcached.git
synced 2024-11-29 03:00:23 +00:00
54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
"""
|
|
Tests for the sprockets.clients.memcached package
|
|
|
|
"""
|
|
import mock
|
|
import os
|
|
import pickle
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
import unittest
|
|
|
|
from sprockets.clients import memcached
|
|
|
|
|
|
class TestGetServers(unittest.TestCase):
|
|
|
|
def tearDown(self):
|
|
for key in ['MEMCACHED_SERVERS', 'TEST1_MEMCACHED_SERVERS']:
|
|
if key in os.environ:
|
|
del os.environ[key]
|
|
|
|
def test_get_servers_for_prefixed_key(self):
|
|
os.environ['TEST1_MEMCACHED_SERVERS'] = '1.1.1.1:11211,1.1.1.2:11211'
|
|
self.assertListEqual(memcached._get_servers('TEST1'),
|
|
['1.1.1.1:11211', '1.1.1.2:11211'])
|
|
|
|
def test_get_servers_for_non_prefixed_key(self):
|
|
os.environ['MEMCACHED_SERVERS'] = '2.1.1.1:11211'
|
|
self.assertListEqual(memcached._get_servers(None), ['2.1.1.1:11211'])
|
|
|
|
def test_get_servers_returns_default_value(self):
|
|
self.assertListEqual(memcached._get_servers(None), ['127.0.0.1:11211'])
|
|
|
|
|
|
class TestClientWrapsMemcacheClient(unittest.TestCase):
|
|
|
|
@mock.patch('memcache.Client.__init__')
|
|
def test_client_super_init(self, mock_init):
|
|
memcached.Client()
|
|
mock_init.assert_called_once_with(['127.0.0.1:11211'],
|
|
0,
|
|
0,
|
|
pickle.Pickler,
|
|
pickle.Unpickler,
|
|
None,
|
|
None,
|
|
memcached.SERVER_MAX_KEY_LENGTH,
|
|
memcached.SERVER_MAX_VALUE_LENGTH,
|
|
memcached._DEAD_RETRY,
|
|
memcached._SOCKET_TIMEOUT,
|
|
False,
|
|
0,
|
|
True)
|