Darin Petkov | f0136cd | 2012-11-07 16:18:02 +0100 | [diff] [blame] | 1 | // Copyright (c) 2012 The Chromium OS 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 "shill/diagnostics_reporter.h" |
| 6 | |
Ben Chan | 11c213f | 2014-09-05 08:21:06 -0700 | [diff] [blame] | 7 | #include <base/files/file_util.h> |
Paul Stewart | 5ad1606 | 2013-02-21 18:10:48 -0800 | [diff] [blame] | 8 | #include <base/files/scoped_temp_dir.h> |
Utkarsh Sanghi | 83bd64b | 2014-07-29 16:01:43 -0700 | [diff] [blame] | 9 | #include <chromeos/minijail/mock_minijail.h> |
Darin Petkov | f0136cd | 2012-11-07 16:18:02 +0100 | [diff] [blame] | 10 | #include <gmock/gmock.h> |
| 11 | #include <gtest/gtest.h> |
| 12 | |
Darin Petkov | 3806676 | 2012-12-17 15:35:45 +0100 | [diff] [blame] | 13 | #include "shill/mock_process_killer.h" |
Peter Qiu | 8d6b597 | 2014-10-28 15:33:34 -0700 | [diff] [blame] | 14 | #include "shill/net/mock_time.h" |
Darin Petkov | f0136cd | 2012-11-07 16:18:02 +0100 | [diff] [blame] | 15 | |
Albert Chaulk | 0e1cdea | 2013-02-27 15:32:55 -0800 | [diff] [blame] | 16 | using base::FilePath; |
Utkarsh Sanghi | 83bd64b | 2014-07-29 16:01:43 -0700 | [diff] [blame] | 17 | using chromeos::MockMinijail; |
Darin Petkov | bd0dcc8 | 2012-11-29 10:51:12 +0100 | [diff] [blame] | 18 | using testing::_; |
Darin Petkov | 3806676 | 2012-12-17 15:35:45 +0100 | [diff] [blame] | 19 | using testing::ElementsAre; |
Darin Petkov | fac01f0 | 2012-12-06 12:44:07 +0100 | [diff] [blame] | 20 | using testing::InSequence; |
Darin Petkov | 3806676 | 2012-12-17 15:35:45 +0100 | [diff] [blame] | 21 | using testing::IsNull; |
Darin Petkov | f0136cd | 2012-11-07 16:18:02 +0100 | [diff] [blame] | 22 | using testing::Return; |
Darin Petkov | 328e19d | 2012-12-04 15:54:07 +0100 | [diff] [blame] | 23 | using testing::SetArgumentPointee; |
Darin Petkov | 3806676 | 2012-12-17 15:35:45 +0100 | [diff] [blame] | 24 | using testing::StrEq; |
Darin Petkov | f0136cd | 2012-11-07 16:18:02 +0100 | [diff] [blame] | 25 | |
| 26 | namespace shill { |
| 27 | |
Darin Petkov | f0136cd | 2012-11-07 16:18:02 +0100 | [diff] [blame] | 28 | namespace { |
| 29 | |
| 30 | class ReporterUnderTest : public DiagnosticsReporter { |
| 31 | public: |
| 32 | ReporterUnderTest() {} |
| 33 | |
| 34 | MOCK_METHOD0(IsReportingEnabled, bool()); |
| 35 | |
| 36 | private: |
| 37 | DISALLOW_COPY_AND_ASSIGN(ReporterUnderTest); |
| 38 | }; |
| 39 | |
| 40 | } // namespace |
| 41 | |
Darin Petkov | 328e19d | 2012-12-04 15:54:07 +0100 | [diff] [blame] | 42 | class DiagnosticsReporterTest : public testing::Test { |
| 43 | public: |
| 44 | DiagnosticsReporterTest() { |
Darin Petkov | 3806676 | 2012-12-17 15:35:45 +0100 | [diff] [blame] | 45 | reporter_.minijail_ = &minijail_; |
| 46 | reporter_.process_killer_ = &process_killer_; |
Darin Petkov | 328e19d | 2012-12-04 15:54:07 +0100 | [diff] [blame] | 47 | reporter_.time_ = &time_; |
Darin Petkov | 328e19d | 2012-12-04 15:54:07 +0100 | [diff] [blame] | 48 | } |
| 49 | |
| 50 | protected: |
| 51 | bool IsReportingEnabled() { |
| 52 | return DiagnosticsReporter::GetInstance()->IsReportingEnabled(); |
| 53 | } |
| 54 | |
Ben Chan | 7fab897 | 2014-08-10 17:14:46 -0700 | [diff] [blame] | 55 | void SetLastLogStash(uint64_t time) { reporter_.last_log_stash_ = time; } |
| 56 | uint64_t GetLastLogStash() { return reporter_.last_log_stash_; } |
| 57 | uint64_t GetLogStashThrottleSeconds() { |
Darin Petkov | 328e19d | 2012-12-04 15:54:07 +0100 | [diff] [blame] | 58 | return DiagnosticsReporter::kLogStashThrottleSeconds; |
| 59 | } |
| 60 | |
Paul Stewart | 3b30ca5 | 2015-06-16 13:13:10 -0700 | [diff] [blame] | 61 | void SetStashedNetLog(const FilePath& stashed_net_log) { |
Darin Petkov | 3806676 | 2012-12-17 15:35:45 +0100 | [diff] [blame] | 62 | reporter_.stashed_net_log_ = stashed_net_log; |
| 63 | } |
| 64 | |
| 65 | MockMinijail minijail_; |
| 66 | MockProcessKiller process_killer_; |
Darin Petkov | 328e19d | 2012-12-04 15:54:07 +0100 | [diff] [blame] | 67 | MockTime time_; |
| 68 | ReporterUnderTest reporter_; |
| 69 | }; |
| 70 | |
Darin Petkov | f0136cd | 2012-11-07 16:18:02 +0100 | [diff] [blame] | 71 | TEST_F(DiagnosticsReporterTest, IsReportingEnabled) { |
| 72 | EXPECT_FALSE(IsReportingEnabled()); |
| 73 | } |
| 74 | |
Darin Petkov | 328e19d | 2012-12-04 15:54:07 +0100 | [diff] [blame] | 75 | TEST_F(DiagnosticsReporterTest, OnConnectivityEventThrottle) { |
Alex Vakulenko | 8a53229 | 2014-06-16 17:18:44 -0700 | [diff] [blame] | 76 | using timeval_seconds_t = decltype(timeval::tv_sec); |
| 77 | const timeval_seconds_t kLastStash = 50; |
| 78 | auto kNow = kLastStash + |
| 79 | static_cast<timeval_seconds_t>(GetLogStashThrottleSeconds() - 1); |
Darin Petkov | 328e19d | 2012-12-04 15:54:07 +0100 | [diff] [blame] | 80 | SetLastLogStash(kLastStash); |
Alex Vakulenko | 8a53229 | 2014-06-16 17:18:44 -0700 | [diff] [blame] | 81 | const struct timeval now = {kNow, 0}; |
Darin Petkov | 328e19d | 2012-12-04 15:54:07 +0100 | [diff] [blame] | 82 | EXPECT_CALL(time_, GetTimeMonotonic(_)) |
| 83 | .WillOnce(DoAll(SetArgumentPointee<0>(now), Return(0))); |
| 84 | reporter_.OnConnectivityEvent(); |
| 85 | EXPECT_EQ(kLastStash, GetLastLogStash()); |
| 86 | } |
| 87 | |
| 88 | TEST_F(DiagnosticsReporterTest, OnConnectivityEvent) { |
Alex Vakulenko | 8a53229 | 2014-06-16 17:18:44 -0700 | [diff] [blame] | 89 | using timeval_seconds_t = decltype(timeval::tv_sec); |
Ben Chan | 7fab897 | 2014-08-10 17:14:46 -0700 | [diff] [blame] | 90 | const uint64_t kInitStash = 0; |
Darin Petkov | 19321e4 | 2012-12-17 15:40:59 +0100 | [diff] [blame] | 91 | SetLastLogStash(kInitStash); |
| 92 | // Test that the initial call is not throttled. |
Alex Vakulenko | 8a53229 | 2014-06-16 17:18:44 -0700 | [diff] [blame] | 93 | auto kNow0 = static_cast<timeval_seconds_t>(kInitStash + 1); |
| 94 | const struct timeval now0 = {kNow0, 0}; |
| 95 | auto kNow1 = kNow0 + |
| 96 | static_cast<timeval_seconds_t>(GetLogStashThrottleSeconds() + 1); |
| 97 | const struct timeval now1 = {kNow1, 0}; |
Paul Stewart | 5ad1606 | 2013-02-21 18:10:48 -0800 | [diff] [blame] | 98 | base::ScopedTempDir temp_dir_; |
Darin Petkov | 3806676 | 2012-12-17 15:35:45 +0100 | [diff] [blame] | 99 | ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| 100 | FilePath stashed_net_log = temp_dir_.path().Append("stashed-net-log"); |
Ben Chan | 6fbf64f | 2014-05-21 18:07:01 -0700 | [diff] [blame] | 101 | EXPECT_EQ(0, base::WriteFile(stashed_net_log, "", 0)); |
Ben Chan | a0ddf46 | 2014-02-06 11:32:42 -0800 | [diff] [blame] | 102 | EXPECT_TRUE(base::PathExists(stashed_net_log)); |
Darin Petkov | 3806676 | 2012-12-17 15:35:45 +0100 | [diff] [blame] | 103 | SetStashedNetLog(stashed_net_log); |
Darin Petkov | 328e19d | 2012-12-04 15:54:07 +0100 | [diff] [blame] | 104 | EXPECT_CALL(time_, GetTimeMonotonic(_)) |
Darin Petkov | fac01f0 | 2012-12-06 12:44:07 +0100 | [diff] [blame] | 105 | .WillOnce(DoAll(SetArgumentPointee<0>(now0), Return(0))) |
| 106 | .WillOnce(DoAll(SetArgumentPointee<0>(now1), Return(0))); |
| 107 | EXPECT_CALL(reporter_, IsReportingEnabled()) |
| 108 | .WillOnce(Return(false)) |
| 109 | .WillOnce(Return(true)); |
Darin Petkov | 3806676 | 2012-12-17 15:35:45 +0100 | [diff] [blame] | 110 | EXPECT_CALL(minijail_, New()).Times(2); |
Utkarsh Sanghi | e4c6aff | 2014-07-30 14:49:03 -0700 | [diff] [blame] | 111 | EXPECT_CALL(minijail_, DropRoot(_, StrEq("syslog"), |
| 112 | StrEq("syslog"))).Times(2); |
Darin Petkov | 3806676 | 2012-12-17 15:35:45 +0100 | [diff] [blame] | 113 | const pid_t kPID = 123; |
| 114 | { |
| 115 | InSequence s; |
| 116 | EXPECT_CALL(minijail_, |
| 117 | RunAndDestroy(_, ElementsAre(_, IsNull()), _)) |
| 118 | .WillOnce(DoAll(SetArgumentPointee<2>(kPID), Return(true))); |
| 119 | EXPECT_CALL( |
| 120 | minijail_, |
| 121 | RunAndDestroy(_, ElementsAre(_, StrEq("--upload"), IsNull()), _)) |
| 122 | .WillOnce(Return(false)); |
| 123 | } |
| 124 | EXPECT_CALL(process_killer_, Wait(kPID, _)).Times(1); |
Darin Petkov | 328e19d | 2012-12-04 15:54:07 +0100 | [diff] [blame] | 125 | reporter_.OnConnectivityEvent(); |
Darin Petkov | fac01f0 | 2012-12-06 12:44:07 +0100 | [diff] [blame] | 126 | EXPECT_EQ(kNow0, GetLastLogStash()); |
Ben Chan | a0ddf46 | 2014-02-06 11:32:42 -0800 | [diff] [blame] | 127 | EXPECT_FALSE(base::PathExists(stashed_net_log)); |
Darin Petkov | fac01f0 | 2012-12-06 12:44:07 +0100 | [diff] [blame] | 128 | reporter_.OnConnectivityEvent(); |
| 129 | EXPECT_EQ(kNow1, GetLastLogStash()); |
Darin Petkov | 328e19d | 2012-12-04 15:54:07 +0100 | [diff] [blame] | 130 | } |
| 131 | |
Darin Petkov | f0136cd | 2012-11-07 16:18:02 +0100 | [diff] [blame] | 132 | } // namespace shill |