Start to add unit tests for dns impl