class TemplateTagsTestCase(TestCase): def setUp(self): self.m1 = Machine(url='http://dspace.wrlc.org', name='DSpace Server') self.m1.save() self.m2 = Machine(url='dspace.wrlc.org', name='Backup DSpace Server') self.m2.save() self.collection = Collection(id='testcoll', name='TestCollection') self.collection.save() self.project = Project(id='testproj', name='TestProject', collection=self.collection) self.project.save() self.item = Item(id='testItem', title='TestItem', collection=self.collection, project=self.project) self.item.save() self.bag1 = Bag(bagname='testBag1', item=self.item, machine=self.m1, bag_type='1', absolute_filesystem_path='/tmp/fakebag') self.bag1.payload = """/data/METADATA/0123456789-dc.xml 2655 /data/METADATA/0123456789-MRC.xml 3256 /data/IMAGES/0123456789_pg1.jp2 1778740 /data/IMAGES/0123456789_pg2.jp2 1878756 /data/IMAGES/0123456789_pg3.jp2 1915879 /data/IMAGES/0123456789_pg1.tiff 1778740 /data/IMAGES/0123456789_pg2.tiff 1878756 /data/IMAGES/0123456789_pg3.tiff 1915879 """ self.bag1.save() self.bag2 = Bag(bagname='testBag2', item=self.item, machine=self.m2, bag_type='1', absolute_filesystem_path='/tmp/fakebag') self.bag2.payload = """/data/METADATA/0123456789-dc.xml 2655 /data/METADATA/0123456789-MRC.xml 3256 /data/IMAGES/0123456789_pg1.jp2 1778740 /data/IMAGES/0123456789_pg2.jp2 1878756 /data/IMAGES/0123456789_pg3.jp2 1915879 /data/IMAGES/0123456789_pg1.tiff 1778740 /data/IMAGES/0123456789_pg2.tiff 1878756 /data/IMAGES/0123456789_pg3.tiff 1915879 """ self.bag2.save() def test_urlize_with_label_templatetag(self): bag1_files = self.bag1.list_payload() bag1_url = invapp_extras.urlize_with_label( urlize(self.bag1.access_url() + bag1_files[0][0]), bag1_files[0][0]) self.assertEqual(bag1_url, '<a href="http://dspace.wrlc.org/tmp/' 'fakebag/data/METADATA/0123456789-dc.xml">' '/data/METADATA/0123456789-dc.xml</a>') bag2_files = self.bag2.list_payload() bag2_url = invapp_extras.urlize_with_label( urlize(self.bag2.access_url() + bag2_files[0][0]), bag2_files[0][0]) self.assertEqual(bag2_url, '/data/METADATA/0123456789-dc.xml')
def test_pagination(self): self.maxDiff = None c1 = Collection(id='collection_1_1_1', name='test-collection-1', created=now()) c1.save() p1 = Project(id='project_1_1_1', name='test-project-1', collection=c1, created=now()) p1.save() i1 = Item(id='item_1_1_1', title='test-item-1', project=p1, collection=c1, created=now(), original_item_type='1') i1.save() m1 = Machine(name='test-machine-1', url='test-url-1') m1.save() b1 = Bag(bagname='test-bag-1', created=now(), item=i1, machine=m1, absolute_filesystem_path='test-path1', bag_type='1') b2 = Bag(bagname='test-bag-4', created=now(), item=i1, machine=m1, absolute_filesystem_path='test-path4', bag_type='1') b3 = Bag(bagname='test-bag-5', created=now(), item=i1, machine=m1, absolute_filesystem_path='test-path5', bag_type='1') b1.payload = '' b2.payload = '' b3.payload = '' for i in range(150): b1.payload += '/data/IMAGES/' + str(i) + '.jp2 ' + \ str(random.randrange(10000, 99999)) + '\n' if i < 70: b2.payload += '/data/IMAGES/' + str(i) + '.jp2' + \ str(random.randrange(10000, 99999)) + '\n' if i < 20: b3.payload += '/data/IMAGES/' + str(i) + '.jp2' + \ str(random.randrange(10000, 99999)) + '\n' b1_files = b1.list_payload() b2_files = b2.list_payload() b3_files = b3.list_payload() b1_paginator = Paginator(b1_files, 10) b2_paginator = Paginator(b2_files, 10) b3_paginator = Paginator(b3_files, 10) b1_files = b1_paginator.page(1) b2_files = b2_paginator.page(1) b3_files = b3_paginator.page(1) expected_b1 = list(range(13)) expected_b1[0] = {'disp': '<<', 'link': None, 'disabled': True} expected_b1[1] = {'disp': '1', 'link': '?files_page=1', 'disabled': True} expected_b1[2] = {'disp': '2', 'link': '?files_page=2', 'disabled': False} expected_b1[3] = {'disp': '3', 'link': '?files_page=3', 'disabled': False} expected_b1[4] = {'disp': '4', 'link': '?files_page=4', 'disabled': False} expected_b1[5] = {'disp': '5', 'link': '?files_page=5', 'disabled': False} expected_b1[6] = {'disp': '6', 'link': '?files_page=6', 'disabled': False} expected_b1[7] = {'disp': '7', 'link': '?files_page=7', 'disabled': False} expected_b1[8] = {'disp': '8', 'link': '?files_page=8', 'disabled': False} expected_b1[9] = {'disp': '9', 'link': '?files_page=9', 'disabled': False} expected_b1[10] = {'disp': '...', 'link': None, 'disabled': True} expected_b1[11] = {'disp': '15', 'link': '?files_page=15', 'disabled': False} expected_b1[12] = {'disp': '>>', 'link': '?files_page=2', 'disabled': False} expected_b2 = list(range(9)) expected_b2[0] = {'disp': '<<', 'link': None, 'disabled': True} expected_b2[1] = {'disp': '1', 'link': '?files_page=1', 'disabled': True} expected_b2[2] = {'disp': '2', 'link': '?files_page=2', 'disabled': False} expected_b2[3] = {'disp': '3', 'link': '?files_page=3', 'disabled': False} expected_b2[4] = {'disp': '4', 'link': '?files_page=4', 'disabled': False} expected_b2[5] = {'disp': '5', 'link': '?files_page=5', 'disabled': False} expected_b2[6] = {'disp': '6', 'link': '?files_page=6', 'disabled': False} expected_b2[7] = {'disp': '7', 'link': '?files_page=7', 'disabled': False} expected_b2[8] = {'disp': '>>', 'link': '?files_page=2', 'disabled': False} expected_b3 = list(range(4)) expected_b3[0] = {'disp': '<<', 'link': None, 'disabled': True} expected_b3[1] = {'disp': '1', 'link': '?files_page=1', 'disabled': True} expected_b3[2] = {'disp': '2', 'link': '?files_page=2', 'disabled': False} expected_b3[3] = {'disp': '>>', 'link': '?files_page=2', 'disabled': False} context = RequestContext(HttpRequest()) # Test for bag with more than 100 files self.assertEqual(expected_b1, invapp_extras.pagination_boxes(context, b1_files, 'files_page')) # Test for bag with less than 100 files self.assertEqual(expected_b2, invapp_extras.pagination_boxes(context, b2_files, 'files_page')) # Test for bag with less than 20 files self.assertEqual(expected_b3, invapp_extras.pagination_boxes(context, b3_files, 'files_page'))