blob: 9d100554983acca17298ca83fb037a28a511dd4e [file] [log] [blame]
// Copyright (c) 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef NET_SPDY_CHROMIUM_SPDY_SESSION_TEST_UTIL_H_
#define NET_SPDY_CHROMIUM_SPDY_SESSION_TEST_UTIL_H_
#include <stdint.h>
#include "base/message_loop/message_loop.h"
#include "base/pending_task.h"
#include "net/third_party/spdy/platform/api/spdy_string.h"
namespace net {
// SpdySessionTestTaskObserver is a MessageLoop::TaskObserver that monitors the
// completion of all tasks executed by the current MessageLoop, recording the
// number of tasks that refer to a specific function and filename.
class SpdySessionTestTaskObserver : public base::MessageLoop::TaskObserver {
public:
// Creates a SpdySessionTaskObserver that will record all tasks that are
// executed that were posted by the function named by |function_name|, located
// in the file |file_name|.
// Example:
// file_name = "foo.cc"
// function = "DoFoo"
SpdySessionTestTaskObserver(const SpdyString& file_name,
const SpdyString& function_name);
~SpdySessionTestTaskObserver() override;
// Implements MessageLoop::TaskObserver.
void WillProcessTask(const base::PendingTask& pending_task) override;
void DidProcessTask(const base::PendingTask& pending_task) override;
// Returns the number of tasks posted by the given function and file.
uint16_t executed_count() const { return executed_count_; }
private:
uint16_t executed_count_;
SpdyString file_name_;
SpdyString function_name_;
};
} // namespace net
#endif // NET_SPDY_CHROMIUM_SPDY_SESSION_TEST_UTIL_H_