Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(51)

Unified Diff: content/test/test_file_error_injector.h

Issue 9426029: Test file errors in downloads. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Added test that injects an error on the 2nd write. Created 8 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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..d74e3553e709afeed4c7a95f2add761d9a557f2c
--- /dev/null
+++ b/content/test/test_file_error_injector.h
@@ -0,0 +1,83 @@
+// 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 "content/public/browser/download_id.h"
cbentzel 2012/03/02 20:45:19 I think you can forward declare download_id since
ahendrickson 2012/03/02 21:58:47 Done.
+#include "net/base/net_errors.h"
+
+class GURL;
+
+namespace content {
+
+// 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.
+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.
+ net::Error net_error; // Error to inject.
+ };
+
+ TestFileErrorInjector() {}
+
+ // Creates an instance.
+ static scoped_refptr<TestFileErrorInjector> Get();
+
+ // Adds an error.
+ // Must be called before |InjectErrors()| for a particular download file.
+ // Only one error per file will be used.
+ virtual bool AddError(const FileErrorInfo& error_info) = 0;
+
+ // Clears all errors.
+ // Doesn't affect files already created.
+ virtual void ClearErrors() = 0;
+
+ // Injects the errors.
+ // The download system must already be initialized before calling this.
+ 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_

Powered by Google App Engine
This is Rietveld 408576698