openobserve/src/common/meta/proxy.rs

// Copyright 2025 OpenObserve Inc.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

use serde::{Deserialize, Serialize};
use utoipa::ToSchema;

#[derive(Clone, Debug, Default, Eq, PartialEq, Serialize, Deserialize, ToSchema)]
pub struct QueryParamProxyURL {
    #[serde(alias = "proxy-token")]
    pub proxy_token: String,
}

#[derive(Clone, Debug, Default, Eq, PartialEq, Serialize, Deserialize, ToSchema)]
pub struct PathParamProxyURL {
    pub target_url: String,
    pub org_id: String,
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_query_param_proxy_url() {
        let proxy_url = QueryParamProxyURL {
            proxy_token: "test-token".to_string(),
        };

        assert_eq!(proxy_url.proxy_token, "test-token");
    }

    #[test]
    fn test_query_param_proxy_url_default() {
        let proxy_url = QueryParamProxyURL::default();
        assert_eq!(proxy_url.proxy_token, "");
    }

    #[test]
    fn test_query_param_proxy_url_serialization() {
        let proxy_url = QueryParamProxyURL {
            proxy_token: "test-token".to_string(),
        };

        let serialized = serde_json::to_string(&proxy_url).unwrap();
        let deserialized: QueryParamProxyURL = serde_json::from_str(&serialized).unwrap();

        assert_eq!(proxy_url, deserialized);
    }

    #[test]
    fn test_query_param_proxy_url_alias() {
        let json = r#"{"proxy-token": "test-token"}"#;
        let proxy_url: QueryParamProxyURL = serde_json::from_str(json).unwrap();
        assert_eq!(proxy_url.proxy_token, "test-token");
    }

    #[test]
    fn test_path_param_proxy_url() {
        let proxy_url = PathParamProxyURL {
            target_url: "http://example.com".to_string(),
            org_id: "org123".to_string(),
        };

        assert_eq!(proxy_url.target_url, "http://example.com");
        assert_eq!(proxy_url.org_id, "org123");
    }

    #[test]
    fn test_path_param_proxy_url_default() {
        let proxy_url = PathParamProxyURL::default();
        assert_eq!(proxy_url.target_url, "");
        assert_eq!(proxy_url.org_id, "");
    }

    #[test]
    fn test_path_param_proxy_url_serialization() {
        let proxy_url = PathParamProxyURL {
            target_url: "http://example.com".to_string(),
            org_id: "org123".to_string(),
        };

        let serialized = serde_json::to_string(&proxy_url).unwrap();
        let deserialized: PathParamProxyURL = serde_json::from_str(&serialized).unwrap();

        assert_eq!(proxy_url, deserialized);
    }

    #[test]
    fn test_path_param_proxy_url_equality() {
        let url1 = PathParamProxyURL {
            target_url: "http://example.com".to_string(),
            org_id: "org123".to_string(),
        };

        let url2 = PathParamProxyURL {
            target_url: "http://example.com".to_string(),
            org_id: "org123".to_string(),
        };

        let url3 = PathParamProxyURL {
            target_url: "http://different.com".to_string(),
            org_id: "org123".to_string(),
        };

        assert_eq!(url1, url2);
        assert_ne!(url1, url3);
    }
}