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
|
// Copyright 2017 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.
#include "components/offline_pages/core/prefetch/get_operation_request.h"
#include "base/test/mock_callback.h"
#include "components/offline_pages/core/prefetch/prefetch_request_test_base.h"
#include "components/offline_pages/core/prefetch/prefetch_types.h"
#include "components/offline_pages/core/prefetch/proto/offline_pages.pb.h"
#include "net/http/http_status_code.h"
#include "net/url_request/url_request_status.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "url/gurl.h"
#include "url/url_constants.h"
using testing::_;
using testing::DoAll;
using testing::Eq;
using testing::SaveArg;
namespace offline_pages {
namespace {
const version_info::Channel kTestChannel = version_info::Channel::UNKNOWN;
// Operations include part of the path in the operation name.
const char kTestOperationName[] = "operations/test-operation-1234";
// This is hard coded so we can express the known URL structure for operations.
const char kServerPathForTestOperation[] = "/v1/operations/test-operation-1234";
} // namespace
// All tests cases here only validate the request data and check for general
// http response. The tests for the Operation proto data returned in the http
// response are covered in PrefetchRequestOperationResponseTest.
class GetOperationRequestTest : public PrefetchRequestTestBase {
public:
std::unique_ptr<GetOperationRequest> CreateRequest(
const PrefetchRequestFinishedCallback& callback) {
return std::unique_ptr<GetOperationRequest>(new GetOperationRequest(
kTestOperationName, kTestChannel, request_context(), callback));
}
};
TEST_F(GetOperationRequestTest, RequestData) {
base::MockCallback<PrefetchRequestFinishedCallback> callback;
std::unique_ptr<GetOperationRequest> request(CreateRequest(callback.Get()));
net::TestURLFetcher* fetcher = GetRunningFetcher();
GURL fetcher_url = fetcher->GetOriginalURL();
EXPECT_TRUE(fetcher_url.SchemeIs(url::kHttpsScheme));
EXPECT_EQ(kServerPathForTestOperation, fetcher_url.path());
EXPECT_TRUE(base::StartsWith(fetcher_url.query(), "key",
base::CompareCase::SENSITIVE));
net::HttpRequestHeaders headers;
fetcher->GetExtraRequestHeaders(&headers);
std::string content_type_header;
headers.GetHeader(net::HttpRequestHeaders::kContentType,
&content_type_header);
EXPECT_EQ("application/x-protobuf", content_type_header);
EXPECT_TRUE(fetcher->upload_content_type().empty());
EXPECT_TRUE(fetcher->upload_data().empty());
}
TEST_F(GetOperationRequestTest, EmptyResponse) {
base::MockCallback<PrefetchRequestFinishedCallback> callback;
std::unique_ptr<GetOperationRequest> request(CreateRequest(callback.Get()));
PrefetchRequestStatus status;
std::string operation_name;
std::vector<RenderPageInfo> pages;
EXPECT_CALL(callback, Run(_, _, _))
.WillOnce(DoAll(SaveArg<0>(&status), SaveArg<1>(&operation_name),
SaveArg<2>(&pages)));
RespondWithData("");
EXPECT_EQ(PrefetchRequestStatus::SHOULD_RETRY_WITH_BACKOFF, status);
EXPECT_TRUE(operation_name.empty());
EXPECT_TRUE(pages.empty());
}
TEST_F(GetOperationRequestTest, InvalidResponse) {
base::MockCallback<PrefetchRequestFinishedCallback> callback;
std::unique_ptr<GetOperationRequest> request(CreateRequest(callback.Get()));
PrefetchRequestStatus status;
std::string operation_name;
std::vector<RenderPageInfo> pages;
EXPECT_CALL(callback, Run(_, _, _))
.WillOnce(DoAll(SaveArg<0>(&status), SaveArg<1>(&operation_name),
SaveArg<2>(&pages)));
RespondWithData("Some invalid data");
EXPECT_EQ(PrefetchRequestStatus::SHOULD_RETRY_WITH_BACKOFF, status);
EXPECT_TRUE(operation_name.empty());
EXPECT_TRUE(pages.empty());
}
} // namespace offline_pages
|
