9a5efe925d
Bug: webrtc:11564 Change-Id: I81d06041b80ce470e4859c4d0ebad7ff0f831af9 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/175134 Reviewed-by: Björn Terelius <terelius@webrtc.org> Commit-Queue: Danil Chapovalov <danilchap@webrtc.org> Cr-Commit-Position: refs/heads/master@{#31276}
81 lines
2.6 KiB
C++
81 lines
2.6 KiB
C++
/*
|
|
* 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/resource.h"
|
|
|
|
#include <memory>
|
|
|
|
#include "api/scoped_refptr.h"
|
|
#include "call/adaptation/test/fake_resource.h"
|
|
#include "rtc_base/event.h"
|
|
#include "rtc_base/task_queue_for_test.h"
|
|
#include "test/gmock.h"
|
|
#include "test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
|
|
using ::testing::_;
|
|
using ::testing::StrictMock;
|
|
|
|
class MockResourceListener : public ResourceListener {
|
|
public:
|
|
MOCK_METHOD(void,
|
|
OnResourceUsageStateMeasured,
|
|
(rtc::scoped_refptr<Resource> resource),
|
|
(override));
|
|
};
|
|
|
|
class ResourceTest : public ::testing::Test {
|
|
public:
|
|
ResourceTest()
|
|
: resource_adaptation_queue_("ResourceAdaptationQueue"),
|
|
encoder_queue_("EncoderQueue"),
|
|
fake_resource_(new FakeResource("FakeResource")) {
|
|
fake_resource_->Initialize(&encoder_queue_, &resource_adaptation_queue_);
|
|
}
|
|
|
|
protected:
|
|
const std::unique_ptr<TaskQueueFactory> task_queue_factory_;
|
|
TaskQueueForTest resource_adaptation_queue_;
|
|
TaskQueueForTest encoder_queue_;
|
|
rtc::scoped_refptr<FakeResource> fake_resource_;
|
|
};
|
|
|
|
TEST_F(ResourceTest, RegisteringListenerReceivesCallbacks) {
|
|
resource_adaptation_queue_.SendTask(
|
|
[this] {
|
|
StrictMock<MockResourceListener> resource_listener;
|
|
fake_resource_->SetResourceListener(&resource_listener);
|
|
EXPECT_CALL(resource_listener, OnResourceUsageStateMeasured(_))
|
|
.Times(1)
|
|
.WillOnce([](rtc::scoped_refptr<Resource> resource) {
|
|
EXPECT_EQ(ResourceUsageState::kOveruse, resource->usage_state());
|
|
});
|
|
fake_resource_->set_usage_state(ResourceUsageState::kOveruse);
|
|
fake_resource_->SetResourceListener(nullptr);
|
|
},
|
|
RTC_FROM_HERE);
|
|
}
|
|
|
|
TEST_F(ResourceTest, UnregisteringListenerStopsCallbacks) {
|
|
resource_adaptation_queue_.SendTask(
|
|
[this] {
|
|
StrictMock<MockResourceListener> resource_listener;
|
|
fake_resource_->SetResourceListener(&resource_listener);
|
|
fake_resource_->SetResourceListener(nullptr);
|
|
EXPECT_CALL(resource_listener, OnResourceUsageStateMeasured(_))
|
|
.Times(0);
|
|
fake_resource_->set_usage_state(ResourceUsageState::kOveruse);
|
|
},
|
|
RTC_FROM_HERE);
|
|
}
|
|
|
|
} // namespace webrtc
|