Chromium Code Reviews| Index: components/data_reduction_proxy/common/data_reduction_proxy_headers_unittest.cc |
| diff --git a/components/data_reduction_proxy/common/data_reduction_proxy_headers_unittest.cc b/components/data_reduction_proxy/common/data_reduction_proxy_headers_unittest.cc |
| index 871d66589e0d4834e51ab59ab0bc56de6e3e1934..1e1d8a2a7013bab6693336b708f189587a6dbae7 100644 |
| --- a/components/data_reduction_proxy/common/data_reduction_proxy_headers_unittest.cc |
| +++ b/components/data_reduction_proxy/common/data_reduction_proxy_headers_unittest.cc |
| @@ -24,6 +24,94 @@ namespace data_reduction_proxy { |
| class DataReductionProxyHeadersTest : public testing::Test {}; |
| +TEST_F(DataReductionProxyHeadersTest, GetDataReductionProxyActionValue) { |
| + const struct { |
| + const char* headers; |
| + std::string action_key; |
| + bool expected_result; |
| + std::string expected_action_value; |
| + } tests[] = { |
| + { "HTTP/1.1 200 OK\n" |
| + "Content-Length: 999\n", |
| + "a=", |
| + false, |
| + "", |
| + }, |
| + { "HTTP/1.1 200 OK\n" |
| + "connection: keep-alive\n" |
| + "Content-Length: 999\n", |
| + "a=", |
| + false, |
| + "", |
| + }, |
| + { "HTTP/1.1 200 OK\n" |
| + "connection: keep-alive\n" |
| + "Chrome-Proxy: bypass=86400\n" |
| + "Content-Length: 999\n", |
| + "bypass=", |
| + true, |
| + "86400", |
| + }, |
| + { "HTTP/1.1 200 OK\n" |
| + "connection: keep-alive\n" |
| + "Chrome-Proxy: bypass=0\n" |
| + "Content-Length: 999\n", |
| + "bypass=", |
| + true, |
| + "0", |
| + }, |
| + { "HTTP/1.1 200 OK\n" |
| + "connection: keep-alive\n" |
| + "Chrome-Proxy: bypass=1500\n" |
| + "Chrome-Proxy: bypass=86400\n" |
| + "Content-Length: 999\n", |
| + "bypass=", |
| + true, |
| + "1500", |
| + }, |
| + { "HTTP/1.1 200 OK\n" |
| + "connection: keep-alive\n" |
| + "Chrome-Proxy: block=1500, block=3600\n" |
| + "Content-Length: 999\n", |
| + "block=", |
| + true, |
| + "1500", |
| + }, |
| + { "HTTP/1.1 200 OK\n" |
| + "connection: proxy-bypass\n" |
| + "Chrome-Proxy: key=123 \n" |
| + "Content-Length: 999\n", |
| + "key=", |
| + true, |
| + "123", |
| + }, |
| + { "HTTP/1.1 200 OK\n" |
| + "connection: proxy-bypass\n" |
| + "Chrome-Proxy: =123 \n" |
| + "Content-Length: 999\n", |
| + "=", |
|
bengr
2014/07/14 18:27:14
This shouldn't be allowed by any of the header pro
xingx
2014/07/14 21:37:17
Done.
|
| + true, |
| + "123", |
| + }, |
| + }; |
| + for (size_t i = 0; i < ARRAYSIZE_UNSAFE(tests); ++i) { |
| + std::string headers(tests[i].headers); |
| + HeadersToRaw(&headers); |
| + scoped_refptr<net::HttpResponseHeaders> parsed( |
| + new net::HttpResponseHeaders(headers)); |
| + |
| + std::string action_value; |
| + bool has_action_key = GetDataReductionProxyActionValue(parsed, |
| + tests[i].action_key, |
| + &action_value); |
| + EXPECT_EQ(tests[i].expected_result, has_action_key); |
| + if (has_action_key) { |
| + EXPECT_EQ(tests[i].expected_action_value, |
| + action_value); |
| + } |
| + } |
| +} |
| + |
| TEST_F(DataReductionProxyHeadersTest, GetProxyBypassInfo) { |
| const struct { |
| const char* headers; |