// 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 "third_party/blink/renderer/platform/wtf/text/movable_string.h"

#include <vector>

#include "testing/gtest/include/gtest/gtest.h"

namespace WTF {

TEST(MovableStringTest, Simple) {
  MovableString movable_abc(String("abc").ReleaseImpl());

  EXPECT_TRUE(MovableString().IsNull());
  EXPECT_FALSE(movable_abc.IsNull());
  EXPECT_TRUE(movable_abc.Is8Bit());
  EXPECT_EQ(3u, movable_abc.length());
  EXPECT_EQ(3u, movable_abc.CharactersSizeInBytes());

  EXPECT_EQ(String("abc"), movable_abc.ToString());
  MovableString copy = movable_abc;
  EXPECT_EQ(copy.Impl(), movable_abc.Impl());
}

TEST(MovableStringTest, Park) {
  MovableString movable(String("abc").Impl());
  EXPECT_FALSE(movable.Impl()->is_parked());
  EXPECT_TRUE(movable.Impl()->Park());
  EXPECT_TRUE(movable.Impl()->is_parked());

  // Not the only one to have a reference to the string.
  String abc("abc");
  MovableString movable2(abc.Impl());
  EXPECT_FALSE(movable2.Impl()->Park());
  abc = String();
  EXPECT_TRUE(movable2.Impl()->Park());
}

TEST(MovableStringTest, Unpark) {
  MovableString movable(String("abc").Impl());
  EXPECT_FALSE(movable.Impl()->is_parked());
  EXPECT_TRUE(movable.Impl()->Park());
  EXPECT_TRUE(movable.Impl()->is_parked());

  String unparked = movable.ToString();
  EXPECT_EQ(String("abc"), unparked);
  EXPECT_FALSE(movable.Impl()->is_parked());
}

TEST(MovableStringTest, TableSimple) {
  std::vector<char> data(20000, 'a');
  MovableString movable(String(data.data(), data.size()).ReleaseImpl());
  ASSERT_FALSE(movable.Impl()->is_parked());

  auto& table = MovableStringTable::Instance();
  EXPECT_EQ(1u, table.table_.size());

  // Small strings are not in the table.
  MovableString small(String("abc").ReleaseImpl());
  EXPECT_EQ(1u, table.table_.size());

  // No parking as the current state is not "backgrounded".
  table.SetRendererBackgrounded(false);
  ASSERT_FALSE(table.IsRendererBackgrounded());
  table.MaybeParkAll();
  EXPECT_FALSE(movable.Impl()->is_parked());

  table.SetRendererBackgrounded(true);
  ASSERT_TRUE(table.IsRendererBackgrounded());
  table.MaybeParkAll();
  EXPECT_TRUE(movable.Impl()->is_parked());

  // Park and unpark.
  movable.ToString();
  EXPECT_FALSE(movable.Impl()->is_parked());
  table.MaybeParkAll();
  EXPECT_TRUE(movable.Impl()->is_parked());

  // More than one reference, no parking.
  String alive_unparked = movable.ToString();
  table.MaybeParkAll();
  EXPECT_FALSE(movable.Impl()->is_parked());

  // Other reference is dropped, OK to park.
  alive_unparked = String();
  table.MaybeParkAll();
  EXPECT_TRUE(movable.Impl()->is_parked());
}

TEST(MovableStringTest, TableMultiple) {
  std::vector<char> data(20000, 'a');
  MovableString movable(String(data.data(), data.size()).ReleaseImpl());
  MovableString movable2(String(data.data(), data.size()).ReleaseImpl());

  auto& table = MovableStringTable::Instance();
  EXPECT_EQ(2u, table.table_.size());

  movable2 = MovableString();
  EXPECT_EQ(1u, table.table_.size());

  MovableString copy = movable;
  movable = MovableString();
  EXPECT_EQ(1u, table.table_.size());
  copy = MovableString();
  EXPECT_EQ(0u, table.table_.size());

  String str(data.data(), data.size());
  MovableString movable3(str.Impl());
  EXPECT_EQ(1u, table.table_.size());
  // De-duplicated.
  MovableString other_movable3(str.Impl());
  EXPECT_EQ(1u, table.table_.size());
  EXPECT_EQ(movable3.Impl(), other_movable3.Impl());

  // If all the references to a string are in the table, park it.
  str = String();
  table.SetRendererBackgrounded(true);
  ASSERT_TRUE(table.IsRendererBackgrounded());
  table.MaybeParkAll();
  EXPECT_TRUE(movable3.Impl()->is_parked());

  // Only drop it from the table when the last one is gone.
  movable3 = MovableString();
  EXPECT_EQ(1u, table.table_.size());
  other_movable3 = MovableString();
  EXPECT_EQ(0u, table.table_.size());
}

}  // namespace WTF