def set_up(self): socket.getaddrinfo = socket_getaddrinfo requests.get = requests_get dns.resolver.query = dns_query with patch.object(etcd.Client, 'machines') as mock_machines: mock_machines.__get__ = Mock(return_value=[ 'http://localhost:2379', 'http://localhost:4001' ]) self.client = Client({'discovery_srv': 'test'}) self.client.http.request = http_request