|
@@ -1,6 +1,6 @@
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
-from __future__ import absolute_import, with_statement
|
|
|
+from __future__ import absolute_import, unicode_literals
|
|
|
|
|
|
from celery.backends import riak as module
|
|
|
from celery.backends.riak import RiakBackend, riak
|
|
@@ -25,9 +25,6 @@ class test_RiakBackend(AppCase):
|
|
|
return self.app.backend
|
|
|
|
|
|
def test_init_no_riak(self):
|
|
|
- """
|
|
|
- test init no riak raises
|
|
|
- """
|
|
|
prev, module.riak = module.riak, None
|
|
|
try:
|
|
|
with self.assertRaises(ImproperlyConfigured):
|
|
@@ -36,20 +33,15 @@ class test_RiakBackend(AppCase):
|
|
|
module.riak = prev
|
|
|
|
|
|
def test_init_no_settings(self):
|
|
|
- """Test init no settings."""
|
|
|
self.app.conf.riak_backend_settings = []
|
|
|
with self.assertRaises(ImproperlyConfigured):
|
|
|
RiakBackend(app=self.app)
|
|
|
|
|
|
def test_init_settings_is_None(self):
|
|
|
- """
|
|
|
- Test init settings is None
|
|
|
- """
|
|
|
self.app.conf.riak_backend_settings = None
|
|
|
self.assertTrue(self.app.backend)
|
|
|
|
|
|
def test_get_client_client_exists(self):
|
|
|
- """Test get existing client."""
|
|
|
with patch('riak.client.RiakClient') as mock_connection:
|
|
|
self.backend._client = sentinel._client
|
|
|
|
|
@@ -60,13 +52,6 @@ class test_RiakBackend(AppCase):
|
|
|
self.assertFalse(mock_connection.called)
|
|
|
|
|
|
def test_get(self):
|
|
|
- """Test get
|
|
|
-
|
|
|
- RiakBackend.get
|
|
|
- should return and take two params
|
|
|
- db conn to riak is mocked
|
|
|
- TODO Should test on key not exists
|
|
|
- """
|
|
|
self.app.conf.couchbase_backend_settings = {}
|
|
|
self.backend._client = Mock(name='_client')
|
|
|
self.backend._bucket = Mock(name='_bucket')
|
|
@@ -77,13 +62,6 @@ class test_RiakBackend(AppCase):
|
|
|
self.backend._bucket.get.assert_called_once_with('1f3fab')
|
|
|
|
|
|
def test_set(self):
|
|
|
- """Test set
|
|
|
-
|
|
|
- RiakBackend.set
|
|
|
- should return None and take two params
|
|
|
- db conn to couchbase is mocked.
|
|
|
-
|
|
|
- """
|
|
|
self.app.conf.couchbase_backend_settings = None
|
|
|
self.backend._client = MagicMock()
|
|
|
self.backend._bucket = MagicMock()
|
|
@@ -92,14 +70,6 @@ class test_RiakBackend(AppCase):
|
|
|
self.assertIsNone(self.backend.set(sentinel.key, sentinel.value))
|
|
|
|
|
|
def test_delete(self):
|
|
|
- """Test get
|
|
|
-
|
|
|
- RiakBackend.get
|
|
|
- should return and take two params
|
|
|
- db conn to couchbase is mocked
|
|
|
- TODO Should test on key not exists
|
|
|
-
|
|
|
- """
|
|
|
self.app.conf.couchbase_backend_settings = {}
|
|
|
|
|
|
self.backend._client = Mock(name='_client')
|
|
@@ -111,11 +81,6 @@ class test_RiakBackend(AppCase):
|
|
|
self.backend._bucket.delete.assert_called_once_with('1f3fab')
|
|
|
|
|
|
def test_config_params(self):
|
|
|
- """
|
|
|
- test celery.conf.riak_backend_settingS
|
|
|
- celery.conf.riak_backend_settingS
|
|
|
- is properly set
|
|
|
- """
|
|
|
self.app.conf.riak_backend_settings = {
|
|
|
'bucket': 'mycoolbucket',
|
|
|
'host': 'there.host.com',
|
|
@@ -126,9 +91,6 @@ class test_RiakBackend(AppCase):
|
|
|
self.assertEqual(self.backend.port, 1234)
|
|
|
|
|
|
def test_backend_by_url(self, url='riak://myhost/mycoolbucket'):
|
|
|
- """
|
|
|
- test get backend by url
|
|
|
- """
|
|
|
from celery import backends
|
|
|
from celery.backends.riak import RiakBackend
|
|
|
backend, url_ = backends.get_backend_by_url(url, self.app.loader)
|
|
@@ -136,19 +98,12 @@ class test_RiakBackend(AppCase):
|
|
|
self.assertEqual(url_, url)
|
|
|
|
|
|
def test_backend_params_by_url(self):
|
|
|
- """
|
|
|
- test get backend params by url
|
|
|
- """
|
|
|
self.app.conf.result_backend = 'riak://myhost:123/mycoolbucket'
|
|
|
self.assertEqual(self.backend.bucket_name, 'mycoolbucket')
|
|
|
self.assertEqual(self.backend.host, 'myhost')
|
|
|
self.assertEqual(self.backend.port, 123)
|
|
|
|
|
|
def test_non_ASCII_bucket_raises(self):
|
|
|
- """test app.conf.riak_backend_settings and
|
|
|
- app.conf.riak_backend_settings
|
|
|
- is properly set
|
|
|
- """
|
|
|
self.app.conf.riak_backend_settings = {
|
|
|
'bucket': 'héhé',
|
|
|
'host': 'there.host.com',
|