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)