Fixing DNS unit tests.

The mock dictionary that was being created did not properly represent the data being returned by the API. I suspect this was the cause of the invalid key error that this branch was created to fix.