test_email.py

#info@sunandsky.co.uk
from faker import Faker
from csirtg_indicator import Indicator

fake = Faker()


def _not(data):
    for d in data:
        d = Indicator(d)
        assert d.itype != 'email'


def test_email_ip():
    data = ['192.168.1.0/24', '192.168.1.1', '2001:1608:10:147::21', '2001:4860:4860::8888']
    _not(data)


def test_email_fqdn():
    data = [
        '1.2.3.4.org',
        'www41.xzmnt.com',
    ]
    _not(data)


def test_email_ok():
    data = [
        'w!es@example.org',
        '{in|fo}@sunandsky.co.uk',
        'w*es.info@sun-and-sky.co.uk',
        'mut-ti?25@minsal.cl'
    ]

    for d in data:
        d = Indicator(d)
        assert d.itype is 'email'

    i = Indicator('WES@barely3am.com')
    assert i.indicator == 'wes@barely3am.com'


def test_email_random():
    for d in range(0, 100):
        assert Indicator(indicator=fake.email()).itype == 'email'