fix: integration tests (#44)

Author: apogiatzis

tests/integration/conftest.py

@@ -24,11 +24,14 @@ def is_docker_running():
     """Checks if docker is running"""
     logger.debug("Checking if docker is running...")
     client = docker.from_env()
+    is_running = True
     try:
         client.info()
-        return True
     except Exception as e:
-        return False
+        is_running = False
+    finally:
+        client.close()
+    return is_running
 
 
 def is_docker_compose_installed():

tests/integration/test_ffs.py

@@ -29,14 +29,14 @@ def test_grpc_ffs_list_api(pygate_client: PowerGateClient, ffs_instance):
 
 
 def test_grpc_ffs_add_to_hot(pygate_client: PowerGateClient, ffs_instance):
-    res = pygate_client.ffs.add_to_hot(test_chunks(), ffs_instance.token)
+    res = pygate_client.ffs.add_to_hot(chunks(), ffs_instance.token)
 
     assert res is not None
     assert res.cid is not None
 
 
 def test_grpc_ffs_add_then_get_content(pygate_client: PowerGateClient, ffs_instance):
-    res = pygate_client.ffs.add_to_hot(test_chunks(), ffs_instance.token)
+    res = pygate_client.ffs.add_to_hot(chunks(), ffs_instance.token)
 
     assert res is not None
 
@@ -101,6 +101,6 @@ def test_send_fil(pygate_client: PowerGateClient, ffs_instance: CreateResponse):
     assert before_receiver_fil.balance < after_receiver_fil.balance
 
 
-def test_chunks():
+def chunks():
     for _ in range(1):
         yield StageRequest(chunk=bytes("test_content", "ASCII"))

tests/integration/test_wallet.py

@@ -39,7 +39,7 @@ def test_grpc_wallet_balance(pygate_client: PowerGateClient):
 
     balance_res = pygate_client.wallet.balance(new_res.address)
     assert type(balance_res) is BalanceResponse
-    assert balance_res.balance == 4000000000000000
+    assert balance_res.balance == 250000000000000000
 
 
 def test_send_file(pygate_client: PowerGateClient):