From 0fe7276d447ef33f5885c465235aa3fccd111cd0 Mon Sep 17 00:00:00 2001
From: Jedrzej Stuczynski <jedrzej.stuczynski@gmail.com>
Date: Wed, 18 Dec 2019 11:22:46 +0000
Subject: [PATCH] Method to update auth token of provider client + removed mut
 from register

---
 src/clients/provider.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/clients/provider.rs b/src/clients/provider.rs
index 6b951dd0f4..cf986041ed 100644
--- a/src/clients/provider.rs
+++ b/src/clients/provider.rs
@@ -57,6 +57,10 @@ impl ProviderClient {
         }
     }
 
+    pub fn update_token(&mut self, auth_token: AuthToken) {
+        self.auth_token = Some(auth_token)
+    }
+
     pub async fn send_request(&self, bytes: Vec<u8>) -> Result<Vec<u8>, ProviderClientError> {
         let mut socket = tokio::net::TcpStream::connect(self.provider_network_address).await?;
         println!("keep alive: {:?}", socket.keepalive());
@@ -99,7 +103,7 @@ impl ProviderClient {
         Ok(())
     }
 
-    pub async fn register(&mut self) -> Result<AuthToken, ProviderClientError> {
+    pub async fn register(&self) -> Result<AuthToken, ProviderClientError> {
         if self.auth_token.is_some() {
             return Err(ProviderClientError::ClientAlreadyRegisteredError);
         }