probe: payment init recv in ProviderProbe (#663)

golemfactory · Jan 12, 2024 · 4a16c73 · 4a16c73
1 parent a4296ba
commit 4a16c73
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/goth/runner/probe/__init__.py b/goth/runner/probe/__init__.py
@@ -498,6 +498,13 @@ class ProviderProbe(MarketApiMixin, PaymentApiMixin, Probe):
     """The agent component running `ya-provider` for this probe.
     This field is added for convenience to make getting this agent instance easier."""
 
+    async def _start_container(self) -> None:
+        await super()._start_container()
+
+        payment_driver = self.payment_config.driver
+        self.cli.payment_fund(payment_driver)
+        self.cli.payment_init(payment_driver, receiver_mode=True)
+
     def __init__(
         self,
         runner: "Runner",