Skip to content

Commit 1d3209b

Browse files
authored
adding faker data (csirtgadgets#92)
1 parent 2fe0716 commit 1d3209b

7 files changed

+40
-3
lines changed

requirements.txt

+2-1
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,5 @@ arrow>=0.7.0
22
pytricia>=0.9.0
33
ipaddress>=1.0.16
44
pendulum>=0.5.2
5-
prettytable>=0.7.2
5+
prettytable>=0.7.2
6+
Faker==0.7.10

setup.py

+1
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@
5151
'ipaddress>=1.0.16',
5252
'pendulum>=0.5.2',
5353
'prettytable>=0.7.2',
54+
'Faker'
5455
],
5556
entry_points={
5657
'console_scripts': [

test/test_email.py

+7-2
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
11
2-
3-
2+
from faker import Faker
43
from csirtg_indicator import Indicator
54

5+
fake = Faker()
6+
67

78
def _not(data):
89
for d in data:
@@ -37,3 +38,7 @@ def test_email_ok():
3738
i = Indicator('[email protected]')
3839
assert i.indicator == '[email protected]'
3940

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

test/test_fqdn.py

+9
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
11
from csirtg_indicator import Indicator
22
from csirtg_indicator.utils import is_subdomain
3+
from faker import Faker
4+
fake = Faker()
5+
36

47
GOOD = [
58
'hdxturkceizle.xn--6frz82g',
@@ -61,3 +64,9 @@ def test_fqdn_subdomain():
6164

6265
for d in data:
6366
assert not Indicator(indicator=d).is_subdomain()
67+
68+
69+
def test_fqdn_random():
70+
for d in range(0, 100):
71+
assert Indicator(indicator=fake.domain_name()).itype == 'fqdn'
72+

test/test_ipv4.py

+7
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,9 @@
11
from csirtg_indicator import Indicator
22
from csirtg_indicator.exceptions import InvalidIndicator
33
from random import randint
4+
from faker import Faker
5+
fake = Faker()
6+
47

58
def _not(data):
69
for d in data:
@@ -77,3 +80,7 @@ def test_ipv4_padded():
7780
for k, v in d.items():
7881
assert Indicator(k).indicator == v
7982

83+
84+
def test_ipv4_random():
85+
for d in range(0, 100):
86+
assert Indicator(indicator=fake.ipv4()).itype == 'ipv4'

test/test_ipv6.py

+6
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
11
from csirtg_indicator import Indicator
22
from csirtg_indicator.exceptions import InvalidIndicator
3+
from faker import Faker
4+
fake = Faker()
5+
36

47
def _not(data):
58
for d in data:
@@ -38,3 +41,6 @@ def test_ipv6_nok():
3841
_not(data)
3942

4043

44+
def test_ipv6_random():
45+
for d in range(0, 100):
46+
assert Indicator(indicator=fake.ipv6()).itype == 'ipv6'

test/test_urls.py

+8
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,9 @@
22
from csirtg_indicator import Indicator
33
from csirtg_indicator.exceptions import InvalidIndicator
44
from csirtg_indicator.utils import url_to_fqdn
5+
from faker import Faker
6+
fake = Faker()
7+
58

69
GOOD = [
710
'http://58.147.128.10:81/val/1.html',
@@ -59,3 +62,8 @@ def test_urls_ok():
5962
def test_urls_fqdns():
6063
for g in GOOD:
6164
assert url_to_fqdn(g)
65+
66+
67+
def test_urls_random():
68+
for d in range(0, 100):
69+
assert Indicator(indicator=fake.uri()).itype == 'url'

0 commit comments

Comments
 (0)