mirror of
https://github.com/ChronosX88/PyNesca.git
synced 2024-10-18 14:41:00 +00:00
22 lines
529 B
Python
22 lines
529 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"), [80])
|
||
|
|
||
|
def testIpGeneration(self):
|
||
|
self.assertEquals(
|
||
|
self.ipgen.get_next_address(None),
|
||
|
(IPv4Address("192.168.1.1"), 80))
|
||
|
print(self.ipgen.get_next_address(None))
|
||
|
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
main()
|