@@ -0,0 +1,308 @@
|
||||
"""
|
||||
Unit tests for TOC overlay functionality.
|
||||
|
||||
Tests the complete workflow of:
|
||||
1. Opening TOC overlay with swipe up gesture
|
||||
2. Selecting a chapter from the TOC
|
||||
3. Closing overlay by tapping outside or swiping down
|
||||
"""
|
||||
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
from dreader import (
|
||||
EbookReader,
|
||||
TouchEvent,
|
||||
GestureType,
|
||||
ActionType,
|
||||
OverlayState
|
||||
)
|
||||
|
||||
|
||||
class TestTOCOverlay(unittest.TestCase):
|
||||
"""Test TOC overlay opening, interaction, and closing"""
|
||||
|
||||
def setUp(self):
|
||||
"""Set up test reader with a book"""
|
||||
self.reader = EbookReader(page_size=(800, 1200))
|
||||
|
||||
# Load a test EPUB
|
||||
test_epub = Path(__file__).parent / 'data' / 'library-epub' / 'alice.epub'
|
||||
if not test_epub.exists():
|
||||
# Try to find any EPUB in test data
|
||||
epub_dir = Path(__file__).parent / 'data' / 'library-epub'
|
||||
epubs = list(epub_dir.glob('*.epub'))
|
||||
if epubs:
|
||||
test_epub = epubs[0]
|
||||
else:
|
||||
self.skipTest("No test EPUB files available")
|
||||
|
||||
success = self.reader.load_epub(str(test_epub))
|
||||
self.assertTrue(success, "Failed to load test EPUB")
|
||||
|
||||
def tearDown(self):
|
||||
"""Clean up"""
|
||||
self.reader.close()
|
||||
|
||||
def test_overlay_manager_initialization(self):
|
||||
"""Test that overlay manager is properly initialized"""
|
||||
self.assertIsNotNone(self.reader.overlay_manager)
|
||||
self.assertEqual(self.reader.overlay_manager.page_size, (800, 1200))
|
||||
self.assertFalse(self.reader.is_overlay_open())
|
||||
self.assertEqual(self.reader.get_overlay_state(), OverlayState.NONE)
|
||||
|
||||
def test_open_toc_overlay_directly(self):
|
||||
"""Test opening TOC overlay using direct API call"""
|
||||
# Initially no overlay
|
||||
self.assertFalse(self.reader.is_overlay_open())
|
||||
|
||||
# Open TOC overlay
|
||||
overlay_image = self.reader.open_toc_overlay()
|
||||
|
||||
# Should return an image
|
||||
self.assertIsNotNone(overlay_image)
|
||||
self.assertEqual(overlay_image.size, (800, 1200))
|
||||
|
||||
# Overlay should be open
|
||||
self.assertTrue(self.reader.is_overlay_open())
|
||||
self.assertEqual(self.reader.get_overlay_state(), OverlayState.TOC)
|
||||
|
||||
def test_close_toc_overlay_directly(self):
|
||||
"""Test closing TOC overlay using direct API call"""
|
||||
# Open overlay first
|
||||
self.reader.open_toc_overlay()
|
||||
self.assertTrue(self.reader.is_overlay_open())
|
||||
|
||||
# Close overlay
|
||||
page_image = self.reader.close_overlay()
|
||||
|
||||
# Should return base page
|
||||
self.assertIsNotNone(page_image)
|
||||
|
||||
# Overlay should be closed
|
||||
self.assertFalse(self.reader.is_overlay_open())
|
||||
self.assertEqual(self.reader.get_overlay_state(), OverlayState.NONE)
|
||||
|
||||
def test_swipe_up_from_bottom_opens_toc(self):
|
||||
"""Test that swipe up from bottom of screen opens TOC overlay"""
|
||||
# Create swipe up event from bottom of screen (y=1100, which is > 80% of 1200)
|
||||
event = TouchEvent(
|
||||
gesture=GestureType.SWIPE_UP,
|
||||
x=400,
|
||||
y=1100
|
||||
)
|
||||
|
||||
# Handle gesture
|
||||
response = self.reader.handle_touch(event)
|
||||
|
||||
# Should open overlay
|
||||
self.assertEqual(response.action, ActionType.OVERLAY_OPENED)
|
||||
self.assertEqual(response.data['overlay_type'], 'toc')
|
||||
self.assertTrue(self.reader.is_overlay_open())
|
||||
|
||||
def test_swipe_up_from_middle_does_not_open_toc(self):
|
||||
"""Test that swipe up from middle of screen does NOT open TOC"""
|
||||
# Create swipe up event from middle of screen (y=600, which is < 80% of 1200)
|
||||
event = TouchEvent(
|
||||
gesture=GestureType.SWIPE_UP,
|
||||
x=400,
|
||||
y=600
|
||||
)
|
||||
|
||||
# Handle gesture
|
||||
response = self.reader.handle_touch(event)
|
||||
|
||||
# Should not open overlay
|
||||
self.assertEqual(response.action, ActionType.NONE)
|
||||
self.assertFalse(self.reader.is_overlay_open())
|
||||
|
||||
def test_swipe_down_closes_overlay(self):
|
||||
"""Test that swipe down closes the overlay"""
|
||||
# Open overlay first
|
||||
self.reader.open_toc_overlay()
|
||||
self.assertTrue(self.reader.is_overlay_open())
|
||||
|
||||
# Create swipe down event
|
||||
event = TouchEvent(
|
||||
gesture=GestureType.SWIPE_DOWN,
|
||||
x=400,
|
||||
y=300
|
||||
)
|
||||
|
||||
# Handle gesture
|
||||
response = self.reader.handle_touch(event)
|
||||
|
||||
# Should close overlay
|
||||
self.assertEqual(response.action, ActionType.OVERLAY_CLOSED)
|
||||
self.assertFalse(self.reader.is_overlay_open())
|
||||
|
||||
def test_tap_outside_overlay_closes_it(self):
|
||||
"""Test that tapping outside the overlay panel closes it"""
|
||||
# Open overlay first
|
||||
self.reader.open_toc_overlay()
|
||||
self.assertTrue(self.reader.is_overlay_open())
|
||||
|
||||
# Tap in the far left (outside the centered panel)
|
||||
# Panel is 60% wide centered, so left edge is at 20%
|
||||
event = TouchEvent(
|
||||
gesture=GestureType.TAP,
|
||||
x=50, # Well outside panel
|
||||
y=600
|
||||
)
|
||||
|
||||
# Handle gesture
|
||||
response = self.reader.handle_touch(event)
|
||||
|
||||
# Should close overlay
|
||||
self.assertEqual(response.action, ActionType.OVERLAY_CLOSED)
|
||||
self.assertFalse(self.reader.is_overlay_open())
|
||||
|
||||
def test_tap_on_chapter_selects_and_closes(self):
|
||||
"""Test that tapping on a chapter navigates to it and closes overlay"""
|
||||
# Open overlay first
|
||||
self.reader.open_toc_overlay()
|
||||
chapters = self.reader.get_chapters()
|
||||
|
||||
if len(chapters) < 2:
|
||||
self.skipTest("Need at least 2 chapters for this test")
|
||||
|
||||
# Calculate tap position for second chapter (index 1 - "Metamorphosis")
|
||||
# Based on actual measurements: chapter 1 link is at screen Y=282, X=200-300
|
||||
tap_x = 250 # Within the link text bounds
|
||||
tap_y = 282 # Chapter 1 "Metamorphosis"
|
||||
|
||||
event = TouchEvent(
|
||||
gesture=GestureType.TAP,
|
||||
x=tap_x,
|
||||
y=tap_y
|
||||
)
|
||||
|
||||
# Handle gesture
|
||||
response = self.reader.handle_touch(event)
|
||||
|
||||
# Should select chapter
|
||||
self.assertEqual(response.action, ActionType.CHAPTER_SELECTED)
|
||||
self.assertIn('chapter_index', response.data)
|
||||
|
||||
# Overlay should be closed
|
||||
self.assertFalse(self.reader.is_overlay_open())
|
||||
|
||||
def test_multiple_overlay_operations(self):
|
||||
"""Test opening and closing overlay multiple times"""
|
||||
# Open and close 3 times
|
||||
for i in range(3):
|
||||
# Open
|
||||
self.reader.open_toc_overlay()
|
||||
self.assertTrue(self.reader.is_overlay_open())
|
||||
|
||||
# Close
|
||||
self.reader.close_overlay()
|
||||
self.assertFalse(self.reader.is_overlay_open())
|
||||
|
||||
def test_overlay_with_page_navigation(self):
|
||||
"""Test that overlay works correctly after navigating pages"""
|
||||
# Navigate to page 2
|
||||
self.reader.next_page()
|
||||
|
||||
# Open overlay
|
||||
overlay_image = self.reader.open_toc_overlay()
|
||||
self.assertIsNotNone(overlay_image)
|
||||
self.assertTrue(self.reader.is_overlay_open())
|
||||
|
||||
# Close overlay
|
||||
self.reader.close_overlay()
|
||||
self.assertFalse(self.reader.is_overlay_open())
|
||||
|
||||
def test_toc_overlay_contains_all_chapters(self):
|
||||
"""Test that TOC overlay includes all book chapters"""
|
||||
chapters = self.reader.get_chapters()
|
||||
|
||||
# Open overlay (this generates the HTML with chapters)
|
||||
overlay_image = self.reader.open_toc_overlay()
|
||||
self.assertIsNotNone(overlay_image)
|
||||
|
||||
# Verify overlay manager has correct chapter count
|
||||
# This is implicit in the rendering - if it renders without error,
|
||||
# all chapters were included
|
||||
self.assertTrue(self.reader.is_overlay_open())
|
||||
|
||||
def test_overlay_state_persistence_ready(self):
|
||||
"""Test that overlay state can be tracked (for future state persistence)"""
|
||||
# This test verifies the state tracking is ready for StateManager integration
|
||||
|
||||
# Start with no overlay
|
||||
self.assertEqual(self.reader.current_overlay_state, OverlayState.NONE)
|
||||
|
||||
# Open TOC
|
||||
self.reader.open_toc_overlay()
|
||||
self.assertEqual(self.reader.current_overlay_state, OverlayState.TOC)
|
||||
|
||||
# Close overlay
|
||||
self.reader.close_overlay()
|
||||
self.assertEqual(self.reader.current_overlay_state, OverlayState.NONE)
|
||||
|
||||
|
||||
class TestOverlayRendering(unittest.TestCase):
|
||||
"""Test overlay rendering and compositing"""
|
||||
|
||||
def setUp(self):
|
||||
"""Set up test reader"""
|
||||
self.reader = EbookReader(page_size=(800, 1200))
|
||||
test_epub = Path(__file__).parent / 'data' / 'library-epub' / 'alice.epub'
|
||||
|
||||
if not test_epub.exists():
|
||||
epub_dir = Path(__file__).parent / 'data' / 'library-epub'
|
||||
epubs = list(epub_dir.glob('*.epub'))
|
||||
if epubs:
|
||||
test_epub = epubs[0]
|
||||
else:
|
||||
self.skipTest("No test EPUB files available")
|
||||
|
||||
self.reader.load_epub(str(test_epub))
|
||||
|
||||
def tearDown(self):
|
||||
"""Clean up"""
|
||||
self.reader.close()
|
||||
|
||||
def test_overlay_image_size(self):
|
||||
"""Test that overlay image matches page size"""
|
||||
overlay_image = self.reader.open_toc_overlay()
|
||||
self.assertEqual(overlay_image.size, (800, 1200))
|
||||
|
||||
def test_overlay_compositing(self):
|
||||
"""Test that overlay is properly composited on base page"""
|
||||
# Get base page
|
||||
base_page = self.reader.get_current_page()
|
||||
|
||||
# Open overlay (creates composited image)
|
||||
overlay_image = self.reader.open_toc_overlay()
|
||||
|
||||
# Composited image should be different from base page
|
||||
self.assertIsNotNone(overlay_image)
|
||||
|
||||
# Images should have same size but different content
|
||||
self.assertEqual(base_page.size, overlay_image.size)
|
||||
|
||||
def test_overlay_html_to_image_conversion(self):
|
||||
"""Test that HTML overlay is correctly converted to image"""
|
||||
from dreader.html_generator import generate_toc_overlay
|
||||
|
||||
# Get chapters
|
||||
chapters = self.reader.get_chapters()
|
||||
chapter_data = [{"index": idx, "title": title} for title, idx in chapters]
|
||||
|
||||
# Generate HTML
|
||||
html = generate_toc_overlay(chapter_data)
|
||||
self.assertIsNotNone(html)
|
||||
self.assertIn("Table of Contents", html)
|
||||
|
||||
# Render HTML to image using overlay manager
|
||||
overlay_manager = self.reader.overlay_manager
|
||||
image = overlay_manager.render_html_to_image(html)
|
||||
|
||||
# Should produce valid image
|
||||
self.assertIsNotNone(image)
|
||||
self.assertEqual(image.size, (800, 1200))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user