1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_MEDIA_ROUTER_DISCOVERY_ACCESS_CODE_ACCESS_CODE_TEST_UTIL_H_
#define CHROME_BROWSER_MEDIA_ROUTER_DISCOVERY_ACCESS_CODE_ACCESS_CODE_TEST_UTIL_H_
#include "chrome/browser/media/router/discovery/access_code/access_code_cast_pref_updater.h"
#include "chrome/browser/media/router/discovery/access_code/access_code_cast_sink_service.h"
#include "chrome/browser/media/router/discovery/access_code/discovery_resources.pb.h"
#include "chrome/browser/media/router/discovery/mdns/media_sink_util.h"
#include "components/media_router/common/discovery/media_sink_internal.h"
#include "testing/gmock/include/gmock/gmock.h"
namespace media_router {
inline constexpr char kExpectedDisplayName[] = "test_device";
inline constexpr char kExpectedSinkId[] = "1234";
inline constexpr char kExpectedPort[] = "666";
inline constexpr char kExpectedIpV4[] = "192.0.2.146";
inline constexpr char kExpectedIpV6[] =
"2001:0db8:85a3:0000:0000:8a2e:0370:7334";
using DiscoveryDevice = chrome_browser_media::proto::DiscoveryDevice;
using NetworkInfo = chrome_browser_media::proto::NetworkInfo;
// This function is used for testing across the access_code directory
DiscoveryDevice BuildDiscoveryDeviceProto(
const char* display_name = kExpectedDisplayName,
const char* sink_id = kExpectedSinkId,
const char* port = kExpectedPort,
const char* ip_v4 = kExpectedIpV4,
const char* ip_v6 = kExpectedIpV6,
bool set_device_capabilities = true,
bool set_network_info = true);
// Mock Access Code Cast Sink Service class. Used for testing purposes.
class MockAccessCodeCastSinkService : public AccessCodeCastSinkService {
public:
MockAccessCodeCastSinkService(
Profile* profile,
MediaRouter* media_router,
CastMediaSinkServiceImpl* cast_media_sink_service_impl,
DiscoveryNetworkMonitor* network_monitor);
~MockAccessCodeCastSinkService() override;
// This method can be passed into
// AccessCodeCastSinkServiceFactory::SetTestingFactory() to
// make the factory return a MockAccessCodeCastSinkService.
static std::unique_ptr<KeyedService> Create(content::BrowserContext* context);
MOCK_METHOD(void,
AddSinkToMediaRouter,
(const MediaSinkInternal& sink,
AddSinkResultCallback add_sink_callback),
(override));
MOCK_METHOD(void,
DiscoverSink,
(const std::string& access_code, AddSinkResultCallback callback),
(override));
};
class MockAccessCodeCastPrefUpdater : public AccessCodeCastPrefUpdater {
public:
MockAccessCodeCastPrefUpdater();
~MockAccessCodeCastPrefUpdater() override;
void UpdateDevicesDict(const MediaSinkInternal& sink,
base::OnceClosure on_updated_callback) override;
void UpdateDeviceAddedTimeDict(
const MediaSink::Id sink_id,
base::OnceClosure on_updated_callback) override;
void GetDevicesDict(base::OnceCallback<void(base::Value::Dict)>
get_devices_callback) override;
void GetDeviceAddedTimeDict(base::OnceCallback<void(base::Value::Dict)>
get_device_added_time_callback) override;
void RemoveSinkIdFromDevicesDict(
const MediaSink::Id sink_id,
base::OnceClosure on_sink_removed_callback) override;
void RemoveSinkIdFromDeviceAddedTimeDict(
const MediaSink::Id sink_id,
base::OnceClosure on_sink_removed_callback) override;
void ClearDevicesDict(base::OnceClosure on_cleared_callback) override;
void ClearDeviceAddedTimeDict(base::OnceClosure on_cleared_callback) override;
MOCK_METHOD(void,
UpdateDevicesDictForTesting,
(const MediaSinkInternal& sink));
void set_devices_dict(base::Value::Dict dict);
void set_device_added_time_dict(base::Value::Dict dict);
const base::Value::Dict& devices_dict() { return devices_dict_; }
const base::Value::Dict& device_added_time_dict() {
return device_added_time_dict_;
}
private:
base::Value::Dict devices_dict_;
base::Value::Dict device_added_time_dict_;
};
MediaRoute CreateRouteForTesting(const MediaSink::Id& sink_id);
} // namespace media_router
#endif // CHROME_BROWSER_MEDIA_ROUTER_DISCOVERY_ACCESS_CODE_ACCESS_CODE_TEST_UTIL_H_
|
