summaryrefslogtreecommitdiffstats
path: root/third_party/libwebrtc/call/adaptation/broadcast_resource_listener_unittest.cc
blob: 9cd80500c2a67c52f21f92c4ab840f1564300834 (plain)
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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/*
 *  Copyright 2020 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.
 */

#include "call/adaptation/broadcast_resource_listener.h"

#include "call/adaptation/test/fake_resource.h"
#include "call/adaptation/test/mock_resource_listener.h"
#include "test/gmock.h"
#include "test/gtest.h"

namespace webrtc {

using ::testing::_;
using ::testing::StrictMock;

TEST(BroadcastResourceListenerTest, CreateAndRemoveAdapterResource) {
  rtc::scoped_refptr<FakeResource> source_resource =
      FakeResource::Create("SourceResource");
  BroadcastResourceListener broadcast_resource_listener(source_resource);
  broadcast_resource_listener.StartListening();

  EXPECT_TRUE(broadcast_resource_listener.GetAdapterResources().empty());
  rtc::scoped_refptr<Resource> adapter =
      broadcast_resource_listener.CreateAdapterResource();
  StrictMock<MockResourceListener> listener;
  adapter->SetResourceListener(&listener);
  EXPECT_EQ(std::vector<rtc::scoped_refptr<Resource>>{adapter},
            broadcast_resource_listener.GetAdapterResources());

  // The removed adapter is not referenced by the broadcaster.
  broadcast_resource_listener.RemoveAdapterResource(adapter);
  EXPECT_TRUE(broadcast_resource_listener.GetAdapterResources().empty());
  // The removed adapter is not forwarding measurements.
  EXPECT_CALL(listener, OnResourceUsageStateMeasured(_, _)).Times(0);
  source_resource->SetUsageState(ResourceUsageState::kOveruse);
  // Cleanup.
  adapter->SetResourceListener(nullptr);
  broadcast_resource_listener.StopListening();
}

TEST(BroadcastResourceListenerTest, AdapterNameIsBasedOnSourceResourceName) {
  rtc::scoped_refptr<FakeResource> source_resource =
      FakeResource::Create("FooBarResource");
  BroadcastResourceListener broadcast_resource_listener(source_resource);
  broadcast_resource_listener.StartListening();

  rtc::scoped_refptr<Resource> adapter =
      broadcast_resource_listener.CreateAdapterResource();
  EXPECT_EQ("FooBarResourceAdapter", adapter->Name());

  broadcast_resource_listener.RemoveAdapterResource(adapter);
  broadcast_resource_listener.StopListening();
}

TEST(BroadcastResourceListenerTest, AdaptersForwardsUsageMeasurements) {
  rtc::scoped_refptr<FakeResource> source_resource =
      FakeResource::Create("SourceResource");
  BroadcastResourceListener broadcast_resource_listener(source_resource);
  broadcast_resource_listener.StartListening();

  StrictMock<MockResourceListener> destination_listener1;
  StrictMock<MockResourceListener> destination_listener2;
  rtc::scoped_refptr<Resource> adapter1 =
      broadcast_resource_listener.CreateAdapterResource();
  adapter1->SetResourceListener(&destination_listener1);
  rtc::scoped_refptr<Resource> adapter2 =
      broadcast_resource_listener.CreateAdapterResource();
  adapter2->SetResourceListener(&destination_listener2);

  // Expect kOveruse to be echoed.
  EXPECT_CALL(destination_listener1, OnResourceUsageStateMeasured(_, _))
      .Times(1)
      .WillOnce([adapter1](rtc::scoped_refptr<Resource> resource,
                           ResourceUsageState usage_state) {
        EXPECT_EQ(adapter1, resource);
        EXPECT_EQ(ResourceUsageState::kOveruse, usage_state);
      });
  EXPECT_CALL(destination_listener2, OnResourceUsageStateMeasured(_, _))
      .Times(1)
      .WillOnce([adapter2](rtc::scoped_refptr<Resource> resource,
                           ResourceUsageState usage_state) {
        EXPECT_EQ(adapter2, resource);
        EXPECT_EQ(ResourceUsageState::kOveruse, usage_state);
      });
  source_resource->SetUsageState(ResourceUsageState::kOveruse);

  // Expect kUnderuse to be echoed.
  EXPECT_CALL(destination_listener1, OnResourceUsageStateMeasured(_, _))
      .Times(1)
      .WillOnce([adapter1](rtc::scoped_refptr<Resource> resource,
                           ResourceUsageState usage_state) {
        EXPECT_EQ(adapter1, resource);
        EXPECT_EQ(ResourceUsageState::kUnderuse, usage_state);
      });
  EXPECT_CALL(destination_listener2, OnResourceUsageStateMeasured(_, _))
      .Times(1)
      .WillOnce([adapter2](rtc::scoped_refptr<Resource> resource,
                           ResourceUsageState usage_state) {
        EXPECT_EQ(adapter2, resource);
        EXPECT_EQ(ResourceUsageState::kUnderuse, usage_state);
      });
  source_resource->SetUsageState(ResourceUsageState::kUnderuse);

  // Adapters have to be unregistered before they or the broadcaster is
  // destroyed, ensuring safe use of raw pointers.
  adapter1->SetResourceListener(nullptr);
  adapter2->SetResourceListener(nullptr);

  broadcast_resource_listener.RemoveAdapterResource(adapter1);
  broadcast_resource_listener.RemoveAdapterResource(adapter2);
  broadcast_resource_listener.StopListening();
}

}  // namespace webrtc