components/memory_pressure/test_memory_pressure_calculator.h

   1 // Copyright 2015 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 #ifndef COMPONENTS_MEMORY_PRESSURE_TEST_MEMORY_PRESSURE_CALCULATOR_H_
   6 #define COMPONENTS_MEMORY_PRESSURE_TEST_MEMORY_PRESSURE_CALCULATOR_H_
   7
   8 #include "components/memory_pressure/memory_pressure_calculator.h"
   9
  10 namespace memory_pressure {
  11
  12 #if defined(MEMORY_PRESSURE_IS_POLLING)
  13
  14 // A mock memory pressure calculator for unittesting.
  15 class TestMemoryPressureCalculator : public MemoryPressureCalculator {
  16  public:
  17   TestMemoryPressureCalculator();
  18   ~TestMemoryPressureCalculator() override {}
  19
  20   // MemoryPressureCalculator implementation.
  21   MemoryPressureLevel CalculateCurrentPressureLevel() override;
  22
  23   // Sets the mock calculator to return no pressure.
  24   void SetNone();
  25
  26   // Sets the mock calculator to return moderate pressure.
  27   void SetModerate();
  28
  29   // Sets the mock calculator to return critical pressure.
  30   void SetCritical();
  31
  32   // Resets the call counter to 'CalculateCurrentPressureLevel'.
  33   void ResetCalls();
  34
  35   // Returns the number of calls to 'CalculateCurrentPressureLevel'.
  36   int calls() const { return calls_; }
  37
  38  private:
  39   MemoryPressureLevel level_;
  40   int calls_;
  41
  42   DISALLOW_COPY_AND_ASSIGN(TestMemoryPressureCalculator);
  43 };
  44
  45 #endif  // defined(MEMORY_PRESSURE_IS_POLLING)
  46
  47 }  // namespace memory_pressure
  48
  49 #endif  // COMPONENTS_MEMORY_PRESSURE_TEST_MEMORY_PRESSURE_CALCULATOR_H_