Index: remoting/host/process_stats_sender_unittest.cc |
diff --git a/remoting/host/process_stats_sender_unittest.cc b/remoting/host/process_stats_sender_unittest.cc |
deleted file mode 100644 |
index 6d08818c5343271c10557df58b0c051dd8bf6661..0000000000000000000000000000000000000000 |
--- a/remoting/host/process_stats_sender_unittest.cc |
+++ /dev/null |
@@ -1,180 +0,0 @@ |
-// Copyright 2017 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "remoting/host/process_stats_sender.h" |
- |
-#include <stdint.h> |
- |
-#include <utility> |
-#include <vector> |
- |
-#include "base/bind.h" |
-#include "base/location.h" |
-#include "base/message_loop/message_loop.h" |
-#include "base/run_loop.h" |
-#include "base/time/time.h" |
-#include "remoting/host/process_stats_agent.h" |
-#include "remoting/proto/process_stats.pb.h" |
-#include "remoting/protocol/process_stats_stub.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-namespace remoting { |
- |
-namespace { |
- |
-class FakeProcessStatsStub : public protocol::ProcessStatsStub { |
- public: |
- FakeProcessStatsStub() = default; |
- ~FakeProcessStatsStub() override = default; |
- |
- void OnProcessStats( |
- const protocol::AggregatedProcessResourceUsage& usage) override { |
- received_.push_back(usage); |
- DCHECK_LE(received_.size(), expected_usage_count_); |
- DCHECK(!quit_closure_.is_null()); |
- if (received_.size() == expected_usage_count_) { |
- quit_closure_.Run(); |
- } |
- } |
- |
- const std::vector<protocol::AggregatedProcessResourceUsage>& received() |
- const { |
- return received_; |
- } |
- |
- void set_quit_closure(base::Closure quit_closure) { |
- quit_closure_ = quit_closure; |
- } |
- |
- void set_expected_usage_count(size_t expected_usage_count) { |
- expected_usage_count_ = expected_usage_count; |
- } |
- |
- private: |
- std::vector<protocol::AggregatedProcessResourceUsage> received_; |
- size_t expected_usage_count_ = 0; |
- base::Closure quit_closure_; |
-}; |
- |
-class FakeProcessStatsAgent : public ProcessStatsAgent { |
- public: |
- FakeProcessStatsAgent() = default; |
- ~FakeProcessStatsAgent() override = default; |
- |
- protocol::ProcessResourceUsage GetResourceUsage() override { |
- protocol::ProcessResourceUsage usage; |
- usage.set_process_name("FakeProcessStatsAgent"); |
- usage.set_processor_usage(index_); |
- usage.set_working_set_size(index_); |
- usage.set_pagefile_size(index_); |
- index_++; |
- return usage; |
- } |
- |
- // Checks the expected usage based on index. |
- static void AssertExpected( |
- const protocol::AggregatedProcessResourceUsage& usage, |
- size_t index) { |
- ASSERT_EQ(usage.processor_usage(), index); |
- ASSERT_EQ(usage.working_set_size(), index); |
- ASSERT_EQ(usage.pagefile_size(), index); |
- } |
- |
- static void AssertExpected(const protocol::ProcessResourceUsage& usage, |
- size_t index) { |
- ASSERT_EQ(usage.processor_usage(), index); |
- ASSERT_EQ(usage.working_set_size(), index); |
- ASSERT_EQ(usage.pagefile_size(), index); |
- } |
- |
- size_t issued_times() const { return index_; } |
- |
- private: |
- size_t index_ = 0; |
-}; |
- |
-} // namespace |
- |
-TEST(ProcessStatsSenderTest, ReportUsage) { |
- base::MessageLoop message_loop; |
- base::RunLoop run_loop; |
- FakeProcessStatsStub stub; |
- std::unique_ptr<ProcessStatsSender> stats; |
- FakeProcessStatsAgent agent; |
- |
- stub.set_expected_usage_count(10); |
- stub.set_quit_closure(base::Bind( |
- [](std::unique_ptr<ProcessStatsSender>* stats, |
- const FakeProcessStatsStub& stub, const FakeProcessStatsAgent& agent, |
- base::RunLoop* run_loop) -> void { |
- ASSERT_EQ(stub.received().size(), agent.issued_times()); |
- stats->reset(); |
- run_loop->Quit(); |
- }, |
- base::Unretained(&stats), base::ConstRef(stub), base::ConstRef(agent), |
- base::Unretained(&run_loop))); |
- message_loop.task_runner()->PostTask( |
- FROM_HERE, |
- base::Bind( |
- [](std::unique_ptr<ProcessStatsSender>* stats, |
- FakeProcessStatsStub* stub, FakeProcessStatsAgent* agent) -> void { |
- stats->reset(new ProcessStatsSender( |
- stub, base::TimeDelta::FromMilliseconds(1), { agent })); |
- }, |
- base::Unretained(&stats), base::Unretained(&stub), |
- base::Unretained(&agent))); |
- run_loop.Run(); |
- |
- ASSERT_EQ(stub.received().size(), 10U); |
- for (size_t i = 0; i < stub.received().size(); i++) { |
- FakeProcessStatsAgent::AssertExpected(stub.received()[i], i); |
- } |
-} |
- |
-TEST(ProcessStatsSenderTest, MergeUsage) { |
- base::MessageLoop message_loop; |
- base::RunLoop run_loop; |
- FakeProcessStatsStub stub; |
- std::unique_ptr<ProcessStatsSender> stats; |
- // Owned by |stats|. |
- FakeProcessStatsAgent agent1; |
- FakeProcessStatsAgent agent2; |
- |
- stub.set_expected_usage_count(10); |
- stub.set_quit_closure(base::Bind( |
- [](std::unique_ptr<ProcessStatsSender>* stats, |
- const FakeProcessStatsStub& stub, const FakeProcessStatsAgent& agent1, |
- const FakeProcessStatsAgent& agent2, base::RunLoop* run_loop) -> void { |
- ASSERT_EQ(stub.received().size(), agent1.issued_times()); |
- ASSERT_EQ(stub.received().size(), agent2.issued_times()); |
- stats->reset(); |
- run_loop->Quit(); |
- }, |
- base::Unretained(&stats), base::ConstRef(stub), base::ConstRef(agent1), |
- base::ConstRef(agent2), base::Unretained(&run_loop))); |
- message_loop.task_runner()->PostTask( |
- FROM_HERE, |
- base::Bind( |
- [](std::unique_ptr<ProcessStatsSender>* stats, |
- FakeProcessStatsStub* stub, FakeProcessStatsAgent* agent1, |
- FakeProcessStatsAgent* agent2) -> void { |
- stats->reset(new ProcessStatsSender( |
- stub, base::TimeDelta::FromMilliseconds(1), |
- { agent1, agent2 } )); |
- }, |
- base::Unretained(&stats), base::Unretained(&stub), |
- base::Unretained(&agent1), base::Unretained(&agent2))); |
- run_loop.Run(); |
- |
- ASSERT_EQ(stub.received().size(), 10U); |
- for (size_t i = 0; i < stub.received().size(); i++) { |
- FakeProcessStatsAgent::AssertExpected(stub.received()[i], i * 2); |
- for (int j = 0; j < stub.received()[i].usages_size(); j++) { |
- FakeProcessStatsAgent::AssertExpected( |
- stub.received()[i].usages().Get(j), i); |
- } |
- } |
-} |
- |
-} // namespace remoting |