summaryrefslogtreecommitdiff
path: root/util/generate-mx-pickle.py
blob: a917e0892ba41c168571affeb201dde999b307b1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import pickle
import sys

import dns.rdata
import dns.version

# Generate a pickled mx RR for the current dnspython version

mx = dns.rdata.from_text("in", "mx", "10 mx.example.")
filename = f"pickled-{dns.version.MAJOR}-{dns.version.MINOR}.pickle"
with open(filename, "wb") as f:
    pickle.dump(mx, f)
with open(filename, "rb") as f:
    mx2 = pickle.load(f)
if mx == mx2:
    print("ok")
else:
    print("DIFFERENT!")
    sys.exit(1)