From 8d8c59b40819b3fdbbdc12b7b6754bcf2d734f85 Mon Sep 17 00:00:00 2001 From: Gonzalo Mellizo-Soto Date: Wed, 26 Jun 2024 16:02:35 +0200 Subject: [PATCH] Fix tests --- tests/test_agent.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_agent.py b/tests/test_agent.py index 9aeed74..e3bb858 100644 --- a/tests/test_agent.py +++ b/tests/test_agent.py @@ -3,6 +3,7 @@ import pytest from ape.exceptions import NetworkError from giza.cli.schemas.jobs import Job, JobList +from giza.cli.schemas.logs import Logs from giza.cli.schemas.proofs import Proof from giza.cli.schemas.verify import VerifyResponse @@ -28,6 +29,9 @@ def verify_proof(self, *args, **kwargs): verification_time=1, ) + def get_logs(self, *args, **kwargs): + return Logs(logs="dummy_logs") + class JobsClientStub: def __init__(self, *args, **kwargs): @@ -40,6 +44,9 @@ def create(self, *args, **kwargs): def get(self, *args, **kwargs): return Job(id=1, size="S", status="COMPLETED") + def get_logs(self, *args, **kwargs): + return Logs(logs="dummy_logs") + class AccountTestStub: def __init__(self, *args, **kwargs):