def get_sample_odb_orm_result(self, is_list): # type: (bool) -> object cluster = Cluster() cluster.id = test_odb_data.cluster_id cluster.name = 'my.cluster' cluster.odb_type = 'sqlite' cluster.broker_host = 'my.broker.host' cluster.broker_port = 1234 cluster.lb_host = 'my.lb.host' cluster.lb_port = 5678 cluster.lb_agent_port = 9012 es = self.ODBTestModelClass() es.name = test_odb_data.name es.is_active = test_odb_data.is_active es.hosts = test_odb_data.es_hosts es.timeout = test_odb_data.es_timeout es.body_as = test_odb_data.es_body_as es.cluster_id = test_odb_data.cluster_id session = self.session_wrapper._session session.add(cluster) session.add(es) session.commit() session = self.session_wrapper._session result = search_es_list(session, test_odb_data.cluster_id) # type: tuple result = result[0] # type: SearchResults # This is a one-element tuple of ElasticSearch ORM objects result = result.result # type: tuple return result if is_list else result[0]
def get_search_es_list(self, cluster_id, needs_columns=False): """ Returns a list of ElasticSearch connections. """ return query.search_es_list(self._session, cluster_id, needs_columns)