From 734cf041083aa9a9f89a250a5f2e282ea080e119 Mon Sep 17 00:00:00 2001
From: "Sean P. Kelly" <seankell@amazon.com>
Date: Mon, 15 Aug 2022 21:57:25 +0000
Subject: [PATCH] schnauzer: add new regions to pause registry map

---
 sources/api/schnauzer/src/helpers.rs | 84 ++++++++++++----------------
 1 file changed, 36 insertions(+), 48 deletions(-)

diff --git a/sources/api/schnauzer/src/helpers.rs b/sources/api/schnauzer/src/helpers.rs
index a3e8d3a7014..2a58adb763e 100644
--- a/sources/api/schnauzer/src/helpers.rs
+++ b/sources/api/schnauzer/src/helpers.rs
@@ -68,18 +68,23 @@ lazy_static! {
         m.insert("ap-northeast-2", "602401143452");
         m.insert("ap-northeast-3", "602401143452");
         m.insert("ap-south-1", "602401143452");
+        m.insert("ap-south-2", "900889452093");
         m.insert("ap-southeast-1", "602401143452");
         m.insert("ap-southeast-2", "602401143452");
         m.insert("ap-southeast-3", "296578399912");
+        m.insert("ap-southeast-4", "491585149902");
         m.insert("ca-central-1", "602401143452");
         m.insert("cn-north-1", "918309763551");
         m.insert("cn-northwest-1", "961992271922");
         m.insert("eu-central-1", "602401143452");
+        m.insert("eu-central-2", "900612956339");
         m.insert("eu-north-1", "602401143452");
         m.insert("eu-south-1", "590381155156");
+        m.insert("eu-south-2", "455263428931");
         m.insert("eu-west-1", "602401143452");
         m.insert("eu-west-2", "602401143452");
         m.insert("eu-west-3", "602401143452");
+        m.insert("me-central-1", "759879836304");
         m.insert("me-south-1", "558608220178");
         m.insert("sa-east-1", "602401143452");
         m.insert("us-east-1", "602401143452");
@@ -1840,56 +1845,39 @@ mod test_pause_registry {
 
     const CONTAINER_TEMPLATE: &str = "{{ pause-prefix settings.aws.region }}/container:tag";
 
-    const EXPECTED_URL_EU_CENTRAL_1: &str =
-        "602401143452.dkr.ecr.eu-central-1.amazonaws.com/container:tag";
-
-    const EXPECTED_URL_AF_SOUTH_1: &str =
-        "877085696533.dkr.ecr.af-south-1.amazonaws.com/container:tag";
-
-    const EXPECTED_URL_XY_ZTOWN_1: &str =
-        "602401143452.dkr.ecr.us-east-1.amazonaws.com/container:tag";
-
-    const EXPECTED_URL_CN_NORTH_1: &str =
-        "918309763551.dkr.ecr.cn-north-1.amazonaws.com.cn/container:tag";
-
-    #[test]
-    fn url_eu_central_1() {
-        let result = setup_and_render_template(
-            CONTAINER_TEMPLATE,
-            &json!({"settings": {"aws": {"region": "eu-central-1"}}}),
-        )
-        .unwrap();
-        assert_eq!(result, EXPECTED_URL_EU_CENTRAL_1);
-    }
-
-    #[test]
-    fn url_af_south_1() {
-        let result = setup_and_render_template(
-            CONTAINER_TEMPLATE,
-            &json!({"settings": {"aws": {"region": "af-south-1"}}}),
-        )
-        .unwrap();
-        assert_eq!(result, EXPECTED_URL_AF_SOUTH_1);
-    }
-
-    #[test]
-    fn url_fallback() {
-        let result = setup_and_render_template(
-            CONTAINER_TEMPLATE,
-            &json!({"settings": {"aws": {"region": "xy-ztown-1"}}}),
-        )
-        .unwrap();
-        assert_eq!(result, EXPECTED_URL_XY_ZTOWN_1);
-    }
+    const PAUSE_REGISTRY_TESTS: &[(&str, &str)] = &[
+        (
+            "eu-central-1",
+            "602401143452.dkr.ecr.eu-central-1.amazonaws.com/container:tag",
+        ),
+        (
+            "af-south-1",
+            "877085696533.dkr.ecr.af-south-1.amazonaws.com/container:tag",
+        ),
+        (
+            "xy-ztown-1",
+            "602401143452.dkr.ecr.us-east-1.amazonaws.com/container:tag",
+        ),
+        (
+            "cn-north-1",
+            "918309763551.dkr.ecr.cn-north-1.amazonaws.com.cn/container:tag",
+        ),
+        (
+            "ap-southeast-4",
+            "491585149902.dkr.ecr.ap-southeast-4.amazonaws.com/container:tag",
+        ),
+    ];
 
     #[test]
-    fn url_china() {
-        let result = setup_and_render_template(
-            CONTAINER_TEMPLATE,
-            &json!({"settings": {"aws": {"region": "cn-north-1"}}}),
-        )
-        .unwrap();
-        assert_eq!(result, EXPECTED_URL_CN_NORTH_1);
+    fn pause_container_registry() {
+        for (region, expected) in PAUSE_REGISTRY_TESTS {
+            let result = setup_and_render_template(
+                CONTAINER_TEMPLATE,
+                &json!({"settings": {"aws": {"region": *region}}}),
+            )
+            .unwrap();
+            assert_eq!(result, *expected);
+        }
     }
 }