ajurna/cbwebreader
1
0
Fork 0
mirror of https://github.com/ajurna/cbwebreader.git synced 2025-12-06 14:17:19 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

misc fixes. also fixed tests.

Browse Source
This commit is contained in:
ajurna@gmail.com
2016-04-05 14:34:41 +01:00
parent dd38e4c3f1
commit 9235380328
6 changed files with 109 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

172
comic/tests.py
View File

@@ -1,25 +1,30 @@
from django.test import TestCase
from django.contrib.auth.models import User
from django.utils.http import urlsafe_base64_encode
from comic.models import ComicBook, ComicPage, Setting, ComicStatus
from os import path
import os
from os import path
from django.contrib.auth.models import User
from django.test import TestCase
from django.utils.http import urlsafe_base64_encode
from comic.models import ComicBook, ComicPage, Setting, ComicStatus, Directory
from comic.util import generate_directory
# Create your tests here.
class ComicBookTests(TestCase):
def setUp(self):
Setting(name='BASE_DIR', value=path.join(os.getcwd(), 'comic', 'test')).save()
user = User(username='test')
user.save()
ComicBook.process_comic_book(os.getcwd(), path.join('comic', 'test', 'test1.rar'), 'test1.rar')
book = ComicBook.process_comic_book(os.getcwd(), path.join('comic', 'test', 'test2.rar'), 'test2.rar')
ComicBook.process_comic_book('test1.rar')
book = ComicBook.process_comic_book('test2.rar')
status = ComicStatus(user=user,
comic=book,
last_read_page=2,
unread=False)
status.save()
ComicBook.process_comic_book(os.getcwd(), path.join('comic', 'test', 'test4.rar'), 'test4.rar')
Setting(name='BASE_DIR', value=os.getcwd()).save()
ComicBook.process_comic_book('test4.rar')
def test_comic_processing(self):
book = ComicBook.objects.get(file_name='test1.rar')
@@ -49,97 +54,95 @@ class ComicBookTests(TestCase):
def test_get_image(self):
book = ComicBook.objects.get(file_name='test1.rar')
comic_path = path.join(os.getcwd(), 'comic', 'test', 'test1.rar')
img, content_type = book.get_image(comic_path, 0)
img, content_type = book.get_image(0)
self.assertEqual(content_type, 'image/jpeg')
self.assertEqual(img.read(), 'img1.jpg')
self.assertEqual(img.read(), b'img1.jpg')
def test_nav_first_page_with_folder_above(self):
book = ComicBook.objects.get(file_name='test1.rar')
comic_path = path.join('comic', 'test', 'test1.rar')
encoded_path = urlsafe_base64_encode(comic_path)
prev_path_encoded = urlsafe_base64_encode(path.join('comic', 'test'))
nav = book.nav(encoded_path, 0)
user = User.objects.get(username='test')
nav = book.nav(0, user)
self.assertEqual(nav.next_index, 1)
self.assertEqual(nav.next_path, encoded_path)
self.assertEqual(nav.next_path, urlsafe_base64_encode(book.selector.bytes))
self.assertEqual(nav.prev_index, -1)
self.assertEqual(nav.prev_path, prev_path_encoded)
self.assertEqual(nav.prev_path, '')
self.assertEqual(nav.cur_index, 0)
self.assertEqual(nav.cur_path, encoded_path)
self.assertEqual(nav.cur_path, urlsafe_base64_encode(book.selector.bytes))
self.assertEqual(nav.q_prev_to_directory, True)
self.assertEqual(nav.q_next_to_directory, False)
def test_nav_first_page_with_comic_above(self):
book = ComicBook.objects.get(file_name='test2.rar')
comic_path = path.join('comic', 'test', 'test2.rar')
encoded_path = urlsafe_base64_encode(comic_path)
prev_path_encoded = urlsafe_base64_encode(path.join('comic', 'test', 'test1.rar'))
nav = book.nav(encoded_path, 0)
prev_book = ComicBook.objects.get(file_name='test1.rar')
book = ComicBook.objects.get(file_name='test2.rar',
directory__isnull=True)
user = User.objects.get(username='test')
nav = book.nav(0, user)
self.assertEqual(nav.next_index, 1)
self.assertEqual(nav.next_path, encoded_path)
self.assertEqual(nav.prev_index, 3)
self.assertEqual(nav.prev_path, prev_path_encoded)
self.assertEqual(nav.next_path, urlsafe_base64_encode(book.selector.bytes))
self.assertEqual(nav.prev_index, 0)
self.assertEqual(nav.prev_path, urlsafe_base64_encode(prev_book.selector.bytes))
self.assertEqual(nav.cur_index, 0)
self.assertEqual(nav.cur_path, encoded_path)
self.assertEqual(nav.cur_path, urlsafe_base64_encode(book.selector.bytes))
self.assertEqual(nav.q_prev_to_directory, False)
self.assertEqual(nav.q_next_to_directory, False)
def test_nav_last_page_with_comic_below(self):
book = ComicBook.objects.get(file_name='test2.rar')
comic_path = path.join('comic', 'test', 'test2.rar')
encoded_path = urlsafe_base64_encode(comic_path)
next_path_encoded = urlsafe_base64_encode(path.join('comic', 'test', 'test3.rar'))
nav = book.nav(encoded_path, 3)
self.assertEqual(nav.next_index, 0)
self.assertEqual(nav.next_path, next_path_encoded)
user = User.objects.get(username='test')
book = ComicBook.objects.get(file_name='test1.rar',
directory__isnull=True)
next_book = ComicBook.objects.get(file_name='test2.rar',
directory__isnull=True)
nav = book.nav(3, user)
self.assertEqual(nav.next_index, 2)
self.assertEqual(nav.next_path, urlsafe_base64_encode(next_book.selector.bytes))
self.assertEqual(nav.prev_index, 2)
self.assertEqual(nav.prev_path, encoded_path)
self.assertEqual(nav.prev_path, urlsafe_base64_encode(book.selector.bytes))
self.assertEqual(nav.cur_index, 3)
self.assertEqual(nav.cur_path, encoded_path)
self.assertEqual(nav.cur_path, urlsafe_base64_encode(book.selector.bytes))
self.assertEqual(nav.q_prev_to_directory, False)
self.assertEqual(nav.q_next_to_directory, False)
def test_nav_last_page_with_nothing_below(self):
user = User.objects.get(username='test')
book = ComicBook.objects.get(file_name='test4.rar')
comic_path = path.join('comic', 'test', 'test4.rar')
encoded_path = urlsafe_base64_encode(comic_path)
next_path_encoded = urlsafe_base64_encode(path.join('comic', 'test'))
nav = book.nav(encoded_path, 3)
nav = book.nav(3, user)
self.assertEqual(nav.next_index, -1)
self.assertEqual(nav.next_path, next_path_encoded)
self.assertEqual(nav.next_path, '')
self.assertEqual(nav.prev_index, 2)
self.assertEqual(nav.prev_path, encoded_path)
self.assertEqual(nav.prev_path, urlsafe_base64_encode(book.selector.bytes))
self.assertEqual(nav.cur_index, 3)
self.assertEqual(nav.cur_path, encoded_path)
self.assertEqual(nav.cur_path, urlsafe_base64_encode(book.selector.bytes))
self.assertEqual(nav.q_prev_to_directory, False)
self.assertEqual(nav.q_next_to_directory, True)
def test_nav_in_comic(self):
book = ComicBook.objects.get(file_name='test1.rar')
comic_path = path.join('comic', 'test', 'test1.rar')
encoded_path = urlsafe_base64_encode(comic_path)
nav = book.nav(encoded_path, 1)
user = User.objects.get(username='test')
book = ComicBook.objects.get(file_name='test1.rar',
directory__isnull=True)
nav = book.nav(1, user)
self.assertEqual(nav.next_index, 2)
self.assertEqual(nav.next_path, encoded_path)
self.assertEqual(nav.next_path, urlsafe_base64_encode(book.selector.bytes))
self.assertEqual(nav.prev_index, 0)
self.assertEqual(nav.prev_path, encoded_path)
self.assertEqual(nav.prev_path, urlsafe_base64_encode(book.selector.bytes))
self.assertEqual(nav.cur_index, 1)
self.assertEqual(nav.cur_path, encoded_path)
self.assertEqual(nav.cur_path, urlsafe_base64_encode(book.selector.bytes))
self.assertEqual(nav.q_prev_to_directory, False)
self.assertEqual(nav.q_next_to_directory, False)
def test_generate_directory(self):
base_dir = Setting.objects.get(name='BASE_DIR').value
user = User.objects.get(username='test')
folders = ComicBook.generate_directory(user, base_dir, path.join('comic', 'test'))
folders = generate_directory(user)
dir1 = folders[0]
self.assertEqual(dir1.name, 'test_folder')
self.assertTrue(dir1.isdir)
self.assertEqual(dir1.icon, 'glyphicon-folder-open')
self.assertFalse(dir1.iscb)
location = urlsafe_base64_encode(path.join('comic', 'test', 'test_folder'))
d = Directory.objects.get(name='test_folder',
parent__isnull=True)
location = '/comic/{0}/'.format(urlsafe_base64_encode(d.selector.bytes).decode())
self.assertEqual(dir1.location, location)
self.assertEqual(dir1.label, '')
self.assertEqual(dir1.label, '<center><span class="label label-default">Empty</span></center>')
self.assertEqual(dir1.cur_page, 0)
dir2 = folders[1]
@@ -147,9 +150,12 @@ class ComicBookTests(TestCase):
self.assertFalse(dir2.isdir)
self.assertEqual(dir2.icon, 'glyphicon-book')
self.assertTrue(dir2.iscb)
location = urlsafe_base64_encode(path.join('comic', 'test', 'test1.rar'))
c = ComicBook.objects.get(file_name='test1.rar',
directory__isnull=True)
location = '/comic/read/{0}/{1}/'.format(urlsafe_base64_encode(c.selector.bytes).decode(),
'0')
self.assertEqual(dir2.location, location)
self.assertEqual(dir2.label, '<span class="label label-default pull-right">Unread</span>')
self.assertEqual(dir2.label, '<center><span class="label label-default">Unread</span></center>')
self.assertEqual(dir2.cur_page, 0)
dir3 = folders[2]
@@ -157,9 +163,12 @@ class ComicBookTests(TestCase):
self.assertFalse(dir3.isdir)
self.assertEqual(dir3.icon, 'glyphicon-book')
self.assertTrue(dir3.iscb)
location = urlsafe_base64_encode(path.join('comic', 'test', 'test2.rar'))
c = ComicBook.objects.get(file_name='test2.rar',
directory__isnull=True)
location = '/comic/read/{0}/{1}/'.format(urlsafe_base64_encode(c.selector.bytes).decode(),
'2')
self.assertEqual(dir3.location, location)
self.assertEqual(dir3.label, '<span class="label label-primary pull-right">3/4</span>')
self.assertEqual(dir3.label, '<center><span class="label label-primary">3/4</span></center>')
self.assertEqual(dir3.cur_page, 2)
dir3 = folders[3]
@@ -167,9 +176,12 @@ class ComicBookTests(TestCase):
self.assertFalse(dir3.isdir)
self.assertEqual(dir3.icon, 'glyphicon-book')
self.assertTrue(dir3.iscb)
location = urlsafe_base64_encode(path.join('comic', 'test', 'test3.rar'))
c = ComicBook.objects.get(file_name='test3.rar',
directory__isnull=True)
location = '/comic/read/{0}/{1}/'.format(urlsafe_base64_encode(c.selector.bytes).decode(),
'0')
self.assertEqual(dir3.location, location)
self.assertEqual(dir3.label, '<span class="label label-danger pull-right">Unprocessed</span>')
self.assertEqual(dir3.label, '<center><span class="label label-default">Unread</span></center>')
self.assertEqual(dir3.cur_page, 0)
def test_pages(self):
@@ -187,41 +199,3 @@ class ComicBookTests(TestCase):
def test_page_name(self):
book = ComicBook.objects.get(file_name='test1.rar')
self.assertEqual(book.page_name(0), 'img1.jpg')