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
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/40285824): Remove this and convert code to safer constructs.
#pragma allow_unsafe_buffers
#endif
#include "chrome/browser/webshare/win/fake_buffer.h"
#include <wrl/implements.h>
#include "testing/gtest/include/gtest/gtest-spi.h"
#include "testing/gtest/include/gtest/gtest.h"
using Microsoft::WRL::Make;
namespace webshare {
TEST(FakeBufferTest, Length) {
auto buffer = Make<FakeBuffer>(47);
UINT32 capacity;
ASSERT_HRESULT_SUCCEEDED(buffer->get_Capacity(&capacity));
ASSERT_EQ(capacity, 47u);
UINT32 length;
ASSERT_HRESULT_SUCCEEDED(buffer->get_Length(&length));
ASSERT_EQ(length, 0u);
ASSERT_HRESULT_SUCCEEDED(buffer->put_Length(23));
ASSERT_HRESULT_SUCCEEDED(buffer->get_Length(&length));
ASSERT_EQ(length, 23u);
ASSERT_HRESULT_SUCCEEDED(buffer->put_Length(47));
ASSERT_HRESULT_SUCCEEDED(buffer->get_Length(&length));
ASSERT_EQ(length, 47u);
EXPECT_NONFATAL_FAILURE(ASSERT_HRESULT_FAILED(buffer->put_Length(48)),
"put_Length");
}
TEST(FakeBufferTest, Bytes) {
auto buffer = Make<FakeBuffer>(2);
byte* raw_buffer;
ASSERT_HRESULT_SUCCEEDED(buffer->Buffer(&raw_buffer));
raw_buffer[0] = 'a';
raw_buffer[1] = 'b';
auto buffer2 = buffer;
byte* raw_buffer_2;
ASSERT_HRESULT_SUCCEEDED(buffer2->Buffer(&raw_buffer_2));
ASSERT_EQ(raw_buffer, raw_buffer_2);
ASSERT_EQ(raw_buffer_2[0], 'a');
ASSERT_EQ(raw_buffer_2[1], 'b');
}
} // namespace webshare
|
