mirror of
https://github.com/ChronosX88/PyNesca.git
synced 2024-10-18 06:31:00 +00:00
28 lines
751 B
Python
28 lines
751 B
Python
from unittest import TestCase, main
|
|
from ipaddress import IPv4Address
|
|
from IpGenerator import IpGenerator
|
|
from Parser import Parser
|
|
|
|
|
|
class testIpGenerator(TestCase):
|
|
|
|
def setUp(self):
|
|
p = Parser()
|
|
self.ipgen = IpGenerator(
|
|
p.parse_address_field("192.168.1.1 - 192.168.1.10"), [80, 90])
|
|
|
|
def testIpGeneration(self):
|
|
'''self.assertEqual(
|
|
self.ipgen.get_next_address(None),
|
|
("192.168.1.1", 80))'''
|
|
previous_address = None
|
|
a = True
|
|
while previous_address or a:
|
|
previous_address=self.ipgen.get_next_address(previous_address)
|
|
a = False
|
|
self.assertEqual(self.ipgen.get_next_address(None), None)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|