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

Side by Side Diff: components/download/internal/test/test_download_driver.h

Issue 2860593003: Refactoring DownloadManager::DownloadURL to add proper annotation. (Closed)
Patch Set: Comment addressed. Created 3 years, 6 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 unified diff | Download patch
OLDNEW
1 // Copyright 2017 The Chromium Authors. All rights reserved. 1 // Copyright 2017 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #ifndef COMPONENTS_DOWNLOAD_INTERNAL_TEST_TEST_DOWNLOAD_DRIVER_H_ 5 #ifndef COMPONENTS_DOWNLOAD_INTERNAL_TEST_TEST_DOWNLOAD_DRIVER_H_
6 #define COMPONENTS_DOWNLOAD_INTERNAL_TEST_TEST_DOWNLOAD_DRIVER_H_ 6 #define COMPONENTS_DOWNLOAD_INTERNAL_TEST_TEST_DOWNLOAD_DRIVER_H_
7 7
8 #include <map> 8 #include <map>
9 #include <memory> 9 #include <memory>
10 10
11 #include "base/macros.h" 11 #include "base/macros.h"
12 #include "components/download/internal/download_driver.h" 12 #include "components/download/internal/download_driver.h"
13 #include "net/traffic_annotation/network_traffic_annotation.h"
13 14
14 namespace download { 15 namespace download {
15 namespace test { 16 namespace test {
16 17
17 // Download driver that simulates content layer download logic. 18 // Download driver that simulates content layer download logic.
18 class TestDownloadDriver : public DownloadDriver { 19 class TestDownloadDriver : public DownloadDriver {
19 public: 20 public:
20 TestDownloadDriver(); 21 TestDownloadDriver();
21 ~TestDownloadDriver() override; 22 ~TestDownloadDriver() override;
22 23
23 // Marks download driver as ready, used to test logic that depends on 24 // Marks download driver as ready, used to test logic that depends on
24 // data initialization. 25 // data initialization.
25 void MakeReady(); 26 void MakeReady();
26 27
27 // Simulates download events from content layer. 28 // Simulates download events from content layer.
28 void NotifyDownloadUpdate(const DriverEntry& entry); 29 void NotifyDownloadUpdate(const DriverEntry& entry);
29 void NotifyDownloadFailed(const DriverEntry& entry, int reason); 30 void NotifyDownloadFailed(const DriverEntry& entry, int reason);
30 void NotifyDownloadSucceeded(const DriverEntry& entry, 31 void NotifyDownloadSucceeded(const DriverEntry& entry,
31 const base::FilePath& path); 32 const base::FilePath& path);
32 33
33 // DownloadDriver implementation. 34 // DownloadDriver implementation.
34 void Initialize(DownloadDriver::Client* client) override; 35 void Initialize(DownloadDriver::Client* client) override;
35 bool IsReady() const override; 36 bool IsReady() const override;
36 void Start(const DownloadParams& params) override; 37 void Start(
38 const DownloadParams& params,
39 const net::NetworkTrafficAnnotationTag& traffic_annotation) override;
37 void Cancel(const std::string& guid) override; 40 void Cancel(const std::string& guid) override;
38 void Pause(const std::string& guid) override; 41 void Pause(const std::string& guid) override;
39 void Resume(const std::string& guid) override; 42 void Resume(const std::string& guid) override;
40 base::Optional<DriverEntry> Find(const std::string& guid) override; 43 base::Optional<DriverEntry> Find(const std::string& guid) override;
41 44
42 private: 45 private:
43 bool is_ready_; 46 bool is_ready_;
44 DownloadDriver::Client* client_; 47 DownloadDriver::Client* client_;
45 48
46 // Map of guid --> DriverEntry. 49 // Map of guid --> DriverEntry.
47 std::map<std::string, DriverEntry> entries_; 50 std::map<std::string, DriverEntry> entries_;
48 51
49 DISALLOW_COPY_AND_ASSIGN(TestDownloadDriver); 52 DISALLOW_COPY_AND_ASSIGN(TestDownloadDriver);
50 }; 53 };
51 54
52 } // namespace test 55 } // namespace test
53 } // namespace download 56 } // namespace download
54 57
55 #endif // COMPONENTS_DOWNLOAD_CONTENT_TEST_TEST_DOWNLOAD_DRIVER_H_ 58 #endif // COMPONENTS_DOWNLOAD_CONTENT_TEST_TEST_DOWNLOAD_DRIVER_H_
OLDNEW
« no previous file with comments | « components/download/internal/download_driver.h ('k') | components/download/internal/test/test_download_driver.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698