def test_xattr_compression(self):
# MB-32669
# subdoc.subdoc_simple_dataset.SubdocSimpleDataset.test_xattr_compression,compression=active
mc = MemcachedClient(self.master.ip, 11210)
mc.sasl_auth_plain(self.master.rest_username,
self.master.rest_password)
mc.bucket_select('default')
self.key = "test_xattr_compression"
self.nesting_level = 5
array = {'i_add': 0, 'i_sub': 1, 'a_i_a': [0, 1], 'ai_sub': [0, 1]}
base_json = self.generate_json_for_nesting()
nested_json = self.generate_nested(base_json, array,
self.nesting_level)
jsonDump = json.dumps(nested_json)
stats = mc.stats()
self.assertEqual(stats['ep_compression_mode'], 'active')
scheme = "http"
host = "{0}:{1}".format(self.master.ip, self.master.port)
self.sdk_client = SDKClient(scheme=scheme,
hosts=[host],
bucket="default")
self.sdk_client.set(self.key, value=jsonDump, ttl=60)
rv = self.sdk_client.cb.mutate_in(self.key,
SD.upsert('my.attr',
"value",
xattr=True,
create_parents=True),
ttl=60)
self.assertTrue(rv.success)
# wait for it to persist and then evict the key
persisted = 0
while persisted == 0:
opaque, rep_time, persist_time, persisted, cas = mc.observe(
self.key)
mc.evict_key(self.key)
time.sleep(65)
try:
self.client.get(self.key)
self.fail("the key should get expired")
except mc_bin_client.MemcachedError as error:
self.assertEqual(error.status, 1)
stats = mc.stats()
self.assertEqual(int(stats['curr_items']), 0)
self.assertEqual(int(stats['curr_temp_items']), 0)
class MemcachedHelper(object):
def __init__(self,
serverip="localhost",
port=11211,
bucket="default",
password=""):
self.client = MemcachedClient(serverip, port)
self.client.sasl_auth_plain(bucket, password)
def write_one_json(self, key, doc):
count = 0
loaded = False
while count < 60 and not loaded:
try:
self.client.set(key, 0, 0, json.dumps(doc))
loaded = True
except MemcachedError as error:
if error.status == 134:
print "Memcached TMP_OOM, Retrying in 5 seconds..."
count += 1
time.sleep(5)
elif error.status == 7:
print "Not my vbucket error. If on MAC, please specify vbuckets as 64."
print "If rebalance is in progress. Please wait for it to finish.\n"
break
else:
print error
break
def read_one_json(self, key):
doc = ""
try:
_, _, doc = self.client.get(key)
except MemcachedError as error:
print error
return doc
def write_batch(self, batch_id, data):
print "Batch Insert not implemented for Memcached"
class MemcachedHelper(object):
def __init__(self, serverip = "localhost", port = 11211, bucket = "default", password = ""):
self.client = MemcachedClient(serverip, port)
self.client.sasl_auth_plain(bucket, password)
def write_one_json(self, key, doc):
count = 0
loaded = False
while count < 60 and not loaded:
try:
#self.client.set(key, 0, 0, json.dumps(doc))
self.client.set(key, 0, 0, doc)
loaded = True
except MemcachedError as error:
if error.status == 134:
print "Memcached TMP_OOM, Retrying in 5 seconds..."
count += 1
time.sleep(5)
elif error.status == 7:
print "Not my vbucket error. If on MAC, please specify vbuckets as 64."
print "If rebalance is in progress. Please wait for it to finish.\n"
break
else:
print error
break
def read_one_json(self, key):
doc = ""
try:
_, _, doc = self.client.get(key)
except MemcachedError as error:
print error
return doc
def write_batch(self, batch_id, data):
print "Batch Insert not implemented for Memcached"
