diff --git a/comic/tests.py b/comic/tests.py index d0f19e6..3f9d1ea 100644 --- a/comic/tests.py +++ b/comic/tests.py @@ -63,11 +63,11 @@ class ComicBookTests(TestCase): user = User.objects.get(username='test') nav = book.nav(0, user) self.assertEqual(nav.next_index, 1) - self.assertEqual(nav.next_path, urlsafe_base64_encode(book.selector.bytes)) + self.assertEqual(nav.next_path.encode(), urlsafe_base64_encode(book.selector.bytes)) self.assertEqual(nav.prev_index, -1) self.assertEqual(nav.prev_path, '') self.assertEqual(nav.cur_index, 0) - self.assertEqual(nav.cur_path, urlsafe_base64_encode(book.selector.bytes)) + self.assertEqual(nav.cur_path.encode(), urlsafe_base64_encode(book.selector.bytes)) self.assertEqual(nav.q_prev_to_directory, True) self.assertEqual(nav.q_next_to_directory, False) @@ -79,11 +79,11 @@ class ComicBookTests(TestCase): nav = book.nav(0, user) self.assertEqual(nav.next_index, 1) - self.assertEqual(nav.next_path, urlsafe_base64_encode(book.selector.bytes)) + self.assertEqual(nav.next_path.encode(), 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.prev_path.encode(), urlsafe_base64_encode(prev_book.selector.bytes)) self.assertEqual(nav.cur_index, 0) - self.assertEqual(nav.cur_path, urlsafe_base64_encode(book.selector.bytes)) + self.assertEqual(nav.cur_path.encode(), urlsafe_base64_encode(book.selector.bytes)) self.assertEqual(nav.q_prev_to_directory, False) self.assertEqual(nav.q_next_to_directory, False) @@ -95,11 +95,11 @@ class ComicBookTests(TestCase): 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.next_path.encode(), urlsafe_base64_encode(next_book.selector.bytes)) self.assertEqual(nav.prev_index, 2) - self.assertEqual(nav.prev_path, urlsafe_base64_encode(book.selector.bytes)) + self.assertEqual(nav.prev_path.encode(), urlsafe_base64_encode(book.selector.bytes)) self.assertEqual(nav.cur_index, 3) - self.assertEqual(nav.cur_path, urlsafe_base64_encode(book.selector.bytes)) + self.assertEqual(nav.cur_path.encode(), urlsafe_base64_encode(book.selector.bytes)) self.assertEqual(nav.q_prev_to_directory, False) self.assertEqual(nav.q_next_to_directory, False) @@ -110,9 +110,9 @@ class ComicBookTests(TestCase): self.assertEqual(nav.next_index, -1) self.assertEqual(nav.next_path, '') self.assertEqual(nav.prev_index, 2) - self.assertEqual(nav.prev_path, urlsafe_base64_encode(book.selector.bytes)) + self.assertEqual(nav.prev_path.encode(), urlsafe_base64_encode(book.selector.bytes)) self.assertEqual(nav.cur_index, 3) - self.assertEqual(nav.cur_path, urlsafe_base64_encode(book.selector.bytes)) + self.assertEqual(nav.cur_path.encode(), urlsafe_base64_encode(book.selector.bytes)) self.assertEqual(nav.q_prev_to_directory, False) self.assertEqual(nav.q_next_to_directory, True) @@ -122,11 +122,11 @@ class ComicBookTests(TestCase): directory__isnull=True) nav = book.nav(1, user) self.assertEqual(nav.next_index, 2) - self.assertEqual(nav.next_path, urlsafe_base64_encode(book.selector.bytes)) + self.assertEqual(nav.next_path.encode(), urlsafe_base64_encode(book.selector.bytes)) self.assertEqual(nav.prev_index, 0) - self.assertEqual(nav.prev_path, urlsafe_base64_encode(book.selector.bytes)) + self.assertEqual(nav.prev_path.encode(), urlsafe_base64_encode(book.selector.bytes)) self.assertEqual(nav.cur_index, 1) - self.assertEqual(nav.cur_path, urlsafe_base64_encode(book.selector.bytes)) + self.assertEqual(nav.cur_path.encode(), urlsafe_base64_encode(book.selector.bytes)) self.assertEqual(nav.q_prev_to_directory, False) self.assertEqual(nav.q_next_to_directory, False) @@ -135,54 +135,46 @@ class ComicBookTests(TestCase): folders = generate_directory(user) dir1 = folders[0] self.assertEqual(dir1.name, 'test_folder') - self.assertTrue(dir1.isdir) + self.assertEqual(dir1.type, 'directory') self.assertEqual(dir1.icon, 'glyphicon-folder-open') - self.assertFalse(dir1.iscb) 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, '