Index: content/test/test_file_error_injector.h |
diff --git a/content/test/test_file_error_injector.h b/content/test/test_file_error_injector.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f0464a25e1f5c25c74de97b7198ab275430dcbf6 |
--- /dev/null |
+++ b/content/test/test_file_error_injector.h |
@@ -0,0 +1,104 @@ |
+// Copyright (c) 2012 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_TEST_TEST_FILE_ERROR_INJECTOR_H_ |
+#define CONTENT_TEST_TEST_FILE_ERROR_INJECTOR_H_ |
+#pragma once |
+ |
+#include <string> |
+ |
+#include "base/memory/ref_counted.h" |
+#include "net/base/net_errors.h" |
+ |
+class GURL; |
+ |
+namespace content { |
+ |
+class DownloadId; |
+ |
+// Test helper for injecting errors into download file operations. |
+// All errors for a download must be injected before it starts. |
+// This class needs to be |RefCountedThreadSafe| because the implementation |
+// is referenced by other classes that live past the time when the user is |
+// nominally done with it. These classes are internal to content/. |
+// |
+// NOTE: No more than one download with the same URL can be in progress at |
+// the same time. You can have multiple simultaneous downloads as long as the |
+// URLs are different, as the URLs are used as keys to get information about |
+// the download. |
+// |
+// Example: |
+// |
+// FileErrorInfo a = { url1, ... }; |
+// FileErrorInfo b = { url2, ... }; |
+// |
+// scoped_refptr<TestFileErrorInjector> injector = |
+// TestFileErrorInjector::Create(); |
+// |
+// injector->AddError(); |
+// injector->AddError(); |
Randy Smith (Not in Mondays)
2012/03/06 23:05:19
Shouldn't these refer to a & b?
ahendrickson
2012/03/08 22:00:08
Done.
|
+// injector->InjectErrors(); |
+// |
+// download_manager->DownloadUrl(url1, ...); |
+// download_manager->DownloadUrl(url2, ...); |
+// ... wait for downloads to finish or get an injected error ... |
Randy Smith (Not in Mondays)
2012/03/06 23:05:19
Thank you--I think this comment makes this class m
|
+// |
+class TestFileErrorInjector |
+ : public base::RefCountedThreadSafe<TestFileErrorInjector> { |
+ public: |
+ enum FileOperationCode { |
+ FILE_OPERATION_INITIALIZE, |
+ FILE_OPERATION_WRITE, |
+ FILE_OPERATION_RENAME |
+ }; |
+ |
+ // Structure that encapsulates the information needed to inject a file error. |
+ struct FileErrorInfo { |
+ std::string url; // Full URL of the download. Identifies the download. |
+ FileOperationCode code; // Operation to affect. |
+ int operation_instance; // 0-based count of operation calls, for each code. |
+ net::Error net_error; // Error to inject. |
+ }; |
+ |
+ TestFileErrorInjector() {} |
Randy Smith (Not in Mondays)
2012/03/06 23:05:19
If you want all instances of this class to be crea
ahendrickson
2012/03/08 22:00:08
Made it protected, as another class inherits from
Randy Smith (Not in Mondays)
2012/03/12 01:44:50
Good point; oops on my part.
|
+ |
+ // Creates an instance. May only be called once. |
Randy Smith (Not in Mondays)
2012/03/06 23:05:19
I'd say something about lifetime here, since many
ahendrickson
2012/03/08 22:00:08
Done.
|
+ static scoped_refptr<TestFileErrorInjector> Create(); |
+ |
+ // Adds an error. |
+ // Must be called before |InjectErrors()| for a particular download file. |
Randy Smith (Not in Mondays)
2012/03/06 23:05:19
This is a little misleading; you can have multiple
ahendrickson
2012/03/08 22:00:08
Done.
Randy Smith (Not in Mondays)
2012/03/12 01:44:50
Not seeing it?
ahendrickson
2012/03/12 11:51:02
I did change the comment. If it's still not clear
Randy Smith (Not in Mondays)
2012/03/12 17:43:50
Yeah, sorry, I think the interleaving in the diff
|
+ // Only one error per file will be used. |
Randy Smith (Not in Mondays)
2012/03/06 23:05:19
We have one comment thread indicating that it's an
ahendrickson
2012/03/08 22:00:08
The implementation currently DCHECKs if the same U
Randy Smith (Not in Mondays)
2012/03/12 01:44:50
Cool; thanks.
|
+ virtual bool AddError(const FileErrorInfo& error_info) = 0; |
+ |
+ // Clears all errors. |
+ // Only affects files created after the next call to InjectErrors(). |
+ virtual void ClearErrors() = 0; |
+ |
+ // Injects the errors such that new download files will be affected. |
+ // The download system must already be initialized before calling this. |
+ // Multiple calls are allowed, but only useful if the errors have changed. |
+ // Replaces the injected error list. |
+ virtual bool InjectErrors() = 0; |
+ |
+ // Get information about the DownloadFiles created. |
+ virtual size_t CurrentFileCount() const = 0; |
+ virtual size_t TotalFileCount() const = 0; |
+ virtual bool HadFile(const GURL& url) const = 0; |
+ virtual const content::DownloadId GetId(const GURL& url) const = 0; |
+ virtual void ClearFoundFiles() = 0; |
+ |
+ static std::string DebugString(FileOperationCode code); |
+ |
+ protected: |
+ virtual ~TestFileErrorInjector(); |
+ |
+ private: |
+ friend class base::RefCountedThreadSafe<TestFileErrorInjector>; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(TestFileErrorInjector); |
+}; |
+ |
+} // namespace content |
+ |
+#endif // CONTENT_TEST_TEST_FILE_ERROR_INJECTOR_H_ |