OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "gtest/gtest.h" | 5 #include "gtest/gtest.h" |
6 | 6 |
7 #include "base/memory/ref_counted.h" | 7 #include "base/memory/ref_counted.h" |
8 #include "base/scoped_comptr_win.h" | 8 #include "base/win/scoped_comptr.h" |
9 #include "chrome_frame/urlmon_upload_data_stream.h" | 9 #include "chrome_frame/urlmon_upload_data_stream.h" |
10 | 10 |
11 TEST(UrlmonUploadDataStreamTest, TestBasicRead) { | 11 TEST(UrlmonUploadDataStreamTest, TestBasicRead) { |
12 char random_string[] = "some random data, no really this totally random"; | 12 char random_string[] = "some random data, no really this totally random"; |
13 int random_string_length = strlen(random_string); | 13 int random_string_length = strlen(random_string); |
14 scoped_refptr<net::UploadData> upload_data = new net::UploadData(); | 14 scoped_refptr<net::UploadData> upload_data = new net::UploadData(); |
15 upload_data->AppendBytes(random_string, random_string_length); | 15 upload_data->AppendBytes(random_string, random_string_length); |
16 | 16 |
17 CComObject<UrlmonUploadDataStream>* upload_stream = NULL; | 17 CComObject<UrlmonUploadDataStream>* upload_stream = NULL; |
18 HRESULT hr = | 18 HRESULT hr = |
19 CComObject<UrlmonUploadDataStream>::CreateInstance(&upload_stream); | 19 CComObject<UrlmonUploadDataStream>::CreateInstance(&upload_stream); |
20 ASSERT_TRUE(SUCCEEDED(hr)); | 20 ASSERT_TRUE(SUCCEEDED(hr)); |
21 | 21 |
22 upload_stream->Initialize(upload_data.get()); | 22 upload_stream->Initialize(upload_data.get()); |
23 ScopedComPtr<IStream> upload_istream(upload_stream); | 23 base::win::ScopedComPtr<IStream> upload_istream(upload_stream); |
24 | 24 |
25 char buffer[500]; | 25 char buffer[500]; |
26 memset(buffer, 0, 500); | 26 memset(buffer, 0, 500); |
27 ULONG bytes_read = 0; | 27 ULONG bytes_read = 0; |
28 hr = upload_istream->Read(buffer, 500, &bytes_read); | 28 hr = upload_istream->Read(buffer, 500, &bytes_read); |
29 | 29 |
30 EXPECT_TRUE(SUCCEEDED(hr)); | 30 EXPECT_TRUE(SUCCEEDED(hr)); |
31 EXPECT_EQ(bytes_read, random_string_length); | 31 EXPECT_EQ(bytes_read, random_string_length); |
32 EXPECT_TRUE(strcmp(buffer, random_string) == 0); | 32 EXPECT_TRUE(strcmp(buffer, random_string) == 0); |
33 | 33 |
(...skipping 14 matching lines...) Expand all Loading... |
48 | 48 |
49 scoped_refptr<net::UploadData> upload_data = new net::UploadData(); | 49 scoped_refptr<net::UploadData> upload_data = new net::UploadData(); |
50 upload_data->AppendBytes(big_buffer, kBigBufferLength); | 50 upload_data->AppendBytes(big_buffer, kBigBufferLength); |
51 | 51 |
52 CComObject<UrlmonUploadDataStream>* upload_stream = NULL; | 52 CComObject<UrlmonUploadDataStream>* upload_stream = NULL; |
53 HRESULT hr = | 53 HRESULT hr = |
54 CComObject<UrlmonUploadDataStream>::CreateInstance(&upload_stream); | 54 CComObject<UrlmonUploadDataStream>::CreateInstance(&upload_stream); |
55 ASSERT_TRUE(SUCCEEDED(hr)); | 55 ASSERT_TRUE(SUCCEEDED(hr)); |
56 | 56 |
57 upload_stream->Initialize(upload_data.get()); | 57 upload_stream->Initialize(upload_data.get()); |
58 ScopedComPtr<IStream> upload_istream(upload_stream); | 58 base::win::ScopedComPtr<IStream> upload_istream(upload_stream); |
59 | 59 |
60 char big_rcv_buffer[kBigBufferLength]; | 60 char big_rcv_buffer[kBigBufferLength]; |
61 int write_pos = 0; | 61 int write_pos = 0; |
62 ULONG bytes_read = 0; | 62 ULONG bytes_read = 0; |
63 hr = E_UNEXPECTED; | 63 hr = E_UNEXPECTED; |
64 | 64 |
65 while ((hr = upload_istream->Read(&big_rcv_buffer[write_pos], | 65 while ((hr = upload_istream->Read(&big_rcv_buffer[write_pos], |
66 kBigBufferLength, | 66 kBigBufferLength, |
67 &bytes_read)) != S_FALSE) { | 67 &bytes_read)) != S_FALSE) { |
68 EXPECT_TRUE(SUCCEEDED(hr)); | 68 EXPECT_TRUE(SUCCEEDED(hr)); |
(...skipping 13 matching lines...) Expand all Loading... |
82 int random_string_length = strlen(random_string); | 82 int random_string_length = strlen(random_string); |
83 scoped_refptr<net::UploadData> upload_data = new net::UploadData(); | 83 scoped_refptr<net::UploadData> upload_data = new net::UploadData(); |
84 upload_data->AppendBytes(random_string, random_string_length); | 84 upload_data->AppendBytes(random_string, random_string_length); |
85 | 85 |
86 CComObject<UrlmonUploadDataStream>* upload_stream = NULL; | 86 CComObject<UrlmonUploadDataStream>* upload_stream = NULL; |
87 HRESULT hr = | 87 HRESULT hr = |
88 CComObject<UrlmonUploadDataStream>::CreateInstance(&upload_stream); | 88 CComObject<UrlmonUploadDataStream>::CreateInstance(&upload_stream); |
89 ASSERT_TRUE(SUCCEEDED(hr)); | 89 ASSERT_TRUE(SUCCEEDED(hr)); |
90 | 90 |
91 upload_stream->Initialize(upload_data.get()); | 91 upload_stream->Initialize(upload_data.get()); |
92 ScopedComPtr<IStream> upload_istream(upload_stream); | 92 base::win::ScopedComPtr<IStream> upload_istream(upload_stream); |
93 | 93 |
94 STATSTG statstg; | 94 STATSTG statstg; |
95 hr = upload_stream->Stat(&statstg, STATFLAG_NONAME); | 95 hr = upload_stream->Stat(&statstg, STATFLAG_NONAME); |
96 EXPECT_TRUE(SUCCEEDED(hr)); | 96 EXPECT_TRUE(SUCCEEDED(hr)); |
97 EXPECT_EQ(static_cast<LONGLONG>(random_string_length), | 97 EXPECT_EQ(static_cast<LONGLONG>(random_string_length), |
98 statstg.cbSize.QuadPart); | 98 statstg.cbSize.QuadPart); |
99 } | 99 } |
100 | 100 |
101 TEST(UrlmonUploadDataStreamTest, TestRepeatedRead) { | 101 TEST(UrlmonUploadDataStreamTest, TestRepeatedRead) { |
102 char random_string[] = "some random data, no really this totally random"; | 102 char random_string[] = "some random data, no really this totally random"; |
103 int random_string_length = strlen(random_string); | 103 int random_string_length = strlen(random_string); |
104 scoped_refptr<net::UploadData> upload_data = new net::UploadData(); | 104 scoped_refptr<net::UploadData> upload_data = new net::UploadData(); |
105 upload_data->AppendBytes(random_string, random_string_length); | 105 upload_data->AppendBytes(random_string, random_string_length); |
106 | 106 |
107 CComObject<UrlmonUploadDataStream>* upload_stream = NULL; | 107 CComObject<UrlmonUploadDataStream>* upload_stream = NULL; |
108 HRESULT hr = | 108 HRESULT hr = |
109 CComObject<UrlmonUploadDataStream>::CreateInstance(&upload_stream); | 109 CComObject<UrlmonUploadDataStream>::CreateInstance(&upload_stream); |
110 ASSERT_TRUE(SUCCEEDED(hr)); | 110 ASSERT_TRUE(SUCCEEDED(hr)); |
111 | 111 |
112 upload_stream->Initialize(upload_data.get()); | 112 upload_stream->Initialize(upload_data.get()); |
113 ScopedComPtr<IStream> upload_istream(upload_stream); | 113 base::win::ScopedComPtr<IStream> upload_istream(upload_stream); |
114 | 114 |
115 char buffer[500]; | 115 char buffer[500]; |
116 memset(buffer, 0, 500); | 116 memset(buffer, 0, 500); |
117 ULONG bytes_read = 0; | 117 ULONG bytes_read = 0; |
118 hr = upload_istream->Read(buffer, 500, &bytes_read); | 118 hr = upload_istream->Read(buffer, 500, &bytes_read); |
119 | 119 |
120 EXPECT_TRUE(SUCCEEDED(hr)); | 120 EXPECT_TRUE(SUCCEEDED(hr)); |
121 EXPECT_EQ(bytes_read, random_string_length); | 121 EXPECT_EQ(bytes_read, random_string_length); |
122 EXPECT_EQ(0, strcmp(buffer, random_string)); | 122 EXPECT_EQ(0, strcmp(buffer, random_string)); |
123 | 123 |
(...skipping 14 matching lines...) Expand all Loading... |
138 int random_string_length = strlen(random_string); | 138 int random_string_length = strlen(random_string); |
139 scoped_refptr<net::UploadData> upload_data = new net::UploadData(); | 139 scoped_refptr<net::UploadData> upload_data = new net::UploadData(); |
140 upload_data->AppendBytes(random_string, random_string_length); | 140 upload_data->AppendBytes(random_string, random_string_length); |
141 | 141 |
142 CComObject<UrlmonUploadDataStream>* upload_stream = NULL; | 142 CComObject<UrlmonUploadDataStream>* upload_stream = NULL; |
143 HRESULT hr = | 143 HRESULT hr = |
144 CComObject<UrlmonUploadDataStream>::CreateInstance(&upload_stream); | 144 CComObject<UrlmonUploadDataStream>::CreateInstance(&upload_stream); |
145 ASSERT_TRUE(SUCCEEDED(hr)); | 145 ASSERT_TRUE(SUCCEEDED(hr)); |
146 | 146 |
147 upload_stream->Initialize(upload_data.get()); | 147 upload_stream->Initialize(upload_data.get()); |
148 ScopedComPtr<IStream> upload_istream(upload_stream); | 148 base::win::ScopedComPtr<IStream> upload_istream(upload_stream); |
149 | 149 |
150 char buffer[500]; | 150 char buffer[500]; |
151 memset(buffer, 0, 500); | 151 memset(buffer, 0, 500); |
152 ULONG bytes_read = 42; | 152 ULONG bytes_read = 42; |
153 hr = upload_istream->Read(&buffer[0], 0, &bytes_read); | 153 hr = upload_istream->Read(&buffer[0], 0, &bytes_read); |
154 | 154 |
155 EXPECT_EQ(S_OK, hr); | 155 EXPECT_EQ(S_OK, hr); |
156 EXPECT_EQ(0, bytes_read); | 156 EXPECT_EQ(0, bytes_read); |
157 | 157 |
158 char buffer2[500]; | 158 char buffer2[500]; |
159 memset(&buffer2[0], 0, 500); | 159 memset(&buffer2[0], 0, 500); |
160 EXPECT_EQ(0, memcmp(buffer, buffer2, 500)); | 160 EXPECT_EQ(0, memcmp(buffer, buffer2, 500)); |
161 } | 161 } |
162 | 162 |
OLD | NEW |