Example #1
0
 def test_set_fail_will_cause_async(self):
     self.mq.reset_mock()
     def async_log(key):
         self.mq.send(key)
     mc = mc_from_config(self.config, async_cleaner = async_log)
     mc.set('test_set_key', 'test_value')
     self.mq.send.assert_called_with('test_set_key')
Example #2
0
class LocalCachedTest(PureMCTest):
    config = {
        'servers': ['127.0.0.1:11211'],
    }
    mc = LocalCached(mc_from_config(config))

    def test_wrapper_is_right(self):
        self.assertTrue(isinstance(self.mc, LocalCached))
Example #3
0
    def setUp(self):
        unittest.TestCase.setUp(self)

        config = {
            'servers': ['127.0.0.1:11211'],
        }
        self.mc = mc_from_config(config)
        self.cache = VersionedLocalCached(self.mc)
Example #4
0
    def setUp(self):
        unittest.TestCase.setUp(self)

        config = {
            'servers': ['127.0.0.1:11211'],
        }
        self.mc = mc_from_config(config)
        self.cache = VersionedLocalCached(self.mc)
Example #5
0
class AdjustMCTest(PureMCTest):
    config = {
        'servers': ['127.0.0.1:11211'],
        'new_servers': ['127.0.0.1:11212'],
    }
    mc = mc_from_config(config)

    def test_wrapper_is_right(self):
        self.assertTrue(isinstance(self.mc.mc, AdjustMC))
Example #6
0
    def test_set_fail_will_cause_async(self):
        self.mq.reset_mock()

        def async_log(key):
            self.mq.send(key)

        mc = mc_from_config(self.config, async_cleaner=async_log)
        mc.set('test_set_key', 'test_value')
        self.mq.send.assert_called_with('test_set_key')
Example #7
0
 def test_set_multi_fail_will_cause_async(self):
     self.mq.reset_mock()
     async_log = lambda key: self.mq.send(key)
     mc = mc_from_config(self.config, async_cleaner=async_log)
     mc.set_multi({'key1': 'value1', 'key2': 'value2'})
     assert self.mq.send.call_count == 2
     calls = self.mq.send.call_args_list
     for call in calls:
         assert call == call('key1') or call == call('key2')
Example #8
0
 def test_set_multi_fail_will_cause_async(self):
     self.mq.reset_mock()
     async_log = lambda key: self.mq.send(key)
     mc = mc_from_config(self.config, async_cleaner = async_log)
     mc.set_multi({'key1':'value1', 'key2':'value2'})
     assert self.mq.send.call_count == 2
     calls = self.mq.send.call_args_list
     for call in calls:
         assert call == call('key1') or call == call('key2')
Example #9
0
 def test_delete_multi_fail_will_cause_async(self):
     self.mq.reset_mock()
     async_log = lambda key: self.mq.send(key)
     mc = mc_from_config(self.config, async_cleaner = async_log)
     mc.delete_multi(['key1', 'key2', 'key3'])
     assert self.mq.send.call_count == 3
     calls = self.mq.send.call_args_list
     for call in calls:
         assert call == call('key1') or call == call('key2') or \
                 call == call('key3')
Example #10
0
class ReplicatedTest(PureMCTest):
    config = {
        'servers': ['127.0.0.1:11211'],
        'new_servers': [],
        'backup_servers': ['127.0.0.1:11213'],
    }
    mc = mc_from_config(config)

    def test_wrapper_is_right(self):
        self.assertTrue(isinstance(self.mc.mc, Replicated))
Example #11
0
 def test_delete_multi_fail_will_cause_async(self):
     self.mq.reset_mock()
     async_log = lambda key: self.mq.send(key)
     mc = mc_from_config(self.config, async_cleaner=async_log)
     mc.delete_multi(['key1', 'key2', 'key3'])
     assert self.mq.send.call_count == 3
     calls = self.mq.send.call_args_list
     for call in calls:
         assert call == call('key1') or call == call('key2') or \
                 call == call('key3')
Example #12
0
MEMCACHED = {
    'servers' : [],
    'disabled' : False,
}

# from corelib.config import MEMCACHED
import sys
import os
sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), "stub"))

from douban.mc import mc_from_config
from douban.mc.wrapper import LocalCached
mc = LocalCached(mc_from_config(MEMCACHED))

from douban.sqlstore import store_from_config
from ORZ.exports import orz_decorate, OrzField, orz_get_multi, start_transaction, OrzForceRollBack

DATABASE = {
    'farms': {
        "luz_farm": {
            "master": "localhost:test_vagrant9010:eye:sauron",
            "tables": ["*"],
            },
    },
    'options': {
        'show_warnings': True,
    }
}

from unittest import TestCase
Example #13
0
MEMCACHED = {
    'servers': [],
    'disabled': False,
}

# from corelib.config import MEMCACHED
import sys
import os
sys.path.insert(
    0, os.path.join(os.path.dirname(os.path.realpath(__file__)), "stub"))

