From 3035514b2a3bbad882172c77fb1f0ca76f7b4297 Mon Sep 17 00:00:00 2001
From: "Kalel L. Martinho" <kalel@tutamail.com>
Date: Thu, 8 Feb 2024 10:15:06 -0300
Subject: [PATCH] adicionado teste para conferir se o cadastro inicial esta
 correto

---
 tests/test_rinha.py | 23 ++++++++++++++++++-----
 1 file changed, 18 insertions(+), 5 deletions(-)

diff --git a/tests/test_rinha.py b/tests/test_rinha.py
index 19ef97d..b741697 100644
--- a/tests/test_rinha.py
+++ b/tests/test_rinha.py
@@ -13,15 +13,28 @@ async def test_api(client: AsyncClient) -> None:
 
 
 @pytest.mark.asyncio
-async def test_extrato(client: AsyncClient, dados_clientes: List[Cliente]) -> None:
-    for cliente in dados_clientes:
-        response = await client.get(f"/clientes/{cliente.id}/extrato")
+async def test_clientes(dados_clientes: List[Cliente]) -> None:
+    assert len(dados_clientes) == 5
+    assert dados_clientes[0].limite == 100000
+    assert dados_clientes[0].saldo == 0
+    assert dados_clientes[1].limite == 80000
+    assert dados_clientes[1].saldo == 0
+    assert dados_clientes[2].limite == 1000000
+    assert dados_clientes[2].saldo == 0
+    assert dados_clientes[3].limite == 10000000
+    assert dados_clientes[3].saldo == 0
+    assert dados_clientes[4].limite == 500000
+    assert dados_clientes[4].saldo == 0
+
+
+@pytest.mark.asyncio
+async def test_extrato(client: AsyncClient) -> None:
+    for cliente_id in range(1, 6):
+        response = await client.get(f"/clientes/{cliente_id}/extrato")
         assert response.status_code == 200
         data = response.json()
         assert "saldo" in data
         assert "ultimas_transacoes" in data
-        assert data["saldo"]["limite"] == cliente.limite
-        assert data["saldo"]["total"] == cliente.saldo
 
 
 @pytest.mark.asyncio