Include the protocol as well.

Author: skytreader

microdata.py

@@ -125,8 +125,11 @@ def __repr__(self):
 
     @staticmethod
     def get_domain(url_string):
+        """
+        Get the domain _including_ the protocol specified, if any.
+        """
         if "://" in url_string:
-            return url_string.split("/")[2]
+            return "/".join(url_string.split("/")[0:3])
         else:
             return url_string.split("/")[0]
 

test.py

@@ -117,7 +117,7 @@ class URITest(unittest.TestCase):
 
     def test_get_domain(self):
         https_start = "https://github.com/edsu/microdata"
-        self.assertEqual("github.com", URI.get_domain(https_start))
+        self.assertEqual("https://github.com", URI.get_domain(https_start))
 
         no_https = "github.com/edsu/microdata"
         self.assertEqual("github.com", URI.get_domain(no_https))