class TestQtypeVSQnamelen(unittest.TestCase):
def reinit(self):
self.__stringbuffer1 = StringIO.StringIO()
self.__p1 = QtypeVSTld(self.__stringbuffer1)
def data_example(self):
data = PacketsExample()
for i in range(50):
data.add_packet({'flags': '0', 'queries': [
{'qname': 'www.nic.cl.', 'qtype': 1}]})
data.add_packet({'flags': '0', 'queries': [
{'qname': 'www.uchile.cl.', 'qtype': 2}]})
for i in range(50):
data.add_packet({'flags': '0', 'queries': [
{'qname': 'www.google.com.', 'qtype': 1}]})
data.add_packet({'flags': '0', 'queries': [
{'qname': 'www.facebook.com.', 'qtype': 2}]})
data.add_packet({'flags': '0', 'queries': [
{'qname': 'www.asdf.net.', 'qtype': 3}]})
data.add_packet({'flags': '0', 'queries': [
{'qname': 'www.qwerty.net.', 'qtype': 3}]})
data.set_expected(1, {'cl': 50, 'com': 50})
data.set_expected(2, {'cl': 50, 'com': 50})
data.set_expected(3, {'net': 2})
data.put_information('QtypeVSTld', [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('QtypeVSTld'):
self.assertTrue(qvst in result)
self.assertEquals(example.expected_value(qvst), result[qvst])
def reinit(self):
self.__stringbuffer1 = StringIO.StringIO()
self.__p1 = QtypeVSTld(self.__stringbuffer1)
