Fixed issue where file extenions not found and auth timeout
This commit is contained in:
@@ -0,0 +1,73 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Test extension case normalization functionality
|
||||
"""
|
||||
|
||||
import pytest
|
||||
from unittest.mock import MagicMock
|
||||
from bdfr.resource import Resource
|
||||
|
||||
|
||||
class TestExtensionNormalization:
|
||||
"""Test that extensions are properly normalized to lowercase"""
|
||||
|
||||
def test_url_extensions_normalized(self):
|
||||
"""Test that extensions from URLs are normalized to lowercase"""
|
||||
test_cases = [
|
||||
("https://example.com/image.JPG", ".jpg"),
|
||||
("https://example.com/image.jpeg", ".jpeg"),
|
||||
("https://example.com/image.JPEG", ".jpeg"),
|
||||
("https://example.com/image.jpg", ".jpg"),
|
||||
("https://example.com/image.PNG", ".png"),
|
||||
("https://example.com/image.GIF", ".gif"),
|
||||
]
|
||||
|
||||
for url, expected in test_cases:
|
||||
mock_submission = MagicMock()
|
||||
mock_submission.id = "test123"
|
||||
|
||||
resource = Resource(mock_submission, url, lambda: None)
|
||||
assert resource.extension == expected, f"URL {url} should normalize to {expected}, got {resource.extension}"
|
||||
|
||||
def test_reddit_media_urls_normalized(self):
|
||||
"""Test that Reddit media URLs are properly normalized"""
|
||||
test_cases = [
|
||||
("https://www.reddit.com/media?url=https%3A%2F%2Fi.redd.it%2Fr2mv10i4vkfd1.JPG", ".jpg"),
|
||||
("https://www.reddit.com/media?url=https%3A%2F%2Fi.redd.it%2Fr2mv10i4vkfd1.jpeg", ".jpeg"),
|
||||
("https://www.reddit.com/media?url=https%3A%2F%2Fi.redd.it%2Fr2mv10i4vkfd1.JPEG", ".jpeg"),
|
||||
("https://www.reddit.com/media?url=https%3A%2F%2Fi.redd.it%2Fr2mv10i4vkfd1.PNG", ".png"),
|
||||
]
|
||||
|
||||
for url, expected in test_cases:
|
||||
mock_submission = MagicMock()
|
||||
mock_submission.id = "test123"
|
||||
|
||||
resource = Resource(mock_submission, url, lambda: None)
|
||||
assert resource.extension == expected, f"Reddit media URL {url} should normalize to {expected}, got {resource.extension}"
|
||||
|
||||
def test_constructor_extensions_normalized(self):
|
||||
"""Test that extensions passed to constructor are normalized"""
|
||||
test_cases = [
|
||||
(".JPG", ".jpg"),
|
||||
(".JPEG", ".jpeg"),
|
||||
(".PNG", ".png"),
|
||||
(".GIF", ".gif"),
|
||||
]
|
||||
|
||||
for input_ext, expected in test_cases:
|
||||
mock_submission = MagicMock()
|
||||
mock_submission.id = "test123"
|
||||
|
||||
resource = Resource(mock_submission, "https://example.com/test", lambda: None, input_ext)
|
||||
assert resource.extension == expected, f"Constructor extension {input_ext} should normalize to {expected}, got {resource.extension}"
|
||||
|
||||
def test_magic_number_detection_normalized(self):
|
||||
"""Test that magic number detection returns normalized extensions"""
|
||||
mock_submission = MagicMock()
|
||||
mock_submission.id = "test123"
|
||||
|
||||
# Test JPEG magic number detection
|
||||
jpeg_content = b'\xFF\xD8\xFF' + b'0' * 100 # JPEG magic number
|
||||
resource = Resource(mock_submission, "https://example.com/no-extension", lambda params: jpeg_content)
|
||||
resource.download() # Trigger content-based detection
|
||||
assert resource.extension == ".jpg", f"Magic number detection should return .jpg, got {resource.extension}"
|
||||
Reference in New Issue
Block a user