diff --git a/openbb_platform/extensions/news/integration/test_news_api.py b/openbb_platform/extensions/news/integration/test_news_api.py index 9b94ac1796b..f7e16ee254f 100644 --- a/openbb_platform/extensions/news/integration/test_news_api.py +++ b/openbb_platform/extensions/news/integration/test_news_api.py @@ -157,3 +157,16 @@ def test_news_company(params, headers): result = requests.get(url, headers=headers, timeout=10) assert isinstance(result, requests.Response) assert result.status_code == 200 + + +@pytest.mark.skip(reason="No providers implement this yet.") +@pytest.mark.parametrize("params", []) +@pytest.mark.integration +def test_news_sector(params, headers): + params = {p: v for p, v in params.items() if v} + + query_str = get_querystring(params, []) + url = f"http://0.0.0.0:8000/api/v1/news/sector?{query_str}" + result = requests.get(url, headers=headers, timeout=10) + assert isinstance(result, requests.Response) + assert result.status_code == 200 diff --git a/openbb_platform/extensions/news/integration/test_news_python.py b/openbb_platform/extensions/news/integration/test_news_python.py index 9874f633b77..b78d548def7 100644 --- a/openbb_platform/extensions/news/integration/test_news_python.py +++ b/openbb_platform/extensions/news/integration/test_news_python.py @@ -150,3 +150,15 @@ def test_news_company(params, obb): assert result assert isinstance(result, OBBject) assert len(result.results) > 0 + + +@pytest.mark.skip(reason="Not providers implement this yet.") +@pytest.mark.parametrize("params", []) +@pytest.mark.integration +def test_news_sector(params, obb): + params = {p: v for p, v in params.items() if v} + + result = obb.news.sector(**params) + assert result + assert isinstance(result, OBBject) + assert len(result.results) > 0 diff --git a/openbb_platform/extensions/news/openbb_news/news_router.py b/openbb_platform/extensions/news/openbb_news/news_router.py index 742b922a383..178ae242aac 100644 --- a/openbb_platform/extensions/news/openbb_news/news_router.py +++ b/openbb_platform/extensions/news/openbb_news/news_router.py @@ -35,3 +35,14 @@ async def company( ) -> OBBject[BaseModel]: """Company News. Get news for one or more companies.""" return await OBBject.from_query(Query(**locals())) + + +@router.command(model="SectorNews") +async def sector( + cc: CommandContext, + provider_choices: ProviderChoices, + standard_params: StandardParams, + extra_params: ExtraParams, +) -> OBBject[BaseModel]: + """Sector News. Get news for one or more sectors.""" + return await OBBject.from_query(Query(**locals()))