def test_create_sort_param_03(self): """sort direction input handled properly""" for sort_type, primary in config.SEARCH_SORT_OPTIONS_MAP.items(): sort = create_sort_param(sort_type=sort_type, sort_dir='desc') self.assertEqual(sort, [(primary[0], 'desc'), config.SEARCH_DEFAULT_SECONDARY_SORT]) sort = create_sort_param(sort_type=sort_type, sort_dir='asc') self.assertEqual(sort, [(primary[0], 'asc'), (config.SEARCH_DEFAULT_SECONDARY_SORT[0], 'asc')])
def test_create_sort_param_01(self): """no options should produce default values""" sort = create_sort_param() default_type = config.SEARCH_DEFAULT_SORT primary = config.SEARCH_SORT_OPTIONS_MAP[default_type] secondary = config.SEARCH_DEFAULT_SECONDARY_SORT self.assertEqual(sort, [primary,secondary])
def test_create_sort_param_04(self): """list type that doesn't appear in ABS_SORT_OPTIONS_MAP should get defaults""" sort = create_sort_param(list_type='foo') default_type = config.SEARCH_DEFAULT_SORT primary = config.SEARCH_SORT_OPTIONS_MAP[default_type] secondary = config.SEARCH_DEFAULT_SECONDARY_SORT self.assertEqual(sort, [primary,secondary])
view that retrieves the only bibcodes for a list of paper """ #list of bibcodes to extract bibcodes_to_export = [] list_type = request.values.get('list_type') numRecs = request.values.get('numRecs', str(config.EXPORT_DEFAULT_ROWS)) try: query_components = json.loads(request.values.get('current_search_parameters')) except (TypeError, JSONDecodeError), e: #@todo: logging of the error return '' #update the query parameters to return only what is necessary query_components.update({'facets':[], 'fields': ['bibcode'], 'highlights':[], 'rows': numRecs}) if 'sort' not in query_components: query_components['sort'] = create_sort_param(list_type=list_type) #execute the query if list_type == 'similar': resp = get_document_similar(**query_components) else: req = solr.create_request(**query_components) url = None if 'bigquery' in request.values: from adsabs.core.solr import bigquery bigquery.prepare_bigquery_request(req, request.values['bigquery']) url = config.SOLRBIGQUERY_URL req = solr.set_defaults(req, query_url=url) resp = solr.get_response(req) if resp.is_error():
def test_create_sort_param_02(self): """sort type input should produce correct values from mapping""" for sort_type, primary in config.SEARCH_SORT_OPTIONS_MAP.items(): sort = create_sort_param(sort_type=sort_type) self.assertEqual(sort, [primary, config.SEARCH_DEFAULT_SECONDARY_SORT])
def test_create_sort_param_06(self): """list type shouldn't override form input""" for list_type, lt_primary in config.ABS_SORT_OPTIONS_MAP.items(): for sort_type, primary in config.SEARCH_SORT_OPTIONS_MAP.items(): sort = create_sort_param(sort_type=sort_type, list_type=list_type) self.assertEqual(sort, [primary, config.SEARCH_DEFAULT_SECONDARY_SORT])
def test_create_sort_param_05(self): """list type in ABS_SORT_OPTIONS_MAP should get produce correct value""" for list_type, primary in config.ABS_SORT_OPTIONS_MAP.items(): sort = create_sort_param(list_type=list_type) self.assertEqual(sort, [primary, config.SEARCH_DEFAULT_SECONDARY_SORT])