6ef3b096c10cbf1403374ff3b41cc7d4e7b5453c
[wolnelektury.git] / apps / oai / tests / oaipmhapi.py
1
2
3 from catalogue.test_utils import WLTestCase
4 from catalogue import models
5 from nose.tools import raises
6 from oai.handlers import *
7 from oaipmh.server import *
8 from os import path
9 from oaipmh.metadata import MetadataRegistry
10 from lxml import etree
11
12
13 class BookMetadataTest(WLTestCase):
14     def setUp(self):
15         super(BookMetadataTest, self).setUp()
16         xml = path.join(path.dirname(__file__), 'files/lubie-kiedy-kobieta.xml')
17         self.book = models.Book.from_xml_file(xml)
18
19         xml = path.join(path.dirname(__file__), 'files/antygona.xml')
20         self.book2 = models.Book.from_xml_file(xml)
21
22         self.catalogue = Catalogue()
23         mr = MetadataRegistry()
24         mr.registerWriter('oai_dc', oai_dc_writer)
25         nsmap = {'oai_dc': NS_OAIDC, 'dc': NS_DC, 'xsi': NS_XSI}
26         self.xml = XMLTreeServer(self.catalogue, mr, nsmap)
27
28     def test_get_record(self):
29         sch = self.xml.getRecord(identifier='lubie-kiedy-kobieta',
30                                  metadataPrefix='oai_dc')
31         sch = self.xml.listRecords(metadataPrefix='oai_dc')
32
33     def test_selecting(self):
34         records, token = self.catalogue.listRecords(**{'set': 'epoch:starozytnosc'})