Index: counter_mock.h |
diff --git a/counter_mock.h b/counter_mock.h |
index 701cce48eed923fd21ee14745961683440ecedb2..cf2a486c19314e21d9aedb98706f8060e4809e87 100644 |
--- a/counter_mock.h |
+++ b/counter_mock.h |
@@ -13,7 +13,7 @@ |
namespace chromeos_metrics { |
-class TaggedCounterMock : public TaggedCounterInterface { |
+class TaggedCounterMock : public TaggedCounter { |
public: |
MOCK_METHOD3(Init, void(const char* filename, |
Reporter reporter, void* reporter_handle)); |
@@ -21,6 +21,17 @@ class TaggedCounterMock : public TaggedCounterInterface { |
MOCK_METHOD0(Flush, void()); |
}; |
+class TaggedCounterReporterMock : public TaggedCounterReporter { |
+ public: |
+ MOCK_METHOD5(Init, void(const char* filename, |
+ const char* histogram_name, |
+ int min, |
+ int max, |
+ int nbuckets)); |
+ MOCK_METHOD2(Update, void(int32 tag, int32 count)); |
+ MOCK_METHOD0(Flush, void()); |
+}; |
+ |
class FrequencyCounterMock : public FrequencyCounter { |
public: |
MOCK_METHOD4(Init, void(const char* filename, |
@@ -28,6 +39,7 @@ class FrequencyCounterMock : public FrequencyCounter { |
void* reporter_handle, |
time_t cycle_duration)); |
MOCK_METHOD1(Update, void(int32 count)); |
+ MOCK_METHOD0(FlushFinishedCycles, void()); |
}; |
} // namespace chromeos_metrics |