Home | History | Annotate | Download | only in trace_event
      1 // Copyright 2016 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 BASE_TRACE_EVENT_TRACE_EVENT_FILTER_TEST_UTILS_H_
      6 #define BASE_TRACE_EVENT_TRACE_EVENT_FILTER_TEST_UTILS_H_
      7 
      8 #include <memory>
      9 #include <string>
     10 
     11 #include "base/macros.h"
     12 #include "base/trace_event/trace_event_filter.h"
     13 
     14 namespace base {
     15 namespace trace_event {
     16 
     17 class TestEventFilter : public TraceEventFilter {
     18  public:
     19   struct HitsCounter {
     20     HitsCounter();
     21     ~HitsCounter();
     22     void Reset();
     23     size_t filter_trace_event_hit_count;
     24     size_t end_event_hit_count;
     25   };
     26 
     27   static const char kName[];
     28 
     29   // Factory method for TraceLog::SetFilterFactoryForTesting().
     30   static std::unique_ptr<TraceEventFilter> Factory(
     31       const std::string& predicate_name);
     32 
     33   TestEventFilter();
     34   ~TestEventFilter() override;
     35 
     36   // TraceEventFilter implementation.
     37   bool FilterTraceEvent(const TraceEvent& trace_event) const override;
     38   void EndEvent(const char* category_name, const char* name) const override;
     39 
     40   static void set_filter_return_value(bool value) {
     41     filter_return_value_ = value;
     42   }
     43 
     44  private:
     45   static bool filter_return_value_;
     46 
     47   DISALLOW_COPY_AND_ASSIGN(TestEventFilter);
     48 };
     49 
     50 }  // namespace trace_event
     51 }  // namespace base
     52 
     53 #endif  // BASE_TRACE_EVENT_TRACE_EVENT_FILTER_TEST_UTILS_H_
     54