From 73353bc70b4c54c62093a26e1b7a30a08b1162d5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Joonas=20Kes=C3=A4niemi?= Date: Fri, 1 Jun 2018 12:35:38 +0300 Subject: [PATCH] Update minimal_api.py --- src/metax_api/tests/api/oaipmh/minimal_api.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/metax_api/tests/api/oaipmh/minimal_api.py b/src/metax_api/tests/api/oaipmh/minimal_api.py index 2f0a4209..a4d72ba2 100644 --- a/src/metax_api/tests/api/oaipmh/minimal_api.py +++ b/src/metax_api/tests/api/oaipmh/minimal_api.py @@ -110,16 +110,18 @@ def test_list_sets(self): self.assertEquals(len(sets), 4) def test_list_identifiers(self): + ms = settings.OAI['BATCH_SIZE'] allRecords = CatalogRecord.objects.filter( - data_catalog__catalog_json__identifier__in=MetaxOAIServer._get_default_set_filter(None))[:settings.OAI['BATCH_SIZE']] + data_catalog__catalog_json__identifier__in=MetaxOAIServer._get_default_set_filter(None))[:ms] response = self.client.get('/oai/?verb=ListIdentifiers&metadataPrefix=oai_dc') self.assertEqual(response.status_code, status.HTTP_200_OK) headers = self._get_results(response.content, '//o:header') self.assertTrue(len(headers) == len(allRecords), len(headers)) def test_list_records(self): + ms = settings.OAI['BATCH_SIZE'] allRecords = CatalogRecord.objects.filter( - data_catalog__catalog_json__identifier__in=MetaxOAIServer._get_default_set_filter(None))[:settings.OAI['BATCH_SIZE']] + data_catalog__catalog_json__identifier__in=MetaxOAIServer._get_default_set_filter(None))[:ms] response = self.client.get('/oai/?verb=ListRecords&metadataPrefix=oai_dc') self.assertEqual(response.status_code, status.HTTP_200_OK) @@ -183,8 +185,9 @@ def test_get_urnresolver_record(self): self.assertTrue(len(identifiers) == 1, response.content) def test_list_records_from_datasets_set(self): + ms = settings.OAI['BATCH_SIZE'] allRecords = CatalogRecord.objects.filter( - data_catalog__catalog_json__identifier__in=MetaxOAIServer._get_default_set_filter(None))[:settings.OAI['BATCH_SIZE']] + data_catalog__catalog_json__identifier__in=MetaxOAIServer._get_default_set_filter(None))[:ms] response = self.client.get('/oai/?verb=ListRecords&metadataPrefix=oai_dc&set=datasets') self.assertEqual(response.status_code, status.HTTP_200_OK)