Skip to content

Commit

Permalink
Add test_google_sheets_real_fastapi_team_end2end test (#825)
Browse files Browse the repository at this point in the history
  • Loading branch information
rjambrecic committed Jul 10, 2024
1 parent 42e0038 commit cf0b903
Showing 1 changed file with 29 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -47,13 +47,12 @@ def test_init(self, mock_get_conv_uuid: Iterator[Any]) -> None:
team_class=GBBGoogleSheetsTeam,
)

@pytest.mark.flaky
@pytest.mark.openai
@pytest.mark.fastapi_openapi_team
def test_google_sheets_team_end2end(
self, google_sheets_fastapi_openapi_url: str, mock_get_conv_uuid: Iterator[Any]
def _test_google_sheets_team_end2end(
self,
user_id: int,
google_sheets_fastapi_openapi_url: str,
mock_get_conv_uuid: Iterator[Any],
) -> None:
user_id = 123
with mock_get_conv_uuid:
google_sheets_team = GBBGoogleSheetsTeam(
task="Do your job.",
Expand Down Expand Up @@ -99,3 +98,27 @@ def test_google_sheets_team_end2end(
assert (
len(expected_messages) == 0
), f"Expected messages left: {expected_messages}"

@pytest.mark.flaky
@pytest.mark.openai
@pytest.mark.fastapi_openapi_team
def test_google_sheets_team_end2end(
self, google_sheets_fastapi_openapi_url: str, mock_get_conv_uuid: Iterator[Any]
) -> None:
self._test_google_sheets_team_end2end(
user_id=123,
google_sheets_fastapi_openapi_url=google_sheets_fastapi_openapi_url,
mock_get_conv_uuid=mock_get_conv_uuid,
)

@pytest.mark.flaky
@pytest.mark.openai
@pytest.mark.fastapi_openapi_team
def test_google_sheets_real_fastapi_team_end2end(
self, mock_get_conv_uuid: Iterator[Any]
) -> None:
self._test_google_sheets_team_end2end(
user_id=13,
google_sheets_fastapi_openapi_url="https://google-sheets.tools.staging.fastagency.ai/openapi.json",
mock_get_conv_uuid=mock_get_conv_uuid,
)

0 comments on commit cf0b903

Please sign in to comment.