2024-10-27 12:34:00+04:00

.github/workflows/build-and-push.yml

@@ -1,3 +1,5 @@
+# build-and-push.yml
+
 name: Build and Push Docker Image
 
 on:

.github/workflows/lint-markdown.yml

@@ -1,3 +1,5 @@
+# lint-markdown.yml
+
 ---
 name: Markdown Lint
 

.markdownlint.yaml

@@ -1,3 +1,5 @@
+# .markdownlint.yaml
+
 ---
 MD013: false
 default: true

.pre-commit-config.yaml

@@ -1,3 +1,5 @@
+# .pre-commit-config.yaml
+
 ---
 repos:
   - hooks:

app/__init__.py

@@ -1 +1 @@
-# app/__init__.py
+# __init__.py

app/api.py

@@ -1,3 +1,5 @@
+# api.py
+
 from fastapi import APIRouter, HTTPException, Body
 from pydantic import BaseModel
 from .url_processing import process_url_request

app/bot.py

@@ -1,4 +1,5 @@
-# app/bot.py
+# bot.py
+
 import asyncio
 import logging
 from aiogram import Bot, Dispatcher, types

app/config.py

@@ -1,4 +1,5 @@
-# app/config.py
+# config.py
+
 import os
 from dotenv import load_dotenv
 from pydantic_settings import BaseSettings  # Updated import

app/download.py

@@ -1,4 +1,5 @@
-# app/download.py
+# download.py
+
 import os
 import logging
 import yt_dlp

app/main.py

@@ -1,4 +1,5 @@
-# app/main.py
+# main.py
+
 import logging
 import asyncio
 from fastapi import FastAPI

app/url_processing.py

@@ -1,4 +1,5 @@
-# app/url_processing.py
+# url_processing.py
+
 import os
 import logging
 from urllib.parse import urlparse, urlunparse
@@ -16,7 +17,7 @@ async def process_url_request(url: str) -> str:
         elif "youtube.com/shorts" in final_url:
             sanitized_url = final_url
         else:
-            return "Unsupported URL"
+            return f"Unsupported URL {final_url}"
         video_path = await yt_dlp_download(sanitized_url)
         return video_path if video_path else "Download failed"
     except Exception as e:
@@ -33,5 +34,4 @@ def follow_redirects(url: str, timeout=10) -> str:
 
 def transform_tiktok_url(url: str) -> str:
     parsed_url = urlparse(url)
-    video_id = parsed_url.path.split("/")[-1]
-    return f"https://www.tiktok.com/embed/v2/{video_id}"
+    return f"{parsed_url.scheme}://{parsed_url.netloc}{parsed_url.path}"

docker-compose.yml

@@ -1,3 +1,5 @@
+# docker-compose.yml
+
 ---
 networks:
   traefik_default:

nginx/conf.d/default.conf

@@ -1,3 +1,5 @@
+# default.conf
+
 server {
     listen 80;
 

pyproject.toml

@@ -1,3 +1,5 @@
+# pyproject.toml
+
 [tool.poetry]
 name = "url-fairy-bot"
 version = "2.0.0"
@@ -21,6 +23,8 @@ black = "^24.4.2"
 flake8 = "^7.0.0"
 pytest =  "^8.2.0"
 yamllint = "^1.35.1"
+httpx = "^0.27.2"
+pytest-asyncio = "^0.24.0"
 
 [tool.poetry.scripts]
 test = "pytest"

pytest.ini

@@ -0,0 +1,5 @@
+# pytest.ini
+
+[pytest]
+testpaths = tests
+addopts = --ignore=lib/python3.11/site-packages

tests/__init__.py

@@ -1 +1 @@
-#
+# __init__.py

tests/test_api.py

@@ -0,0 +1,12 @@
+# test_api.py
+
+import pytest
+from httpx import AsyncClient
+from app.main import app
+
+@pytest.mark.asyncio
+async def test_process_url():
+    async with AsyncClient(app=app, base_url="http://test") as ac:
+        response = await ac.post("/process_url/", json={"url": "https://example.com"})
+    assert response.status_code == 200
+    assert response.json()["status"] == "success"

tests/test_bot.py

@@ -0,0 +1,15 @@
+# test_bot.py
+
+import pytest
+from aiogram.types import Message
+from app.bot import handle_message
+
+@pytest.mark.asyncio
+async def test_handle_message():
+    class MockMessage:
+        text = "https://example.com"
+        async def reply(self, text):
+            assert "Unsupported URL https://example.com/" in text or "success" in text
+
+    message = MockMessage()
+    await handle_message(message)

tests/test_url_processing.py

@@ -0,0 +1,22 @@
+# test_url_processing.py
+
+import os
+import pytest
+from unittest.mock import patch, AsyncMock
+from app.url_processing import process_url_request, transform_tiktok_url
+
+@pytest.mark.asyncio
+async def test_tiktok_url_processing():
+    test_url = "https://vt.tiktok.com/ZSYFmgDeS/"
+    expected_file_path = "/tmp/url-fairy-bot-cache/https___www_tiktok_com___video_7371578973689482539.mp4"
+    transformed_url = transform_tiktok_url("https://www.tiktok.com/video/7371578973689482539")
+
+    # Mock follow_redirects to return the final TikTok URL format
+    with patch("app.url_processing.follow_redirects", return_value="https://www.tiktok.com/video/7371578973689482539"):
+        # Mock yt_dlp_download to simulate successful download
+        with patch("app.url_processing.yt_dlp_download", return_value=expected_file_path) as mock_download:
+            result = await process_url_request(test_url)
+
+            # Assertions
+            mock_download.assert_called_once_with(transformed_url)
+            assert result == expected_file_path