| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 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 "components/webp_transcode/webp_network_client.h" | |
| 6 | |
| 7 #include <stddef.h> | |
| 8 | |
| 9 #include "base/mac/scoped_nsobject.h" | |
| 10 #include "base/macros.h" | |
| 11 #include "base/memory/weak_ptr.h" | |
| 12 #include "base/message_loop/message_loop.h" | |
| 13 #include "base/thread_task_runner_handle.h" | |
| 14 #include "net/http/http_request_headers.h" | |
| 15 #include "net/url_request/url_request_test_util.h" | |
| 16 #include "testing/gtest/include/gtest/gtest.h" | |
| 17 #import "third_party/ocmock/OCMock/OCMock.h" | |
| 18 | |
| 19 namespace { | |
| 20 class WebPNetworkClientTest : public testing::Test { | |
| 21 public: | |
| 22 WebPNetworkClientTest() { | |
| 23 // Set up mock original network client proxy. | |
| 24 OCMockObject* mockProxy_ = [[OCMockObject | |
| 25 niceMockForProtocol:@protocol(CRNNetworkClientProtocol)] retain]; | |
| 26 mockWebProxy_.reset(mockProxy_); | |
| 27 | |
| 28 // Link all the mock objects into the WebPNetworkClient. | |
| 29 webp_client_.reset([[WebPNetworkClient alloc] | |
| 30 initWithTaskRunner:base::ThreadTaskRunnerHandle::Get()]); | |
| 31 [webp_client_ | |
| 32 setUnderlyingClient:(id<CRNNetworkClientProtocol>)mockWebProxy_]; | |
| 33 } | |
| 34 | |
| 35 protected: | |
| 36 base::MessageLoop loop_; | |
| 37 base::scoped_nsobject<WebPNetworkClient> webp_client_; | |
| 38 // Holds a mock CRNNetworkClientProtocol object. | |
| 39 base::scoped_nsobject<OCMockObject> mockWebProxy_; | |
| 40 }; | |
| 41 } // namespace | |
| 42 | |
| 43 TEST_F(WebPNetworkClientTest, TestAcceptHeaders) { | |
| 44 const struct { | |
| 45 const std::string header_in; | |
| 46 const std::string header_out; | |
| 47 } tests[] = { | |
| 48 {"", "image/webp"}, | |
| 49 {"*/*", "*/*,image/webp"}, | |
| 50 {"image/webp", "image/webp"}, | |
| 51 {"text/html,*/*", "text/html,*/*,image/webp"}, | |
| 52 // Desktop Chrome default without image/webp. | |
| 53 {"text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8", | |
| 54 "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8," | |
| 55 "image/webp"}, | |
| 56 // Desktop Chrome default. | |
| 57 {"text/html,application/xhtml+xml,application/xml;q=0.9,image/webp," | |
| 58 "*/*;q=0.8", | |
| 59 "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp," | |
| 60 "*/*;q=0.8"}}; | |
| 61 GURL url("http://www.google.com"); | |
| 62 scoped_ptr<net::URLRequestContext> request_context( | |
| 63 new net::TestURLRequestContext(false)); | |
| 64 for (size_t i = 0; i < arraysize(tests); ++i) { | |
| 65 scoped_ptr<net::URLRequest> request = | |
| 66 request_context->CreateRequest(url, net::DEFAULT_PRIORITY, nullptr); | |
| 67 if (!tests[i].header_in.empty()) | |
| 68 request->SetExtraRequestHeaderByName("Accept", tests[i].header_in, true); | |
| 69 [webp_client_ didCreateNativeRequest:request.get()]; | |
| 70 const net::HttpRequestHeaders& headers = request->extra_request_headers(); | |
| 71 std::string acceptHeader; | |
| 72 EXPECT_TRUE(headers.GetHeader("Accept", &acceptHeader)); | |
| 73 EXPECT_EQ(tests[i].header_out, acceptHeader); | |
| 74 } | |
| 75 } | |
| OLD | NEW |