Refactor news & social modules

- update NewsAPIsTool
- update SocialAPIsTool
- add tests for NewsAPIsTool
- added some missing docs

tests/conftest.py

@@ -23,6 +23,7 @@ def pytest_configure(config:pytest.Config):
         ("social", "marks tests that use social media"),
         ("limited", "marks tests that have limited execution due to API constraints"),
         ("wrapper", "marks tests for wrapper handler"),
+        ("tools", "marks tests for tools"),
     ]
     for marker in markers:
         line = f"{marker[0]}: {marker[1]}"

tests/tools/test_news_tool.py

@@ -0,0 +1,54 @@
+import pytest
+from app.news import NewsAPIsTool
+
+
+@pytest.mark.limited
+@pytest.mark.tools
+@pytest.mark.news
+@pytest.mark.api
+class TestNewsAPITool:
+    def test_news_api_tool(self):
+        tool = NewsAPIsTool()
+        assert tool is not None
+
+    def test_news_api_tool_get_top(self):
+        tool = NewsAPIsTool()
+        result = tool.wrapper_handler.try_call(lambda w: w.get_top_headlines(total=2))
+        assert isinstance(result, list)
+        assert len(result) > 0
+        for article in result:
+            assert article.title is not None
+            assert article.source is not None
+
+    def test_news_api_tool_get_latest(self):
+        tool = NewsAPIsTool()
+        result = tool.wrapper_handler.try_call(lambda w: w.get_latest_news(query="crypto", total=2))
+        assert isinstance(result, list)
+        assert len(result) > 0
+        for article in result:
+            assert article.title is not None
+            assert article.source is not None
+
+    def test_news_api_tool_get_top__all_results(self):
+        tool = NewsAPIsTool()
+        result = tool.wrapper_handler.try_call_all(lambda w: w.get_top_headlines(total=2))
+        assert isinstance(result, dict)
+        assert len(result.keys()) > 0
+        print("Results from providers:", result.keys())
+        for provider, articles in result.items():
+            for article in articles:
+                print(provider, article.title)
+                assert article.title is not None
+                assert article.source is not None
+
+    def test_news_api_tool_get_latest__all_results(self):
+        tool = NewsAPIsTool()
+        result = tool.wrapper_handler.try_call_all(lambda w: w.get_latest_news(query="crypto", total=2))
+        assert isinstance(result, dict)
+        assert len(result.keys()) > 0
+        print("Results from providers:", result.keys())
+        for provider, articles in result.items():
+            for article in articles:
+                print(provider, article.title)
+                assert article.title is not None
+                assert article.source is not None