class TestQtype(unittest.TestCase):
def reinit(self):
self.__stringbuffer1 = StringIO.StringIO()
self.__p1 = Qtype(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("1", 5)
data.put_information("Qtype", ["1"])
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 qtype in example.get_information("Qtype"):
self.assertTrue(qtype in result)
self.assertEquals(example.expected_value(qtype), result[qtype])
def reinit(self):
self.__stringbuffer1 = StringIO.StringIO()
self.__p1 = Qtype(self.__stringbuffer1)
