Add tests for NetworkDiscovery