Add dns discovery to mock