From 8a29ed218fbe88387aff720f117a883afb3211b7 Mon Sep 17 00:00:00 2001 From: Ivanildo Barauna de Souza Junior Date: Thu, 13 Jun 2024 14:09:07 -0300 Subject: [PATCH] tests: refactor tests for consider new lib estructure --- src/api_to_dataframe/__init__.py | 4 ++-- tests/test_run.py | 19 ++++++++++++------- 2 files changed, 14 insertions(+), 9 deletions(-) diff --git a/src/api_to_dataframe/__init__.py b/src/api_to_dataframe/__init__.py index 6e63670..f4f2620 100644 --- a/src/api_to_dataframe/__init__.py +++ b/src/api_to_dataframe/__init__.py @@ -1,2 +1,2 @@ -from api_to_dataframe.controller import client_builder as ClientBuilder -from api_to_dataframe.common.utils import retry_strategies as RetryStrategies +from api_to_dataframe.controller.client_builder import ClientBuilder +from api_to_dataframe.common.utils.retry_strategies import RetryStrategies diff --git a/tests/test_run.py b/tests/test_run.py index 612e84a..89a451b 100644 --- a/tests/test_run.py +++ b/tests/test_run.py @@ -1,17 +1,23 @@ import pytest import pandas as pd -import api_to_dataframe as api_to_dataframe +from api_to_dataframe import ClientBuilder @pytest.fixture() def setup(): - new_client = api_to_dataframe.ClientBuilder(endpoint="https://economia.awesomeapi.com.br/last/USD-BRL") + new_client = ClientBuilder(endpoint="https://economia.awesomeapi.com.br/last/USD-BRL") return new_client +@pytest.fixture() +def response_setup(): + new_client = ClientBuilder(endpoint="https://economia.awesomeapi.com.br/last/USD-BRL") + return new_client.get_api_data() + + def test_constructor_without_param(): with pytest.raises(ValueError): - new_client = api_to_dataframe.ClientBuilder(endpoint="") + new_client = ClientBuilder(endpoint="") def test_constructor_with_param(setup): @@ -22,11 +28,10 @@ def test_constructor_with_param(setup): def test_response_to_json(setup): new_client = setup - response = new_client._response_to_json() + response = new_client.get_api_data() assert isinstance(response, dict) -def test_to_dataframe(setup): - new_client = setup - df = new_client.to_dataframe() +def test_to_dataframe(response_setup): + df = ClientBuilder.api_to_dataframe(response_setup) assert isinstance(df, pd.DataFrame)