Revision control
Copy as Markdown
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
*/
use once_cell::sync::Lazy;
use url::Url;
use url_macro::url;
#[derive(Clone, Copy, Debug, Default, Eq, PartialEq)]
pub enum Environment {
#[default]
Prod,
Staging,
#[cfg(test)]
Test,
}
impl Environment {
pub fn into_url(self, path: &str) -> Url {
let mut base = self.base_url();
// Ensure the path has a trailing slash so that `join` appends
// rather than replacing the last segment.
if !base.path().ends_with('/') {
base.set_path(&format!("{}/", base.path()));
}
base.join(path)
.expect("joining a path to a valid base URL must succeed")
}
fn base_url(self) -> Url {
match self {
Environment::Prod => MARS_API_ENDPOINT_PROD.clone(),
Environment::Staging => MARS_API_ENDPOINT_STAGING.clone(),
#[cfg(test)]
Environment::Test => Url::parse(&mockito::server_url()).unwrap(),
}
}
}
#[cfg(test)]
mod tests {
use url::Host;
use super::*;
#[test]
fn prod_endpoint_parses_and_is_expected() {
let url = Environment::Prod.into_url("ads");
assert_eq!(url.scheme(), "https");
assert_eq!(url.host(), Some(Host::Domain("ads.mozilla.org")));
assert_eq!(url.path(), "/v1/ads");
}
#[test]
fn staging_endpoint_parses_and_is_expected() {
let url = Environment::Staging.into_url("ads");
assert_eq!(url.scheme(), "https");
assert_eq!(url.host(), Some(Host::Domain("ads.allizom.org")));
assert_eq!(url.path(), "/v1/ads");
}
}