1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/download/internal/background_service/ios/background_download_task_helper.h"
#include <memory>
#import "base/files/file_util.h"
#import "base/files/scoped_temp_dir.h"
#import "base/functional/callback_helpers.h"
#import "base/run_loop.h"
#import "base/sequence_checker.h"
#import "base/test/bind.h"
#import "base/test/gmock_callback_support.h"
#import "base/test/task_environment.h"
#import "base/uuid.h"
#import "components/download/internal/background_service/test/background_download_test_base.h"
#import "components/download/public/background_service/download_params.h"
#import "net/test/embedded_test_server/embedded_test_server.h"
#import "net/test/embedded_test_server/http_request.h"
#import "net/test/embedded_test_server/http_response.h"
using net::test_server::HttpRequest;
using net::test_server::HttpResponse;
using net::test_server::HttpMethod;
const char kHeaderValue[] = "abcd1234";
namespace download {
class BackgroundDownloadTaskHelperTest
: public test::BackgroundDownloadTestBase {
protected:
BackgroundDownloadTaskHelperTest() = default;
~BackgroundDownloadTaskHelperTest() override = default;
void SetUp() override {
BackgroundDownloadTestBase::SetUp();
helper_ = BackgroundDownloadTaskHelper::Create();
}
void Download(
const std::string& relative_url,
const std::string& guid,
BackgroundDownloadTaskHelper::CompletionCallback completion_callback) {
DownloadParams params;
params.request_params.url = server_.GetURL(relative_url);
params.request_params.method = "POST";
params.request_params.request_headers.SetHeader(
net::HttpRequestHeaders::kIfMatch, kHeaderValue);
helper_->StartDownload(guid, dir_.GetPath().AppendASCII(guid),
params.request_params, params.scheduling_params,
std::move(completion_callback), base::DoNothing());
}
BackgroundDownloadTaskHelper* helper() { return helper_.get(); }
private:
base::test::TaskEnvironment task_environment_;
std::unique_ptr<BackgroundDownloadTaskHelper> helper_;
};
// Verifies download can be finished.
// TODO(crbug.com/40239993): Re-enable the test.
TEST_F(BackgroundDownloadTaskHelperTest, DISABLED_DownloadComplete) {
base::RunLoop loop;
std::string guid = base::Uuid::GenerateRandomV4().AsLowercaseString();
Download("/test", guid,
base::BindLambdaForTesting([&](bool success,
const base::FilePath& file_path,
int64_t file_size) {
std::string content;
EXPECT_TRUE(success);
ASSERT_TRUE(base::ReadFileToString(file_path, &content));
EXPECT_EQ(BackgroundDownloadTestBase::kDefaultResponseContent,
content);
EXPECT_EQ(file_size, static_cast<int64_t>(content.size()));
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
loop.Quit();
}));
loop.Run();
EXPECT_TRUE(base::PathExists(dir().GetPath().AppendASCII(guid)));
ASSERT_TRUE(request_sent());
auto it = request_sent()->headers.find(net::HttpRequestHeaders::kIfMatch);
EXPECT_EQ(kHeaderValue, it->second);
EXPECT_EQ(HttpMethod::METHOD_POST, request_sent()->method);
}
// Verifies non success http code is treated as error.
// TODO(crbug.com/40239993): Re-enable the test.
TEST_F(BackgroundDownloadTaskHelperTest,
DISABLED_DownloadErrorNonSuccessHttpCode) {
base::RunLoop loop;
std::string guid = base::Uuid::GenerateRandomV4().AsLowercaseString();
Download("/notfound", guid,
base::BindLambdaForTesting([&](bool success,
const base::FilePath& file_path,
int64_t file_size) {
EXPECT_FALSE(success);
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
loop.Quit();
}));
loop.Run();
EXPECT_FALSE(base::PathExists(dir().GetPath().AppendASCII(guid)));
}
// Verifies data URL should result in failure.
// TODO(crbug.com/40239993): Re-enable the test.
TEST_F(BackgroundDownloadTaskHelperTest, DISABLED_DataURL) {
base::RunLoop loop;
std::string guid = base::Uuid::GenerateRandomV4().AsLowercaseString();
DownloadParams params;
params.request_params.url = GURL("data:text/plain;base64,Q2hyb21pdW0=");
helper()->StartDownload(
guid, dir_.GetPath().AppendASCII(guid), params.request_params,
params.scheduling_params,
base::BindLambdaForTesting([&](bool success,
const base::FilePath& file_path,
int64_t file_size) {
EXPECT_FALSE(success);
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
loop.Quit();
}),
base::DoNothing());
loop.Run();
EXPECT_FALSE(base::PathExists(dir().GetPath().AppendASCII(guid)));
}
} // namespace download
|