from douban.mc import mc_from_config
from douban.mc.wrapper import LocalCached
mc = LocalCached(mc_from_config(MEMCACHED))

from douban.sqlstore import store_from_config
from ORZ.exports import orz_decorate, OrzField, orz_get_multi, start_transaction, OrzForceRollBack

DATABASE = {
    'farms': {
        "luz_farm": {
            "master": "localhost:test_vagrant9010:eye:sauron",
            "tables": ["*"],
        },
    },
    'options': {
        'show_warnings': True,
    }
}

from unittest import TestCase
Example #14
0
 def test_get_fail_will_never_use_async(self):
     self.mq.reset_mock()
     async_log = lambda key: self.mq.send(key)
     mc = mc_from_config(self.config, async_cleaner=async_log)
     mc.get('test_key')
     assert not self.mq.send.called
Example #15
0
 def test_get_fail_will_never_use_async(self):
     self.mq.reset_mock()
     async_log = lambda key: self.mq.send(key)
     mc = mc_from_config(self.config, async_cleaner = async_log)
     mc.get('test_key')
     assert not self.mq.send.called
Example #16
0
def get_mc():
    return mc_from_config(MEMCACHED, use_cache=False)
Example #17
0
class PureMCTest(unittest.TestCase):
    config = {
        'servers': ['127.0.0.1:11211'],
    }
    mc = mc_from_config(config)

    def get_random_key(self):
        return 'MCTEST:%s:%s' % (time.time(), random.random())

    def test_wrapper_is_right(self):
        self.assertTrue(isinstance(self.mc.mc, cmemcached.Client))

    def test_incr_decr(self):
        key = self.get_random_key()

        self.assertTrue(self.mc.set(key, 10))
        self.mc.incr(key)
        self.assertEqual(11, self.mc.get(key))
        self.mc.decr(key, 10)
        self.assertEqual(1, self.mc.get(key))
        self.mc.delete(key)

    def test_add_replace(self):
        key = self.get_random_key()
        val1 = str(random.random())
        val2 = str(random.random())

        self.assertTrue(self.mc.get(key) is None)
        self.assertTrue(self.mc.add(key, val1))
        self.assertEquals(val1, self.mc.get(key))

        self.assertFalse(self.mc.add(key, val2))
        self.assertEqual(val1, self.mc.get(key))

        self.mc.delete(key)
        self.assertTrue(self.mc.get(key) is None)
        self.assertFalse(self.mc.replace(key, val1))
        self.assertTrue(self.mc.get(key) is None)
        self.mc.set(key, val1)
        self.assertEqual(val1, self.mc.get(key))
        self.assertTrue(self.mc.replace(key, val2))
        self.assertEqual(val2, self.mc.get(key))

        self.mc.delete(key)

    def test_set_delete_append_prepend(self):
        key = self.get_random_key()
        val = str(random.random())

        self.mc.delete(key)
        self.assertTrue(self.mc.get(key) is None)
        self.mc.set(key, val)
        self.assertEqual(val, self.mc.get(key))

        self.mc.append(key, 'a')
        self.assertEqual(val + 'a', self.mc.get(key))

        self.mc.prepend(key, 'a')
        self.assertEqual('a' + val + 'a', self.mc.get(key))

        self.mc.delete(key)
        self.assertTrue(self.mc.get(key) is None)

    def test_multi_set_delete_append_prepend(self):
        key1, key2 = self.get_random_key(), self.get_random_key()
        val1, val2 = str(random.random()), str(random.random())

        self.mc.delete_multi([key1, key2])
        self.assertEqual({}, self.mc.get_multi([key1, key2]))

        data = {key1: val1, key2: val2}
        self.mc.set_multi(data)
        self.assertEqual(data, self.mc.get_multi([key1, key2]))

        self.mc.append_multi([key1, key2], 'a')
        self.assertEqual({
            key1: val1 + 'a',
            key2: val2 + 'a'
        }, self.mc.get_multi([key1, key2]))

        self.mc.prepend_multi([key1, key2], 'a')
        self.assertEqual({
            key1: 'a' + val1 + 'a',
            key2: 'a' + val2 + 'a'
        }, self.mc.get_multi([key1, key2]))

        self.mc.delete_multi([key1, key2])
        self.assertEqual({}, self.mc.get_multi([key1, key2]))
Example #18
0
# -*- coding: utf-8 -*-

from flask import Flask

from config import MEMCACHED, DOUBANDB

from douban.mc import mc_from_config
from douban.beansdb import beansdb_from_config

mc = mc_from_config(MEMCACHED)
db = beansdb_from_config(DOUBANDB, mc=mc)

app = Flask(__name__)
app.debug = True

@app.route('/')
def hello():
    return "hello world"

if __name__ == "__main__":
    app.run(host="0.0.0.0")
Example #19
0
File: store.py Project: banjin/code
def get_mc():
    return mc_from_config(MEMCACHED, use_cache=False)