// Copyright 2018 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <string>

#include "mojo/public/cpp/base/shared_memory_mojom_traits.h"
#include "mojo/public/cpp/test_support/test_utils.h"
#include "mojo/public/mojom/base/shared_memory.mojom.h"
#include "testing/gtest/include/gtest/gtest.h"

TEST(SharedMemoryMojomTest, ReadOnly) {
  auto region = base::ReadOnlySharedMemoryRegion::Create(64);
  const std::string kTestData = "Hello, world!";
  memcpy(region.mapping.memory(), kTestData.data(), kTestData.size());

  base::ReadOnlySharedMemoryRegion read_only_out;
  ASSERT_TRUE(mojo::test::SerializeAndDeserialize<
              mojo_base::mojom::ReadOnlySharedMemoryRegion>(&region.region,
                                                            &read_only_out));
  base::ReadOnlySharedMemoryMapping mapping = read_only_out.Map();
  EXPECT_EQ(0, memcmp(mapping.memory(), kTestData.data(), kTestData.size()));
}

TEST(SharedMemoryMojomTest, Writable) {
  auto region = base::WritableSharedMemoryRegion::Create(64);
  auto mapping = region.Map();
  const std::string kTestData = "Hello, world!";
  memcpy(mapping.memory(), kTestData.data(), kTestData.size());

  base::WritableSharedMemoryRegion writable_out;
  ASSERT_TRUE(mojo::test::SerializeAndDeserialize<
              mojo_base::mojom::WritableSharedMemoryRegion>(&region,
                                                            &writable_out));

  mapping = writable_out.Map();
  EXPECT_EQ(0, memcmp(mapping.memory(), kTestData.data(), kTestData.size()));
}

TEST(SharedMemoryMojomTest, Unsafe) {
  auto region = base::UnsafeSharedMemoryRegion::Create(64);
  auto mapping = region.Map();
  const std::string kTestData = "Hello, world!";
  memcpy(mapping.memory(), kTestData.data(), kTestData.size());

  base::UnsafeSharedMemoryRegion unsafe_out;
  ASSERT_TRUE(
      mojo::test::SerializeAndDeserialize<
          mojo_base::mojom::UnsafeSharedMemoryRegion>(&region, &unsafe_out));

  mapping = unsafe_out.Map();
  EXPECT_EQ(0, memcmp(mapping.memory(), kTestData.data(), kTestData.size()));
}