Adjust class/module names to changes in goth's master

golemfactory · Apr 9, 2021 · 6ee7836 · 6ee7836
1 parent 75b1b34
commit 6ee7836
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/yagna/module/payments/test_payment_driver_list_cmd.py b/test/yagna/module/payments/test_payment_driver_list_cmd.py
@@ -12,10 +12,10 @@
 )
 from goth.node import node_environment
 from goth.runner import Runner
+from goth.runner.container.build import YagnaBuildEnvironment
 from goth.runner.container.payment import PaymentIdPool
 from goth.runner.container.yagna import YagnaContainerConfig
-from goth.runner.requestor import RequestorProbeWithApiSteps
-from goth.runner.container.build import YagnaBuildEnvironment
+from goth.runner.probe import RequestorProbe
 
 
 logger = logging.getLogger(__name__)
@@ -53,7 +53,7 @@ def _topology(payment_id_pool: PaymentIdPool) -> List[YagnaContainerConfig]:
     return [
         YagnaContainerConfig(
             name="requestor",
-            probe_type=RequestorProbeWithApiSteps,
+            probe_type=RequestorProbe,
             environment=requestor_env,
             payment_id=payment_id_pool.get_id(),
         ),
@@ -73,7 +73,7 @@ async def test_payment_driver_list(
     topology = _topology(payment_id_pool)
 
     async with runner(topology):
-        requestor = runner.get_probes(probe_type=RequestorProbeWithApiSteps)[0]
+        requestor = runner.get_probes(probe_type=RequestorProbe)[0]
 
         res = requestor.cli.payment_drivers()
         assert res and res.items()