| 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 |