theHarvester/tests/myparser_test.py
NotoriousRebel cc7b0d0c51 stashing.
2018-12-29 21:29:25 -05:00

18 lines
452 B
Python

#
# Unit tests for myparser.py
#
from parsers import myparser
import unittest
class TestMyParser(unittest.TestCase):
def test_emails(self):
word = 'domain.com'
results = '@domain.com***a@domain***banotherdomain.com***c@domain.com***d@sub.domain.com***'
p = myparser.parser(results, word)
emails = sorted(p.emails())
self.assertEqual(emails, [ 'c@domain.com', 'd@sub.domain.com' ])
if __name__ == '__main__':
unittest.main()