diff --git a/tests/test_db_clientes.py b/tests/test_db_clientes.py index 456178e..676e1e9 100644 --- a/tests/test_db_clientes.py +++ b/tests/test_db_clientes.py @@ -98,12 +98,13 @@ def test_excluir_cliente_com_contato_bloqueia(cliente_id): finally: conn.close() - with pytest.raises(db_clientes.ClienteComDependenciasError): - db_clientes.excluir_cliente(cliente_id) - - conn = get_connection() try: - with conn.cursor() as cur: - cur.execute("DELETE FROM CLIENTE_CONTATO WHERE CLIENTE_ID = %s", (cliente_id,)) + with pytest.raises(db_clientes.ClienteComDependenciasError): + db_clientes.excluir_cliente(cliente_id) finally: - conn.close() + conn = get_connection() + try: + with conn.cursor() as cur: + cur.execute("DELETE FROM CLIENTE_CONTATO WHERE CLIENTE_ID = %s", (cliente_id,)) + finally: + conn.close()