class TestIdnVSTld(unittest.TestCase):
def reinit(self):
self.__stringbuffer1 = StringIO.StringIO()
self.__p1 = IdnVSTld(self.__stringbuffer1)
def data_example(self):
data = PacketsExample()
for i in range(3):
data.add_packet({"flags": "0", "queries": [{"qname": u"www.nic.xn--cm-5ja."}]})
data.add_packet({"flags": "0", "queries": [{"qname": u"www.uchile.cl."}]})
data.add_packet(
{
"flags": "0",
"queries": [{"qname": u"data-agkn-com-1198526359." + u"us-east-1.elb.amazonaws.com.imunimacu.cl."}],
}
)
data.add_packet(
{
"flags": "0",
"queries": [
{"qname": u"data-agkn-com-1198526359.us-east-1" + u".elb.amazonaws.com.xn--imunimacu-t6a.cl."}
],
}
)
data.set_expected(unicode("xn--cm-5ja", "utf8").encode("idna"), 3)
data.set_expected(unicode("cl", "utf8").encode("idna"), 3)
data.put_information("IdnVSTld", [unicode("cóm", "utf8").encode("idna"), unicode("cl", "utf8").encode("idna")])
return data
def test_data_example(self):
self.reinit()
example = self.data_example()
for packet in example:
# print packet
self.__p1(packet)
result = self.__p1.get_data()
for qvst in example.get_information("IdnVSTld"):
self.assertTrue(qvst in result)
self.assertEquals(example.expected_value(qvst), result[qvst])
def reinit(self):
self.__stringbuffer1 = StringIO.StringIO()
self.__p1 = IdnVSTld(self.__stringbuffer1)