azure_core_test 0.1.0

Utilities for testing client libraries built on azure_core.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

//! Policies for recordings.

use async_trait::async_trait;
use azure_core::{
    http::{
        policies::{Policy, PolicyResult},
        AsyncRawResponse, Context, Request,
    },
    test::RecordingMode,
};
use std::sync::Arc;

/// Adds the `x-recording-mode` header to test responses.
#[derive(Debug)]
pub struct RecordingModePolicy {
    mode: RecordingMode,
}

impl RecordingModePolicy {
    /// Create a new `RecordingModePolicy`.
    pub fn new(mode: RecordingMode) -> Self {
        Self { mode }
    }
}

#[async_trait]
impl Policy for RecordingModePolicy {
    async fn send(
        &self,
        ctx: &Context,
        request: &mut Request,
        next: &[Arc<dyn Policy>],
    ) -> PolicyResult {
        let resp = next[0].send(ctx, request, &next[1..]).await?;
        let (status, mut headers, body) = resp.deconstruct();
        headers.add(self.mode)?;

        Ok(AsyncRawResponse::new(status, headers, Box::pin(body)))
    }
}