Return None when sanitizing shop domain if None passed in

Author: NabeelAhsen

shopify/utils/shop_url.py

@@ -10,7 +10,7 @@
 
 
 def sanitize_shop_domain(shop_domain, myshopify_domain="myshopify.com"):
-    name = str(shop_domain).lower().strip()
+    name = str(shop_domain or "").lower().strip()
     if myshopify_domain not in name and "." not in name:
         name += ".{domain}".format(domain=myshopify_domain)
     name = re.sub(r"https?://", "", name)

test/utils/shop_url_test.py

@@ -42,3 +42,6 @@ def test_returns_hostname_for_custom_shop_domains(self):
         ]
 
         self.assertTrue(all(shop == "my-shop.myshopify.io" for shop in sanitized_shops))
+
+    def test_returns_none_for_none_type(self):
+        self.assertIsNone(shop_url.sanitize_shop_domain(None))