class MemcachedTestCase(unittest.TestCase):
def setUp(self):
self.mc = Client()
LOCK.identifier = "SINGLE_BEAT_MEMCACHED_SERVER"
LOCK.mc = self.mc
def test_acquire_lock(self):
LOCK.acquire_lock()
value = "%s:%s" % (HOST_IDENTIFIER, '0')
self.assertEqual(self.mc.get(LOCK.lock_key), value)
def test_refreshlock(self):
LOCK.refresh_lock(777)
value = "%s:%s" % (HOST_IDENTIFIER, 777)
self.assertEqual(self.mc.get(LOCK.lock_key), value)
def invalidate_cached_object(instance, is_test=False):
"""
Removes cached objects for all users based on the class name of the
instance passed.
Ex: Given a DataSet instance, all possible cached objects holding
DataSets will be deleted to represent the saving, updating,
deletion, or perms change that was performed upon it.
If the is_test flag is set, a new instance of a mockcache Client
will be returned
"""
if not is_test:
try:
cache.delete_many([
'{}-{}'.format(user.id, instance.__class__.__name__)
for user in User.objects.all()
])
except Exception as e:
logger.debug(
"Could not delete %s from cache" % instance.__class__.__name__,
e)
else:
from mockcache import Client
mc = Client()
return mc
class MemcachedTestCase(unittest.TestCase): def setUp(self): self.mc = Client() LOCK.identifier = "SINGLE_BEAT_MEMCACHED_SERVER" LOCK.mc = self.mc def test_acquire_lock(self): LOCK.acquire_lock() value = "%s:%s" % (HOST_IDENTIFIER, '0') self.assertEqual(self.mc.get(LOCK.lock_key),value) def test_refreshlock(self): LOCK.refresh_lock(777) value = "%s:%s" % (HOST_IDENTIFIER, 777) self.assertEqual(self.mc.get(LOCK.lock_key),value)
def setUp(self):
config.tf_server = tf.train.Server.create_local_server()
config.mc = Client()
self.model = Model()
self.model.data = {
"x": [0, 0.1, 0.2, 0.3],
"x2": [5, -2, 3, 10],
"output_y": [1, 2, 3, 4]
}
self.model.norms = {"x": [0, 0.3], "x2": [-2, 10], "output_y": [1, 4]}
self.model.types = {"x": "float", "x2": "float"}
self.model.model_path = "/tmp/mymodel"
self.model2 = Model()
self.model2.data = {
"x0": [0, 0.1, 0.2, 0.3],
"class": ["a", "a", "c", "k"],
"output_y": [1, 2, 3, 4]
}
self.model2.norms = {"x0": [0, 0.3], "class": [], "output_y": [1, 4]}
self.model2.types = {"x0": "float", "class": "str"}
self.model2.model_path = "/tmp/mymodel2"
def setUp(self):
self.mc = Client()
LOCK.identifier = "SINGLE_BEAT_MEMCACHED_SERVER"
LOCK.mc = self.mc
def setUp(self): self.mc = Client() LOCK.identifier = "SINGLE_BEAT_MEMCACHED_SERVER" LOCK.mc = self.mc
def setUp(self):
config.mc = Client()