Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(386)

Side by Side Diff: chrome_frame/urlmon_upload_data_stream_unittest.cc

Issue 6825055: Include base/win/scoped_comptr.h instead of base/scoped_comptr_win.h. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: revert bad indentation, rebase Created 9 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « chrome_frame/urlmon_moniker.cc ('k') | chrome_frame/urlmon_url_request.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
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
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
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
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
OLDNEW
« no previous file with comments | « chrome_frame/urlmon_moniker.cc ('k') | chrome_frame/urlmon_url_request.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698