def test_datetime_format_with_datetime_format():
iso8601 = [fake.iso8601()]
normal = [fake.date(pattern="%d-%m-%Y %H:%M:%S", end_datetime=None)]
values = iso8601 + normal
_paragraph = fake.paragraph_with_salt(values)
extracted_values = extractors.datetime(_paragraph, fmt=None)
assert compare_2_list(values, extracted_values)
def test_number_credit_card_jcb_extractor():
jcb15 = fake.credit_card_number(card_type='jcb15')
jcb16 = fake.credit_card_number(card_type='jcb16')
jcbs = [jcb15, jcb16]
_paragraph = fake.paragraph_with_salt(jcbs)
extracted_jcbs = extractors.credit_card(_paragraph, host='JCB')
assert compare_2_list(jcbs, extracted_jcbs)
def test_number_credit_card_discover_extractor():
discovers = [
fake.credit_card_number(card_type='discover') for _ in range(3)
]
_paragraph = fake.paragraph_with_salt(discovers)
extracted_discovers = extractors.credit_card(_paragraph, host='Discover')
assert compare_2_list(discovers, extracted_discovers)
def test_sha256_with_sha256_extractor():
sha256s = [
hashlib.sha256(fake.name().encode('utf-8')).hexdigest()
for _ in range(3)
]
_paragraph = fake.paragraph_with_salt(sha256s)
extracted_sha256s = extractors.sha256(_paragraph)
assert compare_2_list(sha256s, extracted_sha256s)
def test_number_credit_card_visa_extractor():
visa13 = fake.credit_card_number(card_type='visa13')
visa16 = fake.credit_card_number(card_type='visa16')
visa19 = fake.credit_card_number(card_type='visa19')
visas = [visa13, visa16, visa19]
_paragraph = fake.paragraph_with_salt(visas)
extractor_visas = extractors.credit_card(_paragraph, host='Visa')
assert compare_2_list(visas, extractor_visas)
def test_number_credit_card_master_card_extractor():
mastercards = [
fake.credit_card_number(card_type='mastercard') for _ in range(3)
]
_paragraph = fake.paragraph_with_salt(mastercards)
extracted_mastercards = extractors.credit_card(_paragraph,
host='Master Card')
assert compare_2_list(mastercards, extracted_mastercards)
def test_phone_number_extractor():
phonenumbers = [
'0' + ''.join([str(random.randint(1, 9)) for _ in range(9)])
for _ in range(3)
]
_paragraph = fake.paragraph_with_salt(phonenumbers)
extracted_phonenumbers = extractors.phone_number(_paragraph)
assert compare_2_list(phonenumbers, extracted_phonenumbers)
def test_hex_color_code_with_hex_color_code():
hex_colors = [
'#' + ''.join(
[random.choice(list('ABCDEFabcdef0123456789')) for _ in range(6)])
for _ in range(3)
]
_paragraph = fake.paragraph_with_salt(hex_colors)
extracted_hex_colors = extractors.hex_color_code(_paragraph)
assert compare_2_list(hex_colors, extracted_hex_colors)
def test_time_extractor():
short_times = [
fake.time(pattern="%H:%M", end_datetime=None) for _ in range(3)
]
long_times = [
fake.time(pattern="%H:%M:%S", end_datetime=None) for _ in range(3)
]
times = short_times + long_times
_paragraph = fake.paragraph_with_salt(times)
extracted_times = extractors.time(_paragraph)
assert compare_2_list(times, extracted_times)
def test_number_credit_card_all_type_extractor():
card_type = random.choice(('visa13', 'visa16', 'visa19', 'mastercard',
'diners', 'jcb15', 'jcb16', 'discover'))
value = fake.credit_card_number(card_type=card_type)
ret = extractors.credit_card(value, host=None)
assert ret, value
numbers = {
'Visa': [
fake.credit_card_number(
card_type=random.choice(['visa13', 'visa16', 'visa19']))
],
'Master Card': [fake.credit_card_number(card_type='mastercard')],
'Diners Club': [fake.credit_card_number(card_type='diners')],
'Discover': [fake.credit_card_number(card_type='discover')],
'JCB':
[fake.credit_card_number(card_type=random.choice(['jcb15', 'jcb16']))]
}
random_number = random.choice(list(numbers.items()))
_paragraph = fake.paragraph_with_salt([random_number[1][0]])
extracted = extractors.credit_card(_paragraph, host=None)
assert random_number[1][0] in extracted[random_number[0]]
def test_email_extractor():
emails = [fake.email() for _ in range(3)]
_paragraph = fake.paragraph_with_salt(emails)
extracted_emails = extractors.email(_paragraph)
assert compare_2_list(emails, extracted_emails)
def test_ipv6_with_ipv6_extractor():
ipv6s = [fake.ipv6() for _ in range(3)]
_paragraph = fake.paragraph_with_salt(ipv6s)
extracted_ipv6s = extractors.ipv6(_paragraph)
assert compare_2_list(ipv6s, extracted_ipv6s)
def test_domain_extractor():
domains = [fake.domain_name() for _ in range(3)]
_paragraph = fake.paragraph_with_salt(domains)
extracted_domains = extractors.domain_name(_paragraph)
assert compare_2_list(domains, extracted_domains)
def test_uuid_extractor():
uuids = [str(uuid.uuid4()) for _ in range(3)]
_paragraph = fake.paragraph_with_salt(uuids)
extracted_uuids = extractors.uuid(_paragraph)
assert compare_2_list(uuids, extracted_uuids)
def test_filename_extractor_with_invalid_system():
filenames = [fake.linux_filename() for _ in range(3)]
_paragraph = fake.paragraph_with_salt(filenames)
with pytest.raises(ValueError):
assert extractors.filename(_paragraph, system=fake.word())
def test_filename_extractor_with_system_is_none():
filenames = [fake.linux_filename() for _ in range(3)]
_paragraph = fake.paragraph_with_salt(filenames)
extracted_filenames = extractors.filename(_paragraph, system=None)
assert compare_2_list(filenames, extracted_filenames)
def test_datetime_normal_format_with_dattime_validator_follow_normal():
normal = [fake.date(pattern="%d-%m-%Y %H:%M:%S", end_datetime=None)]
_paragraph = fake.paragraph_with_salt(normal)
extracted_normal = extractors.datetime(_paragraph, fmt='normal')
assert compare_2_list(normal, extracted_normal)
def test_url_extractor():
urls = [fake.url() for _ in range(3)]
_paragraph = fake.paragraph_with_salt(urls)
extracted_urls = extractors.url(_paragraph)
assert compare_2_list(urls, extracted_urls)
def test_iso_8601_datetime_format_with_datetime_format_follow_iso_8601():
iso8601 = [fake.iso8601()]
_paragraph = fake.paragraph_with_salt(iso8601)
extracted_iso8601 = extractors.datetime(_paragraph, fmt='iso-8601')
assert compare_2_list(iso8601, extracted_iso8601)