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)