fixes for edge cases

[librarian.git] / tests / test_dcparser.py

diff --git a/tests/test_dcparser.py b/tests/test_dcparser.py
index fcbc363..64e98d7 100644 (file)
--- a/tests/test_dcparser.py
+++ b/tests/test_dcparser.py
@@ -1,16 +1,23 @@
 # -*- coding: utf-8 -*-
 # -*- coding: utf-8 -*-

+#
+# This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from __future__ import unicode_literals
+

 from librarian import dcparser
 from lxml import etree
 from nose.tools import *
 from os.path import splitext
 from tests.utils import get_all_fixtures
 import codecs
 from librarian import dcparser
 from lxml import etree
 from nose.tools import *
 from os.path import splitext
 from tests.utils import get_all_fixtures
 import codecs

+from datetime import date

 
 
 def check_dcparser(xml_file, result_file):
 
 
 def check_dcparser(xml_file, result_file):

-    xml = file(xml_file).read()
+    xml = open(xml_file, 'rb').read()

     result = codecs.open(result_file, encoding='utf-8').read()
     result = codecs.open(result_file, encoding='utf-8').read()

-    info = dcparser.BookInfo.from_string(xml).to_dict()
+    info = dcparser.BookInfo.from_bytes(xml).to_dict()

     should_be = eval(result)
     for key in should_be:
         assert_equals(info[key], should_be[key])
     should_be = eval(result)
     for key in should_be:
         assert_equals(info[key], should_be[key])


@@ -23,13 +30,13 @@ def test_dcparser():
 
 
 def check_serialize(xml_file):
 
 
 def check_serialize(xml_file):

-    xml = file(xml_file).read()
-    info = dcparser.BookInfo.from_string(xml)
+    xml = open(xml_file, 'rb').read()
+    info = dcparser.BookInfo.from_bytes(xml)

 
     # serialize
 
     # serialize

-    serialized = etree.tostring(info.to_etree(), encoding=unicode).encode('utf-8')
+    serialized = etree.tostring(info.to_etree(), encoding='unicode').encode('utf-8')

     # then parse again
     # then parse again

-    info_bis = dcparser.BookInfo.from_string(serialized)
+    info_bis = dcparser.BookInfo.from_bytes(serialized)

 
     # check if they are the same
     for key in vars(info):
 
     # check if they are the same
     for key in vars(info):


@@ -41,4 +48,3 @@ def check_serialize(xml_file):
 def test_serialize():
     for fixture in get_all_fixtures('dcparser', '*.xml'):
         yield check_serialize, fixture
 def test_serialize():
     for fixture in get_all_fixtures('dcparser', '*.xml'):
         yield check_serialize, fixture

-