class TestChaosTypesAndNames(unittest.TestCase):
def reinit(self):
self.__stringbuffer1 = StringIO.StringIO()
self.__p1 = ChaosTypesAndNames(self.__stringbuffer1)
def data_example(self):
data = PacketsExample()
for i in range(30):
data.add_packet({"flags": "0", "queries": [{"qname": "www.nic.cl.", "qtype": 1}]})
data.add_packet({"flags": "0", "queries": [{"qname": "www.nic.cl.", "qtype": 2}]})
for i in range(25):
data.add_packet({"flags": "0", "queries": [{"qname": "www.uchile.cl.", "qtype": 3}]})
data.set_expected(1, {"www.nic.cl.": 30})
data.set_expected(2, {"www.nic.cl.": 30})
data.set_expected(3, {"www.uchile.cl.": 25})
data.put_information("ChaosTypesAndNames", [1, 2, 3])
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 qvst in example.get_information("ChaosTypesAndNames"):
self.assertTrue(qvst in result)
self.assertEquals(example.expected_value(qvst), result[qvst])
def reinit(self):
self.__stringbuffer1 = StringIO.StringIO()
self.__p1 = ChaosTypesAndNames(self.__stringbuffer1)