add tests

Author: DylanLovesCoffee

dogstatsd/src/aggregator.rs

@@ -7,7 +7,7 @@ use crate::constants;
 use crate::datadog::{self, Metric as MetricToShip, Series};
 use crate::errors;
 use crate::metric::{self, Metric, MetricValue, SortedTags};
-use crate::metrics_origins::get_metric_origin;
+use crate::origins::get_origin;
 use std::time;
 
 use datadog_protos::metrics::{Dogsketch, Sketch, SketchPayload, Metadata};
@@ -270,7 +270,7 @@ fn build_sketch(now: i64, entry: &Metric, mut base_tag_vec: SortedTags) -> Optio
     }
     sketch.set_tags(base_tag_vec.to_chars());
 
-    let metadata: Option<Metadata> = get_metric_origin(&name);
+    let metadata: Option<Metadata> = get_origin(&name);
     if let Some(metadata) = metadata {
         sketch.set_metadata(metadata);
     }

dogstatsd/src/lib.rs

@@ -8,4 +8,4 @@ pub mod dogstatsd;
 pub mod errors;
 pub mod flusher;
 pub mod metric;
-pub mod metrics_origins;
+pub mod origins;

dogstatsd/src/metrics_origins.rs renamed to dogstatsd/src/origins.rs

@@ -53,7 +53,7 @@ impl From<OriginService> for u32 {
     }
 }
 
-pub fn get_metric_origin(name: &str) -> Option<Metadata> {
+pub fn get_origin(name: &str) -> Option<Metadata> {
     let prefix = name.split('.').take(2).collect::<Vec<&str>>().join(".");
 
     match prefix {
@@ -105,3 +105,34 @@ pub fn get_metric_origin(name: &str) -> Option<Metadata> {
         _ => None,
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_origin_product() {
+        let origin_product: u32 = OriginProduct::Serverless.into();
+        assert_eq!(origin_product, 1);
+    }
+
+    #[test]
+    fn test_origin_category() {
+        let origin_category: u32 = OriginCategory::LambdaMetrics.into();
+        assert_eq!(origin_category, 38);
+    }
+
+    #[test]
+    fn test_origin_service() {
+        let origin_service: u32 = OriginService::Other.into();
+        assert_eq!(origin_service, 0);
+    }
+
+    #[test]
+    fn test_get_origin() {
+        let origin = get_origin("aws.lambda.enhanced.invocations");
+        assert_eq!(origin.as_ref().unwrap().origin.as_ref().unwrap().origin_product, 1);
+        assert_eq!(origin.as_ref().unwrap().origin.as_ref().unwrap().origin_category, 38);
+        assert_eq!(origin.as_ref().unwrap().origin.as_ref().unwrap().origin_service, 0);
+    }
+ }