114 lines
3.6 KiB
Python
114 lines
3.6 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 = {"test": "data"}
|
||
|
|
dlp._set_cached_session("http://test.com/video", test_data)
|
||
|
|
|
||
|
|
cached = dlp._get_cached_session("http://test.com/video")
|
||
|
|
assert cached == test_data
|
||
|
|
|
||
|
|
def test_cache_expiry(self):
|
||
|
|
dlp.CACHE_TTL = 1
|
||
|
|
dlp._session_cache.clear()
|
||
|
|
dlp._cache_timestamps.clear()
|
||
|
|
|
||
|
|
dlp._set_cached_session("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"])
|