28 lines
1023 B
Python
28 lines
1023 B
Python
import os
|
|
import re
|
|
import pytest
|
|
from app.api.social.reddit import MAX_COMMENTS, RedditWrapper
|
|
|
|
@pytest.mark.social
|
|
@pytest.mark.api
|
|
@pytest.mark.skipif(not(os.getenv("REDDIT_API_CLIENT_ID")) or not os.getenv("REDDIT_API_CLIENT_SECRET"), reason="REDDIT_CLIENT_ID and REDDIT_API_CLIENT_SECRET not set in environment variables")
|
|
class TestRedditWrapper:
|
|
def test_initialization(self):
|
|
wrapper = RedditWrapper()
|
|
assert wrapper is not None
|
|
assert wrapper.tool is not None
|
|
|
|
def test_get_top_crypto_posts(self):
|
|
wrapper = RedditWrapper()
|
|
posts = wrapper.get_top_crypto_posts(limit=2)
|
|
assert isinstance(posts, list)
|
|
assert len(posts) == 2
|
|
for post in posts:
|
|
assert post.title != ""
|
|
assert re.match(r'\d{4}-\d{2}-\d{2}', post.timestamp)
|
|
|
|
assert isinstance(post.comments, list)
|
|
assert len(post.comments) <= MAX_COMMENTS
|
|
for comment in post.comments:
|
|
assert comment.description != ""
|