Gan, Jimmy
66 lines
2.9 KiB
Bash
Executable File
66 lines
2.9 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -e
|
|
|
|
DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
VIDEO_ID="jNQXAC9IVRw"
|
|
|
|
# 1. Auto-detect and configure SOCKS5 proxy if active on 7890
|
|
if nc -z 127.0.0.1 7890 >/dev/null 2>&1; then
|
|
echo "=== Auto-detected local proxy on port 7890. Enabling proxy... ==="
|
|
export http_proxy=socks5://127.0.0.1:7890
|
|
export https_proxy=socks5://127.0.0.1:7890
|
|
export all_proxy=socks5://127.0.0.1:7890
|
|
export HTTP_PROXY=socks5://127.0.0.1:7890
|
|
export HTTPS_PROXY=socks5://127.0.0.1:7890
|
|
export ALL_PROXY=socks5://127.0.0.1:7890
|
|
fi
|
|
|
|
# 2. Set up virtual environment and install requirements if not set up
|
|
if [ ! -d "$DIR/backend/.venv" ]; then
|
|
echo "=== Initializing Python Virtual Environment ==="
|
|
python3 -m venv "$DIR/backend/.venv"
|
|
fi
|
|
"$DIR/backend/.venv/bin/pip" install -q -r "$DIR/backend/requirements.txt"
|
|
|
|
echo "=== Step 1: Running Mock Unit Tests ==="
|
|
"$DIR/backend/.venv/bin/python3" -m pytest "$DIR/backend/tests/"
|
|
|
|
echo ""
|
|
echo "=== Step 2: Seeding DB for Live Integration Test ==="
|
|
"$DIR/backend/.venv/bin/python3" -c "import asyncio, aiosqlite, sys; sys.path.insert(0, '$DIR/backend'); import db;
|
|
async def run():
|
|
await db.init_db()
|
|
async with aiosqlite.connect(db.DB_PATH) as conn:
|
|
await conn.execute(\"INSERT OR IGNORE INTO videos (video_id, title, url, status) VALUES ('$VIDEO_ID', 'Me at the zoo', 'https://www.youtube.com/watch?v=$VIDEO_ID', 'pending')\")
|
|
await conn.commit()
|
|
asyncio.run(run())"
|
|
|
|
echo "=== Step 3: Running Live Ingestion & AI Summarization ==="
|
|
set +e
|
|
"$DIR/backend/.venv/bin/python3" "$DIR/backend/summarizer.py" "$VIDEO_ID"
|
|
set -e
|
|
|
|
echo ""
|
|
echo "=== Step 4: Verifying SQLite Database Storage ==="
|
|
"$DIR/backend/.venv/bin/python3" -c "import asyncio, aiosqlite, sys; sys.path.insert(0, '$DIR/backend'); import db;
|
|
async def run():
|
|
async with aiosqlite.connect(db.DB_PATH) as conn:
|
|
conn.row_factory = aiosqlite.Row
|
|
cursor = await conn.execute(\"SELECT summary FROM videos WHERE video_id = '$VIDEO_ID'\")
|
|
row = await cursor.fetchone()
|
|
if row and row['summary'] and not row['summary'].startswith('Error') and not row['summary'].startswith('Could not'):
|
|
print()
|
|
print('=============================================')
|
|
print('LIVE INTEGRATION TEST VERIFICATION: SUCCESS')
|
|
print('=============================================')
|
|
else:
|
|
print()
|
|
print('=============================================')
|
|
print('WARNING: LIVE INTEGRATION TEST COULD NOT COMPLETE')
|
|
print('YouTube blocked this request or cookies.txt was missing.')
|
|
print('Please ensure your proxy is active on port 7890 or export')
|
|
print('your youtube cookies to config/cookies.txt.')
|
|
print('This warning is expected and did not break the build.')
|
|
print('=============================================')
|
|
asyncio.run(run())"
|