def test_no_load_called(
self,
):
with self.assertRaises(
expected_exception=ValueError,
):
pydomainextractor.extract('com')
def test_load_called_with_data(
self,
):
pydomainextractor.load(
'com\n'
)
self.assertEqual(
first=pydomainextractor.extract('com'),
second={
'subdomain': '',
'domain': '',
'suffix': 'com',
},
)
pydomainextractor.load(
'net\n'
)
self.assertEqual(
first=pydomainextractor.extract('com'),
second={
'subdomain': '',
'domain': 'com',
'suffix': '',
},
)
pydomainextractor.load(
'customtld\n'
)
self.assertEqual(
first=pydomainextractor.extract('google.customtld'),
second={
'subdomain': '',
'domain': 'google',
'suffix': 'customtld',
},
)
pydomainextractor.load(
'tld\n'
'custom.tld\n'
)
self.assertEqual(
first=pydomainextractor.extract('google.custom.tld'),
second={
'subdomain': '',
'domain': 'google',
'suffix': 'custom.tld',
},
)
def test_extract_only_tld(
self,
):
self.assertEqual(
first=pydomainextractor.extract('com'),
second={
'domain': '',
'subdomain': '',
'suffix': 'com',
},
)
self.assertEqual(
first=pydomainextractor.extract('jp.net'),
second={
'domain': '',
'subdomain': '',
'suffix': 'jp.net',
},
)
self.assertEqual(
first=pydomainextractor.extract('鹿児島.jp'),
second={
'domain': '',
'subdomain': '',
'suffix': '鹿児島.jp',
},
)
self.assertEqual(
first=pydomainextractor.extract('香格里拉'),
second={
'domain': '',
'subdomain': '',
'suffix': '香格里拉',
},
)
self.assertEqual(
first=pydomainextractor.extract('xn--32vp30h.jp'),
second={
'domain': '',
'subdomain': '',
'suffix': 'xn--32vp30h.jp',
},
)
def test_extract_only_domain_and_subdomain(
self,
):
self.assertEqual(
first=pydomainextractor.extract('nonexistenttld.nonexistenttld'),
second={
'domain': 'nonexistenttld',
'subdomain': 'nonexistenttld',
'suffix': '',
},
)
self.assertEqual(
first=pydomainextractor.extract('香格里拉香格里拉香格里拉.nonexistenttld'),
second={
'domain': 'nonexistenttld',
'subdomain': '香格里拉香格里拉香格里拉',
'suffix': '',
},
)
def test_load_called_without_data(
self,
):
pydomainextractor.load()
self.assertEqual(
first=pydomainextractor.extract('com'),
second={
'subdomain': '',
'domain': '',
'suffix': 'com',
},
)
def test_extract_all_parts(
self,
):
self.assertEqual(
first=pydomainextractor.extract('google.com'),
second={
'domain': 'google',
'subdomain': '',
'suffix': 'com',
},
)
self.assertEqual(
first=pydomainextractor.extract('subdomain.google.com'),
second={
'domain': 'google',
'subdomain': 'subdomain',
'suffix': 'com',
},
)
self.assertEqual(
first=pydomainextractor.extract('subsubdomain.subdomain.google.com'),
second={
'domain': 'google',
'subdomain': 'subsubdomain.subdomain',
'suffix': 'com',
},
)
self.assertEqual(
first=pydomainextractor.extract('subsubdomain.subdomain.google.香格里拉'),
second={
'domain': 'google',
'subdomain': 'subsubdomain.subdomain',
'suffix': '香格里拉',
},
)
self.assertEqual(
first=pydomainextractor.extract('subsubdomain.subdomain.google.鹿児島.jp'),
second={
'domain': 'google',
'subdomain': 'subsubdomain.subdomain',
'suffix': '鹿児島.jp',
},
)
self.assertEqual(
first=pydomainextractor.extract('subsubdomain.subdomain.google.xn--32vp30h.jp'),
second={
'domain': 'google',
'subdomain': 'subsubdomain.subdomain',
'suffix': 'xn--32vp30h.jp',
},
)
def test_special_cases(
self,
):
self.assertEqual(
first=pydomainextractor.extract('bla.ck'),
second={
'domain': '',
'subdomain': '',
'suffix': 'bla.ck',
},
)
self.assertEqual(
first=pydomainextractor.extract('a.bla.ck'),
second={
'domain': 'a',
'subdomain': '',
'suffix': 'bla.ck',
},
)
self.assertEqual(
first=pydomainextractor.extract('a.b.bla.ck'),
second={
'domain': 'b',
'subdomain': 'a',
'suffix': 'bla.ck',
},
)
self.assertEqual(
first=pydomainextractor.extract('www.ck'),
second={
'domain': 'www',
'subdomain': '',
'suffix': 'ck',
},
)
self.assertEqual(
first=pydomainextractor.extract('a.www.ck'),
second={
'domain': 'www',
'subdomain': 'a',
'suffix': 'ck',
},
)
self.assertEqual(
first=pydomainextractor.extract('a.bzz.dapps.earth'),
second={
'domain': '',
'subdomain': '',
'suffix': 'a.bzz.dapps.earth',
},
)
self.assertEqual(
first=pydomainextractor.extract('a.b.bzz.dapps.earth'),
second={
'domain': 'a',
'subdomain': '',
'suffix': 'b.bzz.dapps.earth',
},
)