summaryrefslogtreecommitdiffstats
path: root/net/http/http_network_layer_unittest.cc
blob: 90a9adcfc32f6b651932a831b07db72e55c8d7ce (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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
// Copyright (c) 2010 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 "net/base/mock_host_resolver.h"
#include "net/base/net_log.h"
#include "net/base/ssl_config_service_defaults.h"
#include "net/http/http_network_layer.h"
#include "net/http/http_transaction_unittest.h"
#include "net/proxy/proxy_service.h"
#include "net/socket/socket_test_util.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/platform_test.h"

namespace net {

namespace {

class HttpNetworkLayerTest : public PlatformTest {
};

TEST_F(HttpNetworkLayerTest, CreateAndDestroy) {
  MockHostResolver host_resolver;
  net::HttpNetworkLayer factory(
      NULL,
      &host_resolver,
      net::ProxyService::CreateDirect(),
      new net::SSLConfigServiceDefaults,
      NULL,
      NULL,
      NULL);

  scoped_ptr<net::HttpTransaction> trans;
  int rv = factory.CreateTransaction(&trans);
  EXPECT_EQ(net::OK, rv);
  EXPECT_TRUE(trans.get() != NULL);
}

TEST_F(HttpNetworkLayerTest, Suspend) {
  MockHostResolver host_resolver;
  net::HttpNetworkLayer factory(
      NULL,
      &host_resolver,
      net::ProxyService::CreateDirect(),
      new net::SSLConfigServiceDefaults,
      NULL,
      NULL,
      NULL);

  scoped_ptr<net::HttpTransaction> trans;
  int rv = factory.CreateTransaction(&trans);
  EXPECT_EQ(net::OK, rv);

  trans.reset();

  factory.Suspend(true);

  rv = factory.CreateTransaction(&trans);
  EXPECT_EQ(net::ERR_NETWORK_IO_SUSPENDED, rv);

  ASSERT_TRUE(trans == NULL);

  factory.Suspend(false);

  rv = factory.CreateTransaction(&trans);
  EXPECT_EQ(net::OK, rv);
}

TEST_F(HttpNetworkLayerTest, GET) {
  net::MockClientSocketFactory mock_socket_factory;
  net::MockRead data_reads[] = {
    net::MockRead("HTTP/1.0 200 OK\r\n\r\n"),
    net::MockRead("hello world"),
    net::MockRead(false, net::OK),
  };
  net::MockWrite data_writes[] = {
    net::MockWrite("GET / HTTP/1.1\r\n"
                   "Host: www.google.com\r\n"
                   "Connection: keep-alive\r\n"
                   "User-Agent: Foo/1.0\r\n\r\n"),
  };
  net::StaticSocketDataProvider data(data_reads, arraysize(data_reads),
                                     data_writes, arraysize(data_reads));
  mock_socket_factory.AddSocketDataProvider(&data);

  MockHostResolver host_resolver;
  net::HttpNetworkLayer factory(
      &mock_socket_factory,
      &host_resolver,
      net::ProxyService::CreateDirect(),
      new net::SSLConfigServiceDefaults,
      NULL,
      NULL,
      NULL);

  TestCompletionCallback callback;

  scoped_ptr<net::HttpTransaction> trans;
  int rv = factory.CreateTransaction(&trans);
  EXPECT_EQ(net::OK, rv);

  net::HttpRequestInfo request_info;
  request_info.url = GURL("http://www.google.com/");
  request_info.method = "GET";
  request_info.extra_headers.SetHeader(net::HttpRequestHeaders::kUserAgent,
                                       "Foo/1.0");
  request_info.load_flags = net::LOAD_NORMAL;

  rv = trans->Start(&request_info, &callback, net::BoundNetLog());
  if (rv == net::ERR_IO_PENDING)
    rv = callback.WaitForResult();
  ASSERT_EQ(net::OK, rv);

  std::string contents;
  rv = ReadTransaction(trans.get(), &contents);
  EXPECT_EQ(net::OK, rv);
  EXPECT_EQ("hello world", contents);
}

}  // namespace

}  // namespace net