def test_book_inf_authors(self): "Tests getting book authors" Author(name='Name').save() xml_string = ''' <book> <title>Book title </title> <authors> <author> <name>Name2</name> <alias>Alias</alias> </author> <author> <name>Name</name> <alias>Alias</alias> </author> <author> <name> </name> </author> </authors> <files> <file> <link>link</link> </file> </files> </book> '''; xml = etree.fromstring(xml_string) (book, authors, book_files, annotations) = get_book_inf(xml) self.failUnlessEqual(authors[0], Author.objects.all()[1]) self.failUnlessEqual(authors[1], Author.objects.all()[0]) self.failUnlessEqual(Author.objects.all().count(), 2)
def test_book_inf(self): "Tests getting book inf" xml_string = ''' <book> <title>Book title </title> <lang>en</lang> <files> <file> <link>link</link> </file> </files> </book> '''; xml = etree.fromstring(xml_string) (book, authors, book_files, annotations) = get_book_inf(xml) self.failUnlessEqual(book.title, 'Book title') self.failUnlessEqual(book.language, self.unknown_language)
def test_book_inf_files(self): "Tests getting book_files" link_hash = md5('link').hexdigest() BookFile(link='link', link_hash=link_hash).save() xml_string = ''' <book> <title>Book title </title> <files> <file> <link>link </link> <size>wrong</size> </file> <file> <link>link2</link> <size>645</size> </file> </files> </book> '''; xml = etree.fromstring(xml_string) (book, authors, book_files, annotations) = get_book_inf(xml) self.failUnlessEqual(book_files[0], BookFile.objects.all()[0]) self.failUnlessEqual(book_files[1], BookFile.objects.all()[1]) self.failUnlessEqual(BookFile.objects.all().count(), 2)