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

Side by Side Diff: content/browser/download/download_job_unittest.cc

Issue 2645133003: Introduce DownloadJob for DownloadItem refactoring. (Closed)
Patch Set: Comments change. Created 3 years, 11 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
(Empty)
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
3 // found in the LICENSE file.
4
5 #include <memory>
6
7 #include "base/memory/ptr_util.h"
8 #include "content/browser/download/download_job.h"
9 #include "content/browser/download/mock_download_job.h"
10 #include "testing/gmock/include/gmock/gmock.h"
11 #include "testing/gtest/include/gtest/gtest.h"
12
13 namespace content {
14
15 class MockDownloadJobManager : public DownloadJob::Manager {
16 public:
17 MockDownloadJobManager() = default;
18 ~MockDownloadJobManager() = default;
19
20 MOCK_METHOD1(OnSavingStarted, void(DownloadJob* download_job));
21 MOCK_METHOD1(SetDangerType, void(DownloadDangerType));
22 MOCK_METHOD2(OnDownloadInterrupted,
23 void(DownloadJob* download_job, DownloadInterruptReason reason));
24 MOCK_METHOD1(OnDownloadComplete, void(DownloadJob* download_job));
25 };
26
27 // Test for DownloadJob base class functionalities.
28 class DownloadJobTest : public testing::Test {
29 public:
30 DownloadJobTest() = default;
31 ~DownloadJobTest() override = default;
32
33 void SetUp() override {
34 download_job_manager_ = base::MakeUnique<MockDownloadJobManager>();
35 EXPECT_TRUE(download_job_manager_.get());
36 download_job_ = base::MakeUnique<MockDownloadJob>();
37 }
38
39 std::unique_ptr<MockDownloadJobManager> download_job_manager_;
40 std::unique_ptr<MockDownloadJob> download_job_;
41 };
42
43 TEST_F(DownloadJobTest, AttachAndDetach) {
44 // Ensure the manager should be valid only between attach and detach call.
45 EXPECT_FALSE(download_job_->manager());
46 download_job_->OnAttached(download_job_manager_.get());
47 EXPECT_TRUE(download_job_->manager());
48 download_job_->OnBeforeDetach();
49 EXPECT_FALSE(download_job_->manager());
50 }
51
52 } // namespace content
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698