Mikhail Yevchenko
01a376ae21
- Updated AGENTS.md to clarify dlp.py module usage and segment handling. - Modified README.md to include ALLOW_LOCAL configuration for testing. - Refactored app.py to streamline HLS proxy logic and improve error handling. - Enhanced dlp.py to optimize caching and segment retrieval processes. - Updated player.html to ensure proper JSON formatting for proxy URLs. - Improved test_integration.py to validate HLS segment proxying and added test for Pornhub HLS extraction. - Adjusted test_proxy.py to reflect changes in caching functions and data structure.
117 lines
3.8 KiB
Python
117 lines
3.8 KiB
Python
import pytest
|
|
import sys
|
|
import os
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
from utils import is_valid_url, extract_video_id, sanitize_path, get_error_message
|
|
import dlp
|
|
|
|
|
|
class TestURLValidation:
|
|
def test_valid_youtube_url(self):
|
|
assert is_valid_url("https://www.youtube.com/watch?v=dQw4w9WgXcQ")
|
|
assert is_valid_url("https://youtu.be/dQw4w9WgXcQ")
|
|
|
|
def test_valid_youtu_be(self):
|
|
assert is_valid_url("https://youtu.be/abc123")
|
|
|
|
def test_valid_pornhub_url(self):
|
|
assert is_valid_url("https://www.pornhub.com/view_video.php?viewkey=abc123")
|
|
|
|
def test_invalid_url(self):
|
|
assert not is_valid_url("")
|
|
assert not is_valid_url("not-a-url")
|
|
|
|
def test_disallowed_domain(self):
|
|
os.environ["VALIDATION_ENABLED"] = "true"
|
|
assert not is_valid_url("https://evil.com/video")
|
|
|
|
|
|
class TestVideoIDExtraction:
|
|
def test_extract_youtube_id(self):
|
|
assert extract_video_id("https://www.youtube.com/watch?v=dQw4w9WgXcQ") == "dQw4w9WgXcQ"
|
|
assert extract_video_id("https://youtu.be/dQw4w9WgXcQ") == "dQw4w9WgXcQ"
|
|
|
|
def test_extract_pornhub_id(self):
|
|
result = extract_video_id("https://www.pornhub.com/view_video.php?viewkey=ph123456")
|
|
assert result == "ph123456"
|
|
|
|
def test_extract_invalid(self):
|
|
assert extract_video_id("https://example.com/video") == ""
|
|
|
|
|
|
class TestPathSanitization:
|
|
def test_sanitize_normal_path(self):
|
|
assert sanitize_path("path/to/file") == "path/to/file"
|
|
|
|
def test_sanitize_prevents_traversal(self):
|
|
assert sanitize_path("../etc/passwd") == "etc/passwd"
|
|
assert sanitize_path("path/../etc/passwd") == "path/etc/passwd"
|
|
|
|
|
|
class TestCacheMechanics:
|
|
def test_cache_basic(self):
|
|
dlp._session_cache.clear()
|
|
dlp._cache_timestamps.clear()
|
|
|
|
test_data = {"title": "Test Video", "thumbnail": "http://test.com/thumb.jpg", "hls_url": "http://test.com/stream.m3u8"}
|
|
dlp._set_cached_info("http://test.com/video", test_data)
|
|
|
|
cached = dlp._get_cached_info("http://test.com/video")
|
|
assert cached is not None
|
|
assert cached["title"] == "Test Video"
|
|
assert cached["thumbnail"] == "http://test.com/thumb.jpg"
|
|
assert cached["hls_url"] == "http://test.com/stream.m3u8"
|
|
|
|
def test_cache_expiry(self):
|
|
dlp.CACHE_TTL = 1
|
|
dlp._session_cache.clear()
|
|
dlp._cache_timestamps.clear()
|
|
|
|
dlp._set_cached_info("http://test.com/video", {"data": "test"})
|
|
import time
|
|
time.sleep(1.1)
|
|
|
|
assert dlp._is_cache_expired("http://test.com/video") is True
|
|
|
|
dlp.CACHE_TTL = 31536000
|
|
|
|
|
|
class TestErrorMessages:
|
|
def test_get_error_message(self):
|
|
assert "Bad Request" in get_error_message(400)
|
|
assert "Forbidden" in get_error_message(403)
|
|
assert "Not Found" in get_error_message(404)
|
|
assert "Internal Server Error" in get_error_message(500)
|
|
|
|
|
|
class TestFlaskApp:
|
|
def test_index_route(self):
|
|
from app import app
|
|
with app.test_client() as client:
|
|
response = client.get("/")
|
|
assert response.status_code == 200
|
|
|
|
def test_player_route_missing_url(self):
|
|
from app import app
|
|
with app.test_client() as client:
|
|
response = client.get("/player")
|
|
assert response.status_code == 400
|
|
|
|
def test_player_route_invalid_url(self):
|
|
from app import app
|
|
with app.test_client() as client:
|
|
response = client.get("/player?url=https://evil.com/video")
|
|
assert response.status_code == 400
|
|
|
|
def test_hls_proxy_invalid_path(self):
|
|
from app import app
|
|
with app.test_client() as client:
|
|
response = client.get("/hls")
|
|
assert response.status_code == 400
|
|
|
|
|
|
if __name__ == "__main__":
|
|
pytest.main([__file__, "-v"])
|