class TestRcode(unittest.TestCase):
def reinit(self):
self.__stringbuffer1 = StringIO.StringIO()
self.__p1 = ClientSubnet(self.__stringbuffer1)
def data_example(self):
data = PacketsExample()
for i in range(5):
data.add_packet({'dest': 'encrypted(dnsip1)',
'source': 'encrypted(ip1)',
'flags': '0',
'queries': [
{'qname': 'www.ni_c.cl', 'qtype': '1'}]})
data.set_expected("encrypted(ip1)", 5)
data.put_information("src", ["encrypted(ip1)"])
return data
def test_data_example(self):
self.reinit()
example = self.data_example()
for packet in example:
self.__p1(packet)
result = self.__p1.get_data()
for src in example.get_information('src'):
self.assertTrue(src in result)
self.assertEquals(example.expected_value(src), result[src])
def reinit(self):
self.__stringbuffer1 = StringIO.StringIO()
self.__p1 = ClientSubnet(self.__stringbuffer1)