Improving DNS tests

- using mock for dig
- tests for remove_records