Add test for multiservice-discovery

rs/ic-observability/multiservice-discovery-shared/src/builders/prometheus_config_structure.rs

@@ -1,13 +1,13 @@
 use std::collections::{BTreeMap, BTreeSet};
 
 use ic_types::PrincipalId;
-use serde::{Serialize, Serializer};
+use serde::{Deserialize, Serialize, Serializer};
 use service_discovery::job_types::JobType;
 use service_discovery::jobs::Job;
 
 use crate::{builders::ConfigBuilder, contracts::target::TargetDto};
 
-#[derive(Serialize, Debug, Clone, PartialEq, PartialOrd, Ord, Eq)]
+#[derive(Serialize, Deserialize, Debug, Clone, PartialEq, PartialOrd, Ord, Eq)]
 pub struct PrometheusStaticConfig {
     pub targets: BTreeSet<String>,
     pub labels: BTreeMap<String, String>,
@@ -51,10 +51,10 @@ fn get_endpoints(target_group: TargetDto, job: JobType) -> BTreeSet<String> {
         .collect()
 }
 
-const IC_NAME: &str = "ic";
-const IC_NODE: &str = "ic_node";
-const IC_SUBNET: &str = "ic_subnet";
-const JOB: &str = "job";
+pub const IC_NAME: &str = "ic";
+pub const IC_NODE: &str = "ic_node";
+pub const IC_SUBNET: &str = "ic_subnet";
+pub const JOB: &str = "job";
 // TODO: Re-add the labels below once we resolve the issues with the public dashboard queries
 // https://dfinity.atlassian.net/browse/OB-442
 // const DC: &str = "dc";

rs/ic-observability/multiservice-discovery/BUILD.bazel

@@ -7,6 +7,15 @@ DEPS = [
     "//rs/ic-observability/multiservice-discovery-shared"
 ]
 
+DEV_DEPENDENCIES = [
+    "@crate_index_dre//:assert_cmd",
+    "@crate_index_dre//:tokio",
+    "@crate_index_dre//:tempdir",
+    "@crate_index_dre//:anyhow",
+    "@crate_index_dre//:reqwest",
+    "@crate_index_dre//:serde_json",
+]
+
 rust_binary(
     name = "multiservice-discovery",
     srcs = glob(["src/**/*.rs"]),
@@ -40,3 +49,15 @@ rust_test(
         normal_dev = True,
     ) + DEPS,
 )
+
+rust_test(
+    name = "prom_targets_tests",
+    srcs = ["tests/tests.rs"],
+    data = [":multiservice-discovery"] + glob(["tests/test_data/**"]),
+    proc_macro_deps = all_crate_deps(
+        proc_macro_dev = True,
+    ),
+    deps = all_crate_deps(
+        normal_dev = True,
+    ) + DEPS + DEV_DEPENDENCIES,
+    )

rs/ic-observability/multiservice-discovery/Cargo.toml

@@ -29,3 +29,10 @@ slog-term = { workspace = true }
 tokio = { workspace = true }
 url = { workspace = true }
 warp = { workspace = true }
+
+[dev-dependencies]
+tempdir = "0.3.7"
+reqwest = "0.11.23"
+assert_cmd = "2.0.13"
+anyhow = "1.0.79"
+

rs/ic-observability/multiservice-discovery/tests/test_data/mercury/targets/time.local_store.v1.CertificationTime.pb

@@ -0,0 +1 @@
+��������