src/content/browser/power_profiler/power_profiler_service_unittest.cc

   1 // Copyright 2014 The Chromium 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 "base/message_loop/message_loop.h"
   6 #include "base/run_loop.h"
   7 #include "content/browser/browser_thread_impl.h"
   8 #include "content/browser/power_profiler/power_profiler_service.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 namespace content {
  12
  13 namespace {
  14
  15 const int kNumEvents = 3;
  16 const int kDefaultSamplePeriodMs = 50;
  17
  18 // Provide a set number of power events.
  19 class TestPowerDataProvider : public PowerDataProvider {
  20  public:
  21   TestPowerDataProvider(int count) : num_events_to_send_(count) {}
  22   virtual ~TestPowerDataProvider() {}
  23
  24   virtual PowerEventVector GetData() OVERRIDE {
  25     PowerEventVector events;
  26     if (num_events_to_send_ == 0)
  27       return events;
  28
  29     PowerEvent event;
  30     event.type = PowerEvent::SOC_PACKAGE;
  31     event.time = base::TimeTicks::Now();
  32     event.value = 1.0;
  33     events.push_back(event);
  34
  35     num_events_to_send_--;
  36     return events;
  37   }
  38
  39   virtual base::TimeDelta GetSamplingRate() OVERRIDE {
  40     return base::TimeDelta::FromMilliseconds(kDefaultSamplePeriodMs);
  41   }
  42
  43  private:
  44   int num_events_to_send_;
  45   DISALLOW_COPY_AND_ASSIGN(TestPowerDataProvider);
  46 };
  47
  48 class TestPowerProfilerObserver : public PowerProfilerObserver {
  49  public:
  50   TestPowerProfilerObserver()
  51       : valid_event_count_(0),
  52         total_num_events_received_(0) {}
  53   virtual ~TestPowerProfilerObserver() {}
  54
  55   virtual void OnPowerEvent(const PowerEventVector& events) OVERRIDE {
  56     if (IsValidEvent(events[0]))
  57       ++valid_event_count_;
  58
  59     total_num_events_received_++;
  60     if (total_num_events_received_ >= kNumEvents) {
  61       // All expected events received, exiting.
  62       quit_closure_.Run();
  63     }
  64   }
  65
  66   int valid_event_count() const { return valid_event_count_; }
  67   void set_quit_closure(base::Closure closure) { quit_closure_ = closure; }
  68
  69  private:
  70   bool IsValidEvent(const PowerEvent& event) {
  71     return event.type == PowerEvent::SOC_PACKAGE &&
  72            !event.time.is_null() &&
  73            event.value > 0;
  74   }
  75
  76   int valid_event_count_;
  77   int total_num_events_received_;
  78   base::Closure quit_closure_;
  79
  80   DISALLOW_COPY_AND_ASSIGN(TestPowerProfilerObserver);
  81 };
  82
  83 } // namespace
  84
  85 class PowerProfilerServiceTest : public testing::Test {
  86  public:
  87   void ServiceStartTest() {
  88     service_.reset(new PowerProfilerService(
  89         make_scoped_ptr<PowerDataProvider>(
  90             new TestPowerDataProvider(kNumEvents)),
  91         message_loop_.message_loop_proxy(),
  92         base::TimeDelta::FromMilliseconds(1)));
  93     EXPECT_TRUE(service_->IsAvailable());
  94   }
  95
  96   void AddObserverTest() {
  97     service_->AddObserver(&observer_);
  98
  99     // No PowerEvents received.
 100     EXPECT_EQ(observer_.valid_event_count(), 0);
 101   }
 102
 103   void RemoveObserverTest() {
 104      service_->RemoveObserver(&observer_);
 105
 106     // Received |kNumEvents| events.
 107     EXPECT_EQ(observer_.valid_event_count(), kNumEvents);
 108   }
 109
 110  protected:
 111   PowerProfilerServiceTest() : ui_thread_(BrowserThread::UI, &message_loop_) {}
 112   virtual ~PowerProfilerServiceTest() {}
 113
 114   void RegisterQuitClosure(base::Closure closure) {
 115     observer_.set_quit_closure(closure);
 116   }
 117
 118  private:
 119   scoped_ptr<PowerProfilerService> service_;
 120   TestPowerProfilerObserver observer_;
 121
 122   // UI thread.
 123   base::MessageLoopForUI message_loop_;
 124   BrowserThreadImpl ui_thread_;
 125
 126   DISALLOW_COPY_AND_ASSIGN(PowerProfilerServiceTest);
 127 };
 128
 129 // Test whether PowerProfilerService dispatches power events to observer
 130 // properly.
 131 TEST_F(PowerProfilerServiceTest, AvailableService) {
 132   base::RunLoop run_loop;
 133   RegisterQuitClosure(run_loop.QuitClosure());
 134
 135   ServiceStartTest();
 136   AddObserverTest();
 137
 138   run_loop.Run();
 139
 140   RemoveObserverTest();
 141 }
 142
 143 }  // namespace content