discord-havoc/tests/uploader_tests.py

"""
Tests for both Catbox & Litterbox
"""

import os
import logging
import asyncio
from typing import Optional
from util.litterbox import LitterboxAsync
from util.catbox import CatboxAsync

logger = logging.getLogger()
logger.setLevel(logging.INFO)


async def test() -> None:
    f = os.path.join(os.path.expanduser("~"), "qu.png")
    box1: LitterboxAsync = LitterboxAsync()
    box2: CatboxAsync = CatboxAsync()
    url1: Optional[str] = await box1.upload(f)
    url2: Optional[str] = await box2.upload(f)
    logging.info(
        """Uploaded URLs:
          Litter - %s\n
          Cat - %s""",
        url1,
        url2,
    )


asyncio.run(test())