blob: 9c8f48b19d039c1d21854c1c31157393f82fc6e6 (
plain)
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 2014 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.
#import "ios/chrome/browser/net/retryable_url_fetcher.h"
#import "base/mac/scoped_nsobject.h"
#include "ios/web/public/test/test_web_thread.h"
#include "net/url_request/test_url_fetcher_factory.h"
#include "net/url_request/url_fetcher_delegate.h"
#include "net/url_request/url_request_test_util.h"
#import "testing/gtest_mac.h"
#include "testing/platform_test.h"
namespace {
// An arbitrary text string for a fake response.
NSString* const kFakeResponseString = @"Something interesting here.";
}
// Delegate object to provide data for RetryableURLFetcher and
// handles the callback when URL is fetched.
@interface TestRetryableURLFetcherDelegate
: NSObject<RetryableURLFetcherDelegate>
// Counts the number of times that a successful response has been processed.
@property(nonatomic, assign) NSUInteger responsesProcessed;
@end
@implementation TestRetryableURLFetcherDelegate
@synthesize responsesProcessed;
- (NSString*)urlToFetch {
return @"http://www.google.com";
}
- (void)processSuccessResponse:(NSString*)response {
if (response) {
EXPECT_NSEQ(kFakeResponseString, response);
++responsesProcessed;
}
}
@end
namespace {
class RetryableURLFetcherTest : public PlatformTest {
protected:
void SetUp() override {
PlatformTest::SetUp();
test_delegate_.reset([[TestRetryableURLFetcherDelegate alloc] init]);
io_thread_.reset(
new web::TestWebThread(web::WebThread::IO, &message_loop_));
}
net::TestURLFetcherFactory factory_;
scoped_ptr<web::TestWebThread> io_thread_;
base::MessageLoop message_loop_;
base::scoped_nsobject<TestRetryableURLFetcherDelegate> test_delegate_;
};
TEST_F(RetryableURLFetcherTest, TestResponse200) {
scoped_refptr<net::URLRequestContextGetter> request_context_getter =
new net::TestURLRequestContextGetter(message_loop_.task_runner());
base::scoped_nsobject<RetryableURLFetcher> retryableFetcher(
[[RetryableURLFetcher alloc]
initWithRequestContextGetter:request_context_getter.get()
delegate:test_delegate_.get()
backoffPolicy:nil]);
[retryableFetcher startFetch];
// Manually calls the delegate.
net::TestURLFetcher* fetcher = factory_.GetFetcherByID(0);
DCHECK(fetcher);
DCHECK(fetcher->delegate());
[test_delegate_ setResponsesProcessed:0U];
fetcher->set_response_code(200);
fetcher->SetResponseString([kFakeResponseString UTF8String]);
fetcher->delegate()->OnURLFetchComplete(fetcher);
EXPECT_EQ(1U, [test_delegate_ responsesProcessed]);
}
TEST_F(RetryableURLFetcherTest, TestResponse404) {
scoped_refptr<net::URLRequestContextGetter> request_context_getter =
new net::TestURLRequestContextGetter(message_loop_.task_runner());
base::scoped_nsobject<RetryableURLFetcher> retryableFetcher(
[[RetryableURLFetcher alloc]
initWithRequestContextGetter:request_context_getter.get()
delegate:test_delegate_.get()
backoffPolicy:nil]);
[retryableFetcher startFetch];
// Manually calls the delegate.
net::TestURLFetcher* fetcher = factory_.GetFetcherByID(0);
DCHECK(fetcher);
DCHECK(fetcher->delegate());
[test_delegate_ setResponsesProcessed:0U];
fetcher->set_response_code(404);
fetcher->SetResponseString("");
fetcher->delegate()->OnURLFetchComplete(fetcher);
EXPECT_EQ(0U, [test_delegate_ responsesProcessed]);
}
} // namespace
|
