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'))
