35 lines
911 B
Python
35 lines
911 B
Python
import unittest
|
|
from text import TextReader, GZMuleReader, PGZip
|
|
|
|
|
|
class ImportTests(unittest.TestCase):
|
|
|
|
def testpgtext(self):
|
|
fr = TextReader("pg.txt")
|
|
ips = fr.next()
|
|
self.assertEqual("3.0.0.0", ips[0])
|
|
self.assertEqual("3.255.255.255", ips[1])
|
|
|
|
def testMule(self):
|
|
fr = GZMuleReader("nipfilter.dat.gz")
|
|
ips = fr.next()
|
|
self.assertEqual("0.0.0.0", ips[0])
|
|
self.assertEqual("3.255.255.255", ips[1])
|
|
|
|
def testZip(self):
|
|
fr = PGZip("splist.zip")
|
|
ips = fr.next()
|
|
print "wibble wibble",ips
|
|
self.assertEqual("1.1.1.1", ips[0])
|
|
self.assertEqual("3.255.255.255", ips[1])
|
|
|
|
ips = fr.next()
|
|
self.assertEqual("0.0.0.0", ips[0])
|
|
self.assertEqual("3.255.255.255", ips[1])
|
|
|
|
ips = fr.next()
|
|
self.assertEqual(ips, False)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|