def _get_handle(self, snippet): snippet["result"]["handle"]["secret"], snippet["result"]["handle"]["guid"] = HiveServerQueryHandle.get_decoded( snippet["result"]["handle"]["secret"], snippet["result"]["handle"]["guid"] ) snippet["result"]["handle"].pop("statement_id") snippet["result"]["handle"].pop("has_more_statements") return HiveServerQueryHandle(**snippet["result"]["handle"])
def _get_handle(self, snippet): snippet['result']['handle']['secret'], snippet['result']['handle']['guid'] = HiveServerQueryHandle.get_decoded(snippet['result']['handle']['secret'], snippet['result']['handle']['guid']) for key in snippet['result']['handle'].keys(): if key not in ('log_context', 'secret', 'has_result_set', 'operation_type', 'modified_row_count', 'guid'): snippet['result']['handle'].pop(key) return HiveServerQueryHandle(**snippet['result']['handle'])
def _get_handle(self, snippet): snippet['result']['handle']['secret'], snippet['result']['handle']['guid'] = HiveServerQueryHandle.get_decoded(snippet['result']['handle']['secret'], snippet['result']['handle']['guid']) for key in snippet['result']['handle'].keys(): if key not in ('log_context', 'secret', 'has_result_set', 'operation_type', 'modified_row_count', 'guid'): snippet['result']['handle'].pop(key) return HiveServerQueryHandle(**snippet['result']['handle'])
def _get_handle(self, snippet): try: snippet['result']['handle']['secret'], snippet['result']['handle'][ 'guid'] = HiveServerQueryHandle.get_decoded( snippet['result']['handle']['secret'], snippet['result']['handle']['guid']) except KeyError, ex: raise Exception('Operation has no valid handle attached')
def _get_handle(self, snippet): snippet['result']['handle']['secret'], snippet['result']['handle'][ 'guid'] = HiveServerQueryHandle.get_decoded( snippet['result']['handle']['secret'], snippet['result']['handle']['guid']) snippet['result']['handle'].pop('statement_id') snippet['result']['handle'].pop('has_more_statements') return HiveServerQueryHandle(**snippet['result']['handle'])
def test_open_session(self): query = Mock( get_query_statement=Mock(return_value=['SELECT 1']), settings=[] ) with patch('beeswax.server.hive_server2_lib.thrift_util.get_client') as get_client: original_secret = b's\xb6\x0ePP\xbdL\x17\xa3\x0f\\\xf7K\xe8Y\x1d' original_guid = b'\xd9\xe0hT\xd6wO\xe1\xa3S\xfb\x04\xca\x93V\x01' get_client.return_value = Mock( OpenSession=Mock( return_value=Mock( status=Mock( statusCode=TStatusCode.SUCCESS_STATUS ), configuration={}, sessionHandle=Mock( sessionId=Mock( secret=original_secret, guid=original_guid ) ), serverProtocolVersion=11 ) ), get_coordinator_host=Mock(return_value='hive-host') ) session_count = Session.objects.filter(owner=self.user, application=self.query_server['server_name']).count() # Send open session session = HiveServerClient(self.query_server, self.user).open_session(self.user) assert_equal( session_count + 1, # +1 as setUp resets the user which deletes cascade the sessions Session.objects.filter(owner=self.user, application=self.query_server['server_name']).count() ) session = Session.objects.get_session(self.user, self.query_server['server_name']) secret, guid = session.get_adjusted_guid_secret() secret, guid = HiveServerQueryHandle.get_decoded(secret, guid) assert_equal( original_secret, secret ) assert_equal( original_guid, guid ) handle = session.get_handle() assert_equal( original_secret, handle.sessionId.secret ) assert_equal( original_guid, handle.sessionId.guid )
def _get_handle(self, snippet): try: snippet['result']['handle']['secret'], snippet['result']['handle']['guid'] = HiveServerQueryHandle.get_decoded(snippet['result']['handle']['secret'], snippet['result']['handle']['guid']) except KeyError: raise Exception('Operation has no valid handle attached') for key in snippet['result']['handle'].keys(): if key not in ('log_context', 'secret', 'has_result_set', 'operation_type', 'modified_row_count', 'guid'): snippet['result']['handle'].pop(key) return HiveServerQueryHandle(**snippet['result']['handle'])
def _get_handle(self, snippet): try: snippet['result']['handle']['secret'], snippet['result']['handle']['guid'] = HiveServerQueryHandle.get_decoded(snippet['result']['handle']['secret'], snippet['result']['handle']['guid']) except KeyError: raise Exception('Operation has no valid handle attached') except binascii.Error: LOG.warn('Handle already base 64 decoded') for key in snippet['result']['handle'].keys(): if key not in ('log_context', 'secret', 'has_result_set', 'operation_type', 'modified_row_count', 'guid'): snippet['result']['handle'].pop(key) return HiveServerQueryHandle(**snippet['result']['handle'])
def _get_handle(self, snippet): snippet['result']['handle']['secret'], snippet['result']['handle'][ 'guid'] = HiveServerQueryHandle.get_decoded( snippet['result']['handle']['secret'], snippet['result']['handle']['guid']) return HiveServerQueryHandle(**snippet['result']['handle'])
def _get_handle(self, snippet): snippet['result']['handle']['secret'], snippet['result']['handle']['guid'] = HiveServerQueryHandle.get_decoded(snippet['result']['handle']['secret'], snippet['result']['handle']['guid']) return HiveServerQueryHandle(**snippet['result']['handle'])
def _get_handle(self, snippet): snippet['result']['handle']['secret'], snippet['result']['handle']['guid'] = HiveServerQueryHandle.get_decoded(snippet['result']['handle']['secret'], snippet['result']['handle']['guid']) snippet['result']['handle'].pop('statement_id') snippet['result']['handle'].pop('has_more_statements') snippet['result']['handle'].pop('statements_count') return HiveServerQueryHandle(**snippet['result']['handle'])
def _get_handle(self, snippet): snippet["result"]["handle"]["secret"], snippet["result"]["handle"]["guid"] = HiveServerQueryHandle.get_decoded( snippet["result"]["handle"]["secret"], snippet["result"]["handle"]["guid"] ) return HiveServerQueryHandle(**snippet["result"]["handle"])