Changed auth token in policies (#339)

Author: MindFlavor

sdk/core/src/pipeline.rs

@@ -95,6 +95,18 @@ where
         self.http_client.as_ref()
     }
 
+    pub fn replace_policy(
+        &mut self,
+        policy: Arc<dyn Policy<C>>,
+        position: usize,
+    ) -> Arc<dyn Policy<C>> {
+        std::mem::replace(&mut self.pipeline[position], policy)
+    }
+
+    pub fn policies(&self) -> &[Arc<dyn Policy<C>>] {
+        &self.pipeline
+    }
+
     pub async fn send(
         &self,
         ctx: &mut PipelineContext<C>,

sdk/cosmos/src/clients/cosmos_client.rs

@@ -139,7 +139,16 @@ impl CosmosClient {
 
     /// Set the auth token used
     pub fn auth_token(&mut self, auth_token: AuthorizationToken) {
-        self.auth_token = auth_token;
+        // TODO: To remove once everything uses the AutorizationPolicy
+        self.auth_token = auth_token.clone();
+
+        // we replace the AuthorizationPolicy. This is
+        // the last-1 policy by construction.
+        let auth_policy: Arc<dyn azure_core::Policy<CosmosContext>> =
+            Arc::new(crate::AuthorizationPolicy::new(auth_token));
+
+        self.pipeline
+            .replace_policy(auth_policy, self.pipeline.policies().len() - 2);
     }
 
     /// Create a database