blob: 3eb384d17af3e77b59c33ac3fbb133d19a7e254e [file] [log] [blame]
// Copyright 2015 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.
#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/test/test_mock_time_task_runner.h"
#include "base/threading/thread_task_runner_handle.h"
#include "net/base/backoff_entry.h"
#include "net/url_request/test_url_fetcher_factory.h"
#include "net/url_request/url_request_test_util.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace gcm {
// The base class for testing various GCM requests that contains all the common
// logic to set up a task runner and complete a request with retries.
class GCMRequestTestBase : public testing::Test {
~GCMRequestTestBase() override;
const net::BackoffEntry::Policy& GetBackoffPolicy() const;
net::TestURLFetcher* GetFetcher() const;
// Set the response status and body that will be returned by the URL fetch.
void SetResponse(net::HttpStatusCode status_code,
const std::string& response_body);
// Completes the URL fetch.
// It can be overridden by the test class to add additional logic.
virtual void CompleteFetch();
// Verifies that the Fetcher's upload_data exactly matches the given
// properties. The map will be cleared as a side-effect. Wrap calls to this
void VerifyFetcherUploadData(
std::map<std::string, std::string>* expected_pairs);
net::URLRequestContextGetter* url_request_context_getter() const {
return url_request_context_getter_.get();
// Fast forward the timer used in the test to retry the request immediately.
void FastForwardToTriggerNextRetry();
scoped_refptr<base::TestMockTimeTaskRunner> task_runner_;
base::ThreadTaskRunnerHandle task_runner_handle_;
scoped_refptr<net::TestURLRequestContextGetter> url_request_context_getter_;
net::TestURLFetcherFactory url_fetcher_factory_;
// Tracks the number of retries so far.
int retry_count_;
} // namespace gcm