DNS: Rewrite unit tests