def tearDownModule():
global testrun_search_bucket, testrun_props_bucket, \
testrun_sibs_bucket, testrun_yz_bucket
c = RiakClient(protocol='http', host=HTTP_HOST, http_port=HTTP_PORT,
pb_port=PB_PORT, credentials=SECURITY_CREDS)
c.bucket(testrun_sibs_bucket).clear_properties()
c.bucket(testrun_props_bucket).clear_properties()
if not SKIP_SEARCH and not RUN_YZ:
b = c.bucket(testrun_search_bucket)
b.clear_properties()
if RUN_YZ:
c.protocol = 'pbc'
yzbucket = c.bucket(testrun_yz_bucket)
yzbucket.set_property('search_index', '_dont_index_')
c.delete_search_index(testrun_yz_bucket)
for keys in yzbucket.stream_keys():
for key in keys:
yzbucket.delete(key)
mrtype = c.bucket_type(testrun_mr_btype)
mrbucket = mrtype.bucket(testrun_mr_bucket)
mrbucket.set_property('search_index', '_dont_index_')
c.delete_search_index(testrun_mr_bucket)
for keys in mrbucket.stream_keys():
for key in keys:
mrbucket.delete(key)
def setUpModule():
global testrun_search_bucket, testrun_props_bucket, \
testrun_sibs_bucket, testrun_yz_bucket
c = RiakClient(protocol='http', host=HTTP_HOST, http_port=HTTP_PORT,
pb_port=PB_PORT)
testrun_props_bucket = 'propsbucket'
testrun_sibs_bucket = 'sibsbucket'
c.bucket(testrun_sibs_bucket).allow_mult = True
if (not SKIP_SEARCH and not RUN_YZ):
testrun_search_bucket = 'searchbucket'
b = c.bucket(testrun_search_bucket)
b.enable_search()
if RUN_YZ:
c.protocol = 'pbc'
testrun_yz_bucket = 'yzbucket'
c.create_search_index(testrun_yz_bucket)
b = c.bucket(testrun_yz_bucket)
index_set = False
while not index_set:
try:
b.set_property('search_index', testrun_yz_bucket)
index_set = True
except RiakError:
pass
def setUpModule():
global testrun_search_bucket, testrun_props_bucket, \
testrun_sibs_bucket, testrun_yz_bucket, testrun_mr_btype, \
testrun_mr_bucket
c = RiakClient(protocol='pbc', host=PB_HOST, http_port=HTTP_PORT,
pb_port=PB_PORT, credentials=SECURITY_CREDS)
testrun_props_bucket = 'propsbucket'
testrun_sibs_bucket = 'sibsbucket'
c.bucket(testrun_sibs_bucket).allow_mult = True
if (not SKIP_SEARCH and not RUN_YZ):
testrun_search_bucket = 'searchbucket'
b = c.bucket(testrun_search_bucket)
b.enable_search()
if RUN_YZ:
c.protocol = 'pbc'
testrun_yz_bucket = 'yzbucket'
c.create_search_index(testrun_yz_bucket)
b = c.bucket(testrun_yz_bucket)
index_set = False
while not index_set:
try:
b.set_property('search_index', testrun_yz_bucket)
index_set = True
except RiakError:
pass
# Add bucket and type for Search -> MapReduce
testrun_mr_btype = 'pytest-mr'
testrun_mr_bucket = 'mrbucket'
c.create_search_index(testrun_mr_bucket, '_yz_default')
t = c.bucket_type(testrun_mr_btype)
b = t.bucket(testrun_mr_bucket)
index_set = False
while not index_set:
try:
b.set_property('search_index', testrun_mr_bucket)
index_set = True
except RiakError:
pass
