Skip to content

Commit

Permalink
Fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Gonmeso committed Jun 26, 2024
1 parent 6aa1524 commit 8d8c59b
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions tests/test_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand All @@ -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):
Expand All @@ -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):
Expand Down

0 comments on commit 8d8c59b

Please sign in to comment.