Add unit tests for dns client implementation