test_chinadns.py 文件源码

def test_run(udp_server_process):
    addr1, addr2 = udp_server_process
    forward_addr = ("127.0.0.1", 42353)
    argv = ["-p", "%s:%d" % (forward_addr[0], forward_addr[1]),
            "-r", "chinadns",
            "-u", "%s:%d,%s:%d" %
            (addr1[0], addr1[1], addr2[0], addr2[1]),
            "-l", "debug",
            "-f", "%s/chnroute_test.txt" % (mydir),
            "-b", "%s/iplist_test.txt" % (mydir),
            "--cache"]
    p = Process(target=chinadns.run, args=(argv,))
    p.start()
    time.sleep(0.5)
    client = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    q = dnslib.DNSRecord.question(qname)
    client.sendto(bytes(q.pack()), forward_addr)
    data, _ = client.recvfrom(1024)
    d = dnslib.DNSRecord.parse(data)
    assert str(d.rr[0].rdata) == "101.226.103.106"
    client.close()
    os.kill(p.pid, signal.SIGINT)
    p.join()