headers/common/
access_control_allow_headers.rs1use std::iter::FromIterator;
2
3use http::{HeaderName, HeaderValue};
4
5use crate::util::FlatCsv;
6
7#[derive(Clone, Debug, PartialEq)]
35pub struct AccessControlAllowHeaders(FlatCsv);
36
37derive_header! {
38    AccessControlAllowHeaders(_),
39    name: ACCESS_CONTROL_ALLOW_HEADERS
40}
41
42impl AccessControlAllowHeaders {
43    pub fn iter(&self) -> impl Iterator<Item = HeaderName> + '_ {
45        self.0
46            .iter()
47            .map(|s| s.parse().ok())
48            .take_while(|val| val.is_some())
49            .flatten()
50    }
51}
52
53impl FromIterator<HeaderName> for AccessControlAllowHeaders {
54    fn from_iter<I>(iter: I) -> Self
55    where
56        I: IntoIterator<Item = HeaderName>,
57    {
58        let flat = iter.into_iter().map(HeaderValue::from).collect();
59        AccessControlAllowHeaders(flat)
60    }
61}
62
63#[cfg(test)]
64mod tests {
65    use super::super::{test_decode, test_encode};
66    use super::*;
67
68    #[test]
69    fn iter() {
70        let allow_headers = test_decode::<AccessControlAllowHeaders>(&["foo, bar"]).unwrap();
71
72        let as_vec = allow_headers.iter().collect::<Vec<_>>();
73        assert_eq!(as_vec.len(), 2);
74        assert_eq!(as_vec[0], "foo");
75        assert_eq!(as_vec[1], "bar");
76    }
77
78    #[test]
79    fn from_iter() {
80        let allow: AccessControlAllowHeaders =
81            vec![::http::header::CACHE_CONTROL, ::http::header::IF_RANGE]
82                .into_iter()
83                .collect();
84
85        let headers = test_encode(allow);
86        assert_eq!(
87            headers["access-control-allow-headers"],
88            "cache-control, if-range"
89        );
90    }
91
92    #[test]
93    fn test_with_invalid() {
94        let allow_headers = test_decode::<AccessControlAllowHeaders>(&["foo foo, bar"]).unwrap();
95
96        assert!(allow_headers.iter().collect::<Vec<_>>().is_empty());
97    }
98}