def test_domain_name_more_than_two_levels_after_cn_tld(
self, mock_tld, mock_domain_word, faker):
provider = ZhCnInternetProvider(faker)
mock_tld.return_value = 'cn'
mock_domain_word.return_value = 'li'
for levels in range(3, 10):
with patch('faker.providers.internet.zh_CN.Provider.domain_name',
wraps=faker.domain_name) as mock_domain_name:
mock_tld.reset_mock()
mock_domain_word.reset_mock()
mock_domain_name.reset_mock()
domain_name = faker.domain_name(levels=levels)
domain_parts = domain_name.split('.')
# Same assertions as levels=2 for tld and second level if tld is cn
# But every level henceforth should return the mocked value
assert domain_parts[-1] == 'cn'
assert domain_parts[-2] in provider.second_level_domains
assert all(domain_part == 'li'
for domain_part in domain_parts[:-2])
# tld() method should only be called once, domain_word() will be called for each
# level after tld except the second, and recursive calls to domain_name() will be
# made for each level starting from the third level after tld
assert mock_tld.call_count == 1
assert mock_domain_word.call_count == levels - 1
assert mock_domain_name.call_count == levels - 2
def test_domain_name_one_level_after_tld(self, faker):
provider = ZhCnInternetProvider(faker)
for _ in range(100):
domain_name = faker.domain_name(levels=1)
domain_parts = domain_name.split('.')
assert len(domain_parts) == 2
assert domain_parts[-1] in provider.tlds.keys()
assert domain_parts[0] not in provider.second_level_domains
def test_domain_name_two_levels_after_cn_tld(self, mock_tld, mock_domain_word, faker):
provider = ZhCnInternetProvider(faker)
# If tld() returns cn, second level name should be selected from second_level_domains
# and domain_word() will only be called once which will be used for the third level
mock_tld.return_value = 'cn'
mock_domain_word.return_value = 'li'
for _ in range(100):
mock_domain_word.reset_mock()
domain_name = faker.domain_name(levels=2)
domain_parts = domain_name.split('.')
assert len(domain_parts) == 3
assert domain_parts[-1] == 'cn'
assert domain_parts[-2] in provider.second_level_domains
assert domain_parts[0] == 'li'
assert mock_domain_word.call_count == 1
