diff --git a/metrics_service/tests/test_endpoints.py b/metrics_service/tests/test_endpoints.py index 2c6212c..df308c2 100644 --- a/metrics_service/tests/test_endpoints.py +++ b/metrics_service/tests/test_endpoints.py @@ -538,7 +538,7 @@ def test_get_everything_single_bibcode_no_types(self, mock_execute_SQL_query): data=json.dumps({'bibcodes': testset[:1], 'types':[]})) self.assertTrue(r.status_code == 200) # Check that the right info is returned; nothing more, nothing less - expected = ['skipped bibcodes', 'basic stats', 'basic stats refereed', 'citation stats', 'citation stats refereed', 'histograms'] + expected = ['skipped bibcodes', 'basic stats', 'basic stats refereed', 'citation stats', 'citation stats refereed', 'histograms', 'indicators', 'indicators refereed'] self.assertEqual(sorted(list(r.json.keys())), sorted(expected)) @mock.patch('metrics_service.models.execute_SQL_query', return_value=testdata) diff --git a/metrics_service/views.py b/metrics_service/views.py index 8835594..79472cc 100644 --- a/metrics_service/views.py +++ b/metrics_service/views.py @@ -65,7 +65,7 @@ def post(self): elif len(bibcodes) == 1: current_app.logger.debug('Metrics requested for single record') if len(types) > 0: - types = [t for t in types if t in ['basic', 'citations', 'histograms']] + types = [t for t in types if t in ['basic', 'citations', 'histograms', 'indicators']] # if len(types) == 0: # types=['basic', 'citations', 'histograms'] if len(histograms) > 0: