Berack96
dfe3b4ad90
- Implemented DuckDuckGoWrapper for news retrieval using DuckDuckGo tools. - Added GoogleNewsWrapper for accessing Google News RSS feed. - Refactored CryptoPanicWrapper to unify get_top_headlines and get_latest_news methods. - Updated NewsApiWrapper to simplify top headlines retrieval. - Added tests for DuckDuckGo and Google News wrappers. - Enhanced documentation for CryptoPanicWrapper and NewsApiWrapper. - Created base module for social media integrations.
39 lines
1.5 KiB
Python
39 lines
1.5 KiB
Python
import os
|
|
import pytest
|
|
from app.news import CryptoPanicWrapper
|
|
|
|
|
|
@pytest.mark.limited
|
|
@pytest.mark.news
|
|
@pytest.mark.api
|
|
@pytest.mark.skipif(not os.getenv("CRYPTOPANIC_API_KEY"), reason="CRYPTOPANIC_API_KEY not set")
|
|
class TestCryptoPanicAPI:
|
|
|
|
def test_crypto_panic_api_initialization(self):
|
|
crypto = CryptoPanicWrapper()
|
|
assert crypto is not None
|
|
|
|
def test_crypto_panic_api_get_latest_news(self):
|
|
crypto = CryptoPanicWrapper()
|
|
articles = crypto.get_latest_news(query="", total=2)
|
|
assert isinstance(articles, list)
|
|
assert len(articles) == 2
|
|
for article in articles:
|
|
assert article.source is not None or article.source != ""
|
|
assert article.time is not None or article.time != ""
|
|
assert article.title is not None or article.title != ""
|
|
assert article.description is not None or article.description != ""
|
|
|
|
# Useless since both methods use the same endpoint
|
|
# def test_crypto_panic_api_get_top_headlines(self):
|
|
# crypto = CryptoPanicWrapper()
|
|
# articles = crypto.get_top_headlines(total=2)
|
|
# assert isinstance(articles, list)
|
|
# assert len(articles) == 2
|
|
# for article in articles:
|
|
# assert article.source is not None or article.source != ""
|
|
# assert article.time is not None or article.time != ""
|
|
# assert article.title is not None or article.title != ""
|
|
# assert article.description is not None or article.description != ""
|
|
|