Chromium Code Reviews| Index: content/browser/download/mock_download_file.h |
| diff --git a/content/browser/download/mock_download_file.h b/content/browser/download/mock_download_file.h |
| new file mode 100755 |
| index 0000000000000000000000000000000000000000..ed202030ce1bce91f6153aad5e5f03e23b8ee619 |
| --- /dev/null |
| +++ b/content/browser/download/mock_download_file.h |
| @@ -0,0 +1,116 @@ |
| +// Copyright (c) 2011 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 CONTENT_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_FILE_H_ |
| +#define CONTENT_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_FILE_H_ |
| +#pragma once |
| + |
| +#include <string> |
| +#include <map> |
| + |
| +#include "base/file_path.h" |
| +#include "base/memory/ref_counted.h" |
| +#include "content/browser/download/download_file.h" |
| +#include "content/browser/download/download_id.h" |
| +#include "content/browser/download/download_manager.h" |
| +#include "content/browser/download/download_request_handle.h" |
| +#include "net/base/net_errors.h" |
| + |
| +struct DownloadCreateInfo; |
| + |
| +class MockDownloadFile : virtual public DownloadFile { |
|
Randy Smith (Not in Mondays)
2011/11/14 21:58:39
What are the advantages of creating this class as
ahendrickson
2011/11/16 19:39:11
It's a lot more flexible. You can monitor stuff,
|
| + public: |
| + // Class to extract statistics from the usage of |MockDownloadFile|. |
| + class StatisticsRecorder { |
| + public: |
| + enum StatisticsIndex { |
| + STAT_INITIALIZE, |
| + STAT_APPEND, |
| + STAT_RENAME, |
| + STAT_DETACH, |
| + STAT_CANCEL, |
| + STAT_FINISH, |
| + STAT_BYTES |
| + }; |
| + |
| + StatisticsRecorder(); |
| + ~StatisticsRecorder(); |
| + |
| + // Records the statistic. |
| + // |index| indicates what statistic to use. |
| + void Record(StatisticsIndex index); |
| + |
| + // Adds to the statistic. |
| + // |index| indicates what statistic to use. |
| + void Add(StatisticsIndex index, int count); |
| + |
| + // Returns the statistic value. |
| + // |index| indicates what statistic to use. |
| + int Count(StatisticsIndex index); |
| + |
| + private: |
| + typedef std::map<StatisticsIndex, int> StatisticsMap; |
| + |
| + StatisticsMap map_; |
| + }; |
| + |
| + MockDownloadFile(const DownloadCreateInfo* info, |
| + const DownloadRequestHandle& request_handle, |
| + DownloadManager* download_manager, |
| + StatisticsRecorder* recorder); |
| + virtual ~MockDownloadFile(); |
| + |
| + // DownloadFile functions. |
| + virtual net::Error Initialize(bool calculate_hash) OVERRIDE; |
| + virtual net::Error AppendDataToFile(const char* data, |
| + size_t data_len) OVERRIDE; |
| + virtual net::Error Rename(const FilePath& full_path) OVERRIDE; |
| + virtual void Detach() OVERRIDE; |
| + virtual void Cancel() OVERRIDE; |
| + virtual void Finish() OVERRIDE; |
| + virtual void AnnotateWithSourceInformation() OVERRIDE; |
| + virtual FilePath FullPath() const OVERRIDE; |
| + virtual bool InProgress() const OVERRIDE; |
| + virtual int64 BytesSoFar() const OVERRIDE; |
| + virtual bool GetSha256Hash(std::string* hash) OVERRIDE; |
| + virtual void CancelDownloadRequest() OVERRIDE; |
| + virtual int Id() const OVERRIDE; |
| + virtual DownloadManager* GetDownloadManager() OVERRIDE; |
| + virtual const DownloadId& GlobalId() const OVERRIDE; |
| + virtual std::string DebugString() const OVERRIDE; |
| + |
| + // Functions relating to setting and checking expectations. |
| + size_t rename_count() const { return rename_count_; } |
| + void SetExpectedPath(size_t index, const FilePath& path); |
| + |
| + private: |
| + // The unique identifier for this download, assigned at creation by |
| + // the DownloadFileManager for its internal record keeping. |
| + DownloadId id_; |
| + |
| + // The handle to the request information. Used for operations outside the |
| + // download system, specifically canceling a download. |
| + DownloadRequestHandle request_handle_; |
| + |
| + // DownloadManager this download belongs to. |
| + scoped_refptr<DownloadManager> download_manager_; |
| + |
| + // Records usage statistics. Not owned by this class (survives destruction). |
| + StatisticsRecorder* recorder_; |
| + |
| + // The number of times |Rename()| has been called. |
| + // Used instead of checking |recorder_| because the latter can be NULL. |
| + size_t rename_count_; |
| + |
| + // A vector of paths that we expect to call |Rename()| with. |
| + std::vector<FilePath> expected_rename_path_list_; |
| + |
| + // A buffer to hold the data we write. |
| + std::string data_; |
| + |
| + // Dummy in-progress flag. |
| + bool in_progress_; |
| +}; |
| + |
| +#endif // CONTENT_BROWSER_DOWNLOAD_MOCK_DOWNLOAD_FILE_H_ |