Provide initialization and reset fns for InnerMessageSender

Signed-off-by: Matthias Beyer <matthias.beyer@ifm.com>

1 files changed, 10 insertions, 0 deletions

diff --git a/crates/core/tedge_api/src/address.rs b/crates/core/tedge_api/src/address.rs
index 199e7852..4f786ea5 100644
--- a/crates/core/tedge_api/src/address.rs
+++ b/crates/core/tedge_api/src/address.rs
@@ -57,6 +57,16 @@ impl InnerMessageSender {
         Self { send_provider }
     }
 
+    pub async fn init_with(&self, producer: Box<MessageFutureProducer>) {
+        let mut lock = self.send_provider.write().await;
+        *lock = Some(producer);
+    }
+
+    pub async fn reset(&self) {
+        let mut lock = self.send_provider.write().await;
+        *lock = None;
+    }
+
     #[instrument(skip_all, level = "trace")]
     async fn send(&self, message: InternalMessage) -> Result<(), InternalMessage> {
         let lock = self.send_provider.read().await;