winrt_storage_util_unittest.cc 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. // Copyright 2017 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "base/win/winrt_storage_util.h"
  5. #include <string.h>
  6. #include <wrl/client.h>
  7. #include <vector>
  8. #include "base/strings/string_util.h"
  9. #include "base/win/core_winrt_util.h"
  10. #include "base/win/scoped_com_initializer.h"
  11. #include "base/win/scoped_hstring.h"
  12. #include "testing/gtest/include/gtest/gtest.h"
  13. namespace base {
  14. namespace win {
  15. TEST(WinrtStorageUtilTest, CreateBufferFromData) {
  16. ScopedCOMInitializer com_initializer(ScopedCOMInitializer::kMTA);
  17. if (!ResolveCoreWinRTDelayload() ||
  18. !ScopedHString::ResolveCoreWinRTStringDelayload()) {
  19. return;
  20. }
  21. const std::vector<uint8_t> data = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
  22. Microsoft::WRL::ComPtr<ABI::Windows::Storage::Streams::IBuffer> buffer;
  23. ASSERT_HRESULT_SUCCEEDED(
  24. CreateIBufferFromData(data.data(), data.size(), &buffer));
  25. uint8_t* p_buffer_data;
  26. uint32_t length;
  27. ASSERT_HRESULT_SUCCEEDED(
  28. GetPointerToBufferData(buffer.Get(), &p_buffer_data, &length));
  29. ASSERT_EQ(data.size(), length);
  30. EXPECT_EQ(0, memcmp(p_buffer_data, data.data(), data.size()));
  31. }
  32. } // namespace win
  33. } // namespace base