mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 06:10:40 +01:00

The input SocketAddress for STUN host lookup is constructed with just the hostname, so the family is AF_UNSPEC. So added an overload with a target family to distinguish this from the family of the input addr. Bug: webrtc:14319, webrtc:14131 Change-Id: Ia5ac5aa2e894e0c4dfb4417e3e8a76a6cec3ea71 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/270624 Reviewed-by: Tomas Gunnarsson <tommi@webrtc.org> Commit-Queue: Sameer Vijaykar <samvi@google.com> Reviewed-by: Jonas Oreland <jonaso@webrtc.org> Reviewed-by: Jonas Oreland <jonaso@google.com> Cr-Commit-Position: refs/heads/main@{#37750}
62 lines
1.9 KiB
C++
62 lines
1.9 KiB
C++
/*
|
|
* Copyright 2021 The WebRTC Project Authors. All rights reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef API_TEST_MOCK_ASYNC_DNS_RESOLVER_H_
|
|
#define API_TEST_MOCK_ASYNC_DNS_RESOLVER_H_
|
|
|
|
#include <functional>
|
|
#include <memory>
|
|
|
|
#include "api/async_dns_resolver.h"
|
|
#include "test/gmock.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class MockAsyncDnsResolverResult : public AsyncDnsResolverResult {
|
|
public:
|
|
MOCK_METHOD(bool,
|
|
GetResolvedAddress,
|
|
(int, rtc::SocketAddress*),
|
|
(const, override));
|
|
MOCK_METHOD(int, GetError, (), (const, override));
|
|
};
|
|
|
|
class MockAsyncDnsResolver : public AsyncDnsResolverInterface {
|
|
public:
|
|
MOCK_METHOD(void,
|
|
Start,
|
|
(const rtc::SocketAddress&, std::function<void()>),
|
|
(override));
|
|
MOCK_METHOD(void,
|
|
Start,
|
|
(const rtc::SocketAddress&, int family, std::function<void()>),
|
|
(override));
|
|
MOCK_METHOD(AsyncDnsResolverResult&, result, (), (const, override));
|
|
};
|
|
|
|
class MockAsyncDnsResolverFactory : public AsyncDnsResolverFactoryInterface {
|
|
public:
|
|
MOCK_METHOD(std::unique_ptr<webrtc::AsyncDnsResolverInterface>,
|
|
CreateAndResolve,
|
|
(const rtc::SocketAddress&, std::function<void()>),
|
|
(override));
|
|
MOCK_METHOD(std::unique_ptr<webrtc::AsyncDnsResolverInterface>,
|
|
CreateAndResolve,
|
|
(const rtc::SocketAddress&, int, std::function<void()>),
|
|
(override));
|
|
MOCK_METHOD(std::unique_ptr<webrtc::AsyncDnsResolverInterface>,
|
|
Create,
|
|
(),
|
|
(override));
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // API_TEST_MOCK_ASYNC_DNS_RESOLVER_H_
|