test_service_discovery_client.cc revision 5c02ac1a9c1b504631c0a3d2b6e737b5d738bae1 
1// Copyright 2013 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "chrome/browser/local_discovery/test_service_discovery_client.h"
6
7#include "chrome/common/local_discovery/service_discovery_client_impl.h"
8#include "net/dns/mdns_client_impl.h"
9
10namespace local_discovery {
11
12TestServiceDiscoveryClient::TestServiceDiscoveryClient() {
13}
14
15TestServiceDiscoveryClient::~TestServiceDiscoveryClient() {
16}
17
18void TestServiceDiscoveryClient::Start() {
19  mdns_client_.reset(new net::MDnsClientImpl());
20  service_discovery_client_impl_.reset(new ServiceDiscoveryClientImpl(
21      mdns_client_.get()));
22  mdns_client_->StartListening(&mock_socket_factory_);
23
24  EXPECT_CALL(mock_socket_factory_, OnSendTo(testing::_))
25      .Times(testing::AnyNumber())
26      .WillRepeatedly(testing::Invoke(this,
27                                      &TestServiceDiscoveryClient::OnSendTo));
28}
29
30scoped_ptr<ServiceWatcher> TestServiceDiscoveryClient::CreateServiceWatcher(
31      const std::string& service_type,
32      const ServiceWatcher::UpdatedCallback& callback) {
33  return service_discovery_client_impl_->CreateServiceWatcher(service_type,
34                                                              callback);
35}
36
37scoped_ptr<ServiceResolver> TestServiceDiscoveryClient::CreateServiceResolver(
38      const std::string& service_name,
39      const ServiceResolver::ResolveCompleteCallback& callback) {
40  return service_discovery_client_impl_->CreateServiceResolver(service_name,
41                                                              callback);
42}
43
44scoped_ptr<LocalDomainResolver>
45TestServiceDiscoveryClient::CreateLocalDomainResolver(
46      const std::string& domain,
47      net::AddressFamily address_family,
48      const LocalDomainResolver::IPAddressCallback& callback) {
49  return service_discovery_client_impl_->CreateLocalDomainResolver(
50      domain, address_family, callback);
51}
52
53void TestServiceDiscoveryClient::SimulateReceive(const uint8* packet,
54                                                 int size) {
55  mock_socket_factory_.SimulateReceive(packet, size);
56}
57
58}  // namespace local_discovery
59