Make custom PDF task clean up after itself.

[wolnelektury.git] / apps / oai / tests / oaipmhapi.py

diff --git a/apps/oai/tests/oaipmhapi.py b/apps/oai/tests/oaipmhapi.py
index ae5e543..db57bc1 100644 (file)
--- a/apps/oai/tests/oaipmhapi.py
+++ b/apps/oai/tests/oaipmhapi.py
@@ -1,20 +1,36 @@
-
-import sys
-print sys.path
-
+# -*- coding: utf-8 -*-
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
 from catalogue.test_utils import WLTestCase
 from catalogue import models
-from nose.tools import raises
+from oai.handlers import *
+from oaipmh.server import *
 from os import path
-from oai.handlers import Catalogue
+from oaipmh.metadata import MetadataRegistry
+from lxml import etree
+
 
 class BookMetadataTest(WLTestCase):
     def setUp(self):
-        super(BookMetadata, self).setUp()
+        super(BookMetadataTest, self).setUp()
         xml = path.join(path.dirname(__file__), 'files/lubie-kiedy-kobieta.xml')
         self.book = models.Book.from_xml_file(xml)
-        self.catalogue = Catalogue()
+
+        xml = path.join(path.dirname(__file__), 'files/antygona.xml')
+        self.book2 = models.Book.from_xml_file(xml)
+
+        mr = MetadataRegistry()
+        self.catalogue = Catalogue(mr)
+
+        mr.registerWriter('oai_dc', oai_dc_writer)
+        nsmap = {'oai_dc': NS_OAIDC, 'dc': NS_DC, 'xsi': NS_XSI}
+        self.xml = XMLTreeServer(self.catalogue, mr, nsmap)
 
     def test_get_record(self):
-        r = self.catalogue.getRecord(record='lubie-kiedy-kobieta')
-        print r
+        self.xml.getRecord(identifier='lubie-kiedy-kobieta',
+                                 metadataPrefix='oai_dc')
+        self.xml.listRecords(metadataPrefix='oai_dc')
+
+    def test_selecting(self):
+        records, token = self.catalogue.listRecords(**{'set': 'epoch:starozytnosc'})