def setUp(self):
self.set_up()
self.handle_config = get_handle_config(tenant_id)
self.create = 'CREATE NAMESPACE ' + namespace
self.drop = 'DROP NAMESPACE ' + namespace + ' CASCADE'
self.sys_request = SystemRequest().set_timeout(timeout)
self.sys_status = SystemStatusRequest().set_timeout(timeout)
def testNoSQLHandleConfigClone(self):
max_content_length = 1024 * 1024
config = get_handle_config(tenant_id)
clone_config = config.clone()
self.assertEqual(clone_config.get_endpoint(), endpoint)
self.assertEqual(clone_config.get_default_timeout(), timeout)
self.assertEqual(clone_config.get_default_table_request_timeout(),
table_request_timeout)
self.assertEqual(clone_config.get_default_consistency(), consistency)
self.assertEqual(clone_config.get_protocol(), config.get_protocol())
self.assertEqual(clone_config.get_host(), config.get_host())
self.assertEqual(clone_config.get_port(), config.get_port())
self.assertEqual(clone_config.get_timeout(), timeout)
self.assertEqual(clone_config.get_table_request_timeout(),
table_request_timeout)
self.assertEqual(clone_config.get_sec_info_timeout(), sec_info_timeout)
self.assertEqual(clone_config.get_consistency(), consistency)
self.assertEqual(clone_config.get_pool_connections(), pool_connections)
self.assertEqual(clone_config.get_pool_maxsize(), pool_maxsize)
self.assertEqual(clone_config.get_max_content_length(),
max_content_length)
self.assertEqual(clone_config.get_retry_handler().get_num_retries(),
retry_handler.get_num_retries())
self.assertIsNotNone(clone_config.get_authorization_provider())
self.assertEqual(clone_config.get_proxy_host(), proxy_host)
self.assertEqual(clone_config.get_proxy_port(), proxy_port)
self.assertEqual(clone_config.get_proxy_username(), proxy_username)
self.assertEqual(clone_config.get_proxy_password(), proxy_password)
self.assertIsNotNone(clone_config.get_logger())
def setUp(self):
self.set_up()
self.handle_config = get_handle_config(tenant_id)
self.show = 'SHOW AS JSON NAMESPACES'
self.create = 'CREATE NAMESPACE ' + namespace
self.drop = 'DROP NAMESPACE ' + namespace + ' CASCADE'
self.sys_request = SystemRequest()
def setUp(self):
TestBase.set_up(self)
self.handle_config = get_handle_config(tenant_id)
shardkeys = [0, 1]
ids = [0, 1, 2, 3, 4, 5]
write_multiple_request = WriteMultipleRequest()
for sk in shardkeys:
for i in ids:
row = {
'fld_sid': sk,
'fld_id': i,
'fld_long': 2147483648,
'fld_float': 3.1414999961853027,
'fld_double': 3.1415,
'fld_bool': True,
'fld_str': '{"name": u1, "phone": null}',
'fld_bin': bytearray(pack('>i', 4)),
'fld_time': datetime.now(),
'fld_num': Decimal(5),
'fld_json': {
'a': '1',
'b': None,
'c': '3'
},
'fld_arr': ['a', 'b', 'c'],
'fld_map': {
'a': '1',
'b': '2',
'c': '3'
},
'fld_rec': {
'fld_id': 1,
'fld_bool': False,
'fld_str': None
}
}
write_multiple_request.add(
PutRequest().set_value(row).set_table_name(table_name),
True)
self.handle.write_multiple(write_multiple_request)
write_multiple_request.clear()
prepare_statement_update = (
'DECLARE $fld_sid INTEGER; $fld_id INTEGER; UPDATE ' + table_name +
' u SET u.fld_long = u.fld_long + 1 WHERE fld_sid = $fld_sid ' +
'AND fld_id = $fld_id')
prepare_request_update = PrepareRequest().set_statement(
prepare_statement_update)
self.prepare_result_update = self.handle.prepare(
prepare_request_update)
prepare_statement_select = (
'DECLARE $fld_long LONG; SELECT fld_sid, fld_id, fld_long FROM ' +
table_name + ' WHERE fld_long = $fld_long')
prepare_request_select = PrepareRequest().set_statement(
prepare_statement_select)
self.prepare_result_select = self.handle.prepare(
prepare_request_select)
self.query_request = QueryRequest().set_timeout(timeout)
self.get_request = GetRequest().set_table_name(table_name)
def setUp(self):
self.set_up()
self.handle_config = get_handle_config(tenant_id)
self.min_time = list()
self.max_time = list()
shardkeys = 2
ids = 6
write_multiple_request = WriteMultipleRequest()
for sk in range(shardkeys):
for i in range(ids):
row = get_row()
if i == 0:
self.min_time.append(row['fld_time'])
elif i == ids - 1:
self.max_time.append(row['fld_time'])
row['fld_sid'] = sk
row['fld_id'] = i
row['fld_bool'] = False if sk == 0 else True
row['fld_str'] = (
'{"name": u' +
str(shardkeys * ids - sk * ids - i - 1).zfill(2) + '}')
row['fld_json']['location']['coordinates'] = ([
23.549 - sk * 0.5 - i, 35.2908 + sk * 0.5 + i
])
write_multiple_request.add(
PutRequest().set_value(row).set_table_name(table_name),
True)
self.handle.write_multiple(write_multiple_request)
write_multiple_request.clear()
prepare_statement_update = (
'DECLARE $fld_sid INTEGER; $fld_id INTEGER; UPDATE ' + table_name +
' u SET u.fld_long = u.fld_long + 1 WHERE fld_sid = $fld_sid ' +
'AND fld_id = $fld_id')
prepare_request_update = PrepareRequest().set_statement(
prepare_statement_update)
self.prepare_result_update = self.handle.prepare(
prepare_request_update)
prepare_statement_select = (
'DECLARE $fld_long LONG; SELECT fld_sid, fld_id, fld_long FROM ' +
table_name + ' WHERE fld_long = $fld_long')
prepare_request_select = PrepareRequest().set_statement(
prepare_statement_select)
self.prepare_result_select = self.handle.prepare(
prepare_request_select)
self.query_request = QueryRequest().set_timeout(timeout)
self.get_request = GetRequest().set_table_name(table_name)
def testNoSQLHandleConfigGets(self):
max_content_length = 1024 * 1024
config = get_handle_config(tenant_id)
self.assertEqual(config.get_endpoint(), endpoint)
self.assertEqual(config.get_default_timeout(), timeout)
self.assertEqual(config.get_default_table_request_timeout(),
table_request_timeout)
self.assertEqual(config.get_default_consistency(), consistency)
self.assertEqual(config.get_timeout(), timeout)
self.assertEqual(config.get_table_request_timeout(),
table_request_timeout)
self.assertEqual(config.get_sec_info_timeout(), sec_info_timeout)
self.assertEqual(config.get_consistency(), consistency)
self.assertEqual(config.get_pool_connections(), pool_connections)
self.assertEqual(config.get_pool_maxsize(), pool_maxsize)
self.assertEqual(config.get_max_content_length(), max_content_length)
self.assertEqual(config.get_retry_handler(), retry_handler)
self.assertIsNotNone(config.get_authorization_provider())
self.assertIsNotNone(config.get_logger())
def setUp(self):
self.set_up()
self.handle_config = get_handle_config(tenant_id)
self.create_tb_statement = ('CREATE TABLE ' + table_name +
'(fld_id INTEGER, fld_long LONG, \
fld_float FLOAT, fld_double DOUBLE, fld_bool BOOLEAN, fld_str STRING, \
fld_bin BINARY, fld_time TIMESTAMP(4), fld_num NUMBER, fld_json JSON, \
fld_arr ARRAY(STRING), fld_map MAP(STRING), \
fld_rec RECORD(fld_id LONG, fld_bool BOOLEAN, fld_str STRING), \
PRIMARY KEY(fld_id)) USING TTL 30 DAYS')
self.create_idx_statement = ('CREATE INDEX ' + index_name + ' ON ' +
table_name + '(fld_str, fld_double)')
self.alter_fld_statement = ('ALTER TABLE ' + table_name +
'(DROP fld_num)')
self.alter_ttl_statement = ('ALTER TABLE ' + table_name +
' USING TTL 16 HOURS')
self.drop_idx_statement = ('DROP INDEX ' + index_name + ' ON ' +
table_name)
self.drop_tb_statement = ('DROP TABLE IF EXISTS ' + table_name)
self.table_request = TableRequest()
self.table_limits = TableLimits(100, 100, 1)
def testNoSQLHandleConfigGets(self):
config = get_handle_config(tenant_id)
self._check_config(config, None, retry_handler)
def testNoSQLHandleConfigClone(self):
config = get_handle_config(tenant_id)
clone_config = config.clone()
self._check_config(clone_config, config.get_service_url(),
retry_handler.get_num_retries())
def setUp(self):
handle_config = get_handle_config(tenant_id).set_retry_handler(
DefaultRetryHandler()).set_consistency(Consistency.EVENTUAL)
self.handle = NoSQLHandle(handle_config)
