tools/aapt2/StringPool_test.cpp - platform/frameworks/base - Git at Google

 /*
  * Copyright (C) 2015 The Android Open Source Project
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */

 #include "StringPool.h"

 #include <string>

 #include "androidfw/StringPiece.h"

 #include "test/Test.h"
 #include "util/Util.h"

 using android::StringPiece;
 using android::StringPiece16;

 namespace aapt {

 TEST(StringPoolTest, InsertOneString) {
   StringPool pool;

   StringPool::Ref ref = pool.MakeRef("wut");
   EXPECT_EQ(*ref, "wut");
 }

 TEST(StringPoolTest, InsertTwoUniqueStrings) {
   StringPool pool;

   StringPool::Ref ref = pool.MakeRef("wut");
   StringPool::Ref ref2 = pool.MakeRef("hey");

   EXPECT_EQ(*ref, "wut");
   EXPECT_EQ(*ref2, "hey");
 }

 TEST(StringPoolTest, DoNotInsertNewDuplicateString) {
   StringPool pool;

   StringPool::Ref ref = pool.MakeRef("wut");
   StringPool::Ref ref2 = pool.MakeRef("wut");

   EXPECT_EQ(*ref, "wut");
   EXPECT_EQ(*ref2, "wut");
   EXPECT_EQ(1u, pool.size());
 }

 TEST(StringPoolTest, MaintainInsertionOrderIndex) {
   StringPool pool;

   StringPool::Ref ref = pool.MakeRef("z");
   StringPool::Ref ref2 = pool.MakeRef("a");
   StringPool::Ref ref3 = pool.MakeRef("m");

   EXPECT_EQ(0u, ref.index());
   EXPECT_EQ(1u, ref2.index());
   EXPECT_EQ(2u, ref3.index());
 }

 TEST(StringPoolTest, PruneStringsWithNoReferences) {
   StringPool pool;

   StringPool::Ref refA = pool.MakeRef("foo");
   {
     StringPool::Ref ref = pool.MakeRef("wut");
     EXPECT_EQ(*ref, "wut");
     EXPECT_EQ(2u, pool.size());
   }
   StringPool::Ref refB = pool.MakeRef("bar");

   EXPECT_EQ(3u, pool.size());
   pool.Prune();
   EXPECT_EQ(2u, pool.size());
   StringPool::const_iterator iter = begin(pool);
   EXPECT_EQ((*iter)->value, "foo");
   EXPECT_LT((*iter)->index, 2u);
   ++iter;
   EXPECT_EQ((*iter)->value, "bar");
   EXPECT_LT((*iter)->index, 2u);
 }

 TEST(StringPoolTest, SortAndMaintainIndexesInReferences) {
   StringPool pool;

   StringPool::Ref ref = pool.MakeRef("z");
   StringPool::StyleRef ref2 = pool.MakeRef(StyleString{{"a"}});
   StringPool::Ref ref3 = pool.MakeRef("m");

   EXPECT_EQ(*ref, "z");
   EXPECT_EQ(0u, ref.index());

   EXPECT_EQ(*(ref2->str), "a");
   EXPECT_EQ(1u, ref2.index());

   EXPECT_EQ(*ref3, "m");
   EXPECT_EQ(2u, ref3.index());

   pool.Sort([](const StringPool::Entry& a, const StringPool::Entry& b) -> bool {
     return a.value < b.value;
   });

   EXPECT_EQ(*ref, "z");
   EXPECT_EQ(2u, ref.index());

   EXPECT_EQ(*(ref2->str), "a");
   EXPECT_EQ(0u, ref2.index());

   EXPECT_EQ(*ref3, "m");
   EXPECT_EQ(1u, ref3.index());
 }

 TEST(StringPoolTest, SortAndStillDedupe) {
   StringPool pool;

   StringPool::Ref ref = pool.MakeRef("z");
   StringPool::Ref ref2 = pool.MakeRef("a");
   StringPool::Ref ref3 = pool.MakeRef("m");

   pool.Sort([](const StringPool::Entry& a, const StringPool::Entry& b) -> bool {
     return a.value < b.value;
   });

   StringPool::Ref ref4 = pool.MakeRef("z");
   StringPool::Ref ref5 = pool.MakeRef("a");
   StringPool::Ref ref6 = pool.MakeRef("m");

   EXPECT_EQ(ref4.index(), ref.index());
   EXPECT_EQ(ref5.index(), ref2.index());
   EXPECT_EQ(ref6.index(), ref3.index());
 }

 TEST(StringPoolTest, AddStyles) {
   StringPool pool;

   StyleString str{{"android"}, {Span{{"b"}, 2, 6}}};

   StringPool::StyleRef ref = pool.MakeRef(str);

   EXPECT_EQ(0u, ref.index());
   EXPECT_EQ(std::string("android"), *(ref->str));
   ASSERT_EQ(1u, ref->spans.size());

   const StringPool::Span& span = ref->spans.front();
   EXPECT_EQ(*(span.name), "b");
   EXPECT_EQ(2u, span.first_char);
   EXPECT_EQ(6u, span.last_char);
 }

 TEST(StringPoolTest, DoNotDedupeStyleWithSameStringAsNonStyle) {
   StringPool pool;

   StringPool::Ref ref = pool.MakeRef("android");

   StyleString str{{"android"}};
   StringPool::StyleRef styleRef = pool.MakeRef(str);

   EXPECT_NE(ref.index(), styleRef.index());
 }

 TEST(StringPoolTest, FlattenEmptyStringPoolUtf8) {
   using namespace android;  // For NO_ERROR on Windows.

   StringPool pool;
   BigBuffer buffer(1024);
   StringPool::FlattenUtf8(&buffer, pool);

   std::unique_ptr<uint8_t[]> data = util::Copy(buffer);
   ResStringPool test;
   ASSERT_EQ(test.setTo(data.get(), buffer.size()), NO_ERROR);
 }

 TEST(StringPoolTest, FlattenOddCharactersUtf16) {
   using namespace android;  // For NO_ERROR on Windows.

   StringPool pool;
   pool.MakeRef("\u093f");
   BigBuffer buffer(1024);
   StringPool::FlattenUtf16(&buffer, pool);

   std::unique_ptr<uint8_t[]> data = util::Copy(buffer);
   ResStringPool test;
   ASSERT_EQ(test.setTo(data.get(), buffer.size()), NO_ERROR);
   size_t len = 0;
   const char16_t* str = test.stringAt(0, &len);
   EXPECT_EQ(1u, len);
   EXPECT_EQ(u'\u093f', *str);
   EXPECT_EQ(0u, str[1]);
 }

 constexpr const char* sLongString =
     "バッテリーを長持ちさせるため、バッテリーセーバーは端末のパフォーマンスを抑"
     "え、バイブレーション、位置情報サービス、大半のバックグラウンドデータを制限"
     "します。メール、SMSや、同期を使 "
     "用するその他のアプリは、起動しても更新されないことがあります。バッテリーセ"
     "ーバーは端末の充電中は自動的にOFFになります。";

 TEST(StringPoolTest, Flatten) {
   using namespace android;  // For NO_ERROR on Windows.

   StringPool pool;

   StringPool::Ref ref1 = pool.MakeRef("hello");
   StringPool::Ref ref2 = pool.MakeRef("goodbye");
   StringPool::Ref ref3 = pool.MakeRef(sLongString);
   StringPool::Ref ref4 = pool.MakeRef("");
   StringPool::StyleRef ref5 = pool.MakeRef(
       StyleString{{"style"}, {Span{{"b"}, 0, 1}, Span{{"i"}, 2, 3}}});

   EXPECT_EQ(0u, ref1.index());
   EXPECT_EQ(1u, ref2.index());
   EXPECT_EQ(2u, ref3.index());
   EXPECT_EQ(3u, ref4.index());
   EXPECT_EQ(4u, ref5.index());

   BigBuffer buffers[2] = {BigBuffer(1024), BigBuffer(1024)};
   StringPool::FlattenUtf8(&buffers[0], pool);
   StringPool::FlattenUtf16(&buffers[1], pool);

   // Test both UTF-8 and UTF-16 buffers.
   for (const BigBuffer& buffer : buffers) {
     std::unique_ptr<uint8_t[]> data = util::Copy(buffer);

     ResStringPool test;
     ASSERT_EQ(test.setTo(data.get(), buffer.size()), NO_ERROR);

     EXPECT_EQ(std::string("hello"), util::GetString(test, 0));
     EXPECT_EQ(StringPiece16(u"hello"), util::GetString16(test, 0));

     EXPECT_EQ(std::string("goodbye"), util::GetString(test, 1));
     EXPECT_EQ(StringPiece16(u"goodbye"), util::GetString16(test, 1));

     EXPECT_EQ(StringPiece(sLongString), util::GetString(test, 2));
     EXPECT_EQ(util::Utf8ToUtf16(sLongString), util::GetString16(test, 2).to_string());

     size_t len;
     EXPECT_TRUE(test.stringAt(3, &len) != nullptr ||
                 test.string8At(3, &len) != nullptr);

     EXPECT_EQ(std::string("style"), util::GetString(test, 4));
     EXPECT_EQ(StringPiece16(u"style"), util::GetString16(test, 4));

     const ResStringPool_span* span = test.styleAt(4);
     ASSERT_NE(nullptr, span);
     EXPECT_EQ(std::string("b"), util::GetString(test, span->name.index));
     EXPECT_EQ(StringPiece16(u"b"), util::GetString16(test, span->name.index));
     EXPECT_EQ(0u, span->firstChar);
     EXPECT_EQ(1u, span->lastChar);
     span++;

     ASSERT_NE(ResStringPool_span::END, span->name.index);
     EXPECT_EQ(std::string("i"), util::GetString(test, span->name.index));
     EXPECT_EQ(StringPiece16(u"i"), util::GetString16(test, span->name.index));
     EXPECT_EQ(2u, span->firstChar);
     EXPECT_EQ(3u, span->lastChar);
     span++;

     EXPECT_EQ(ResStringPool_span::END, span->name.index);
   }
 }

 }  // namespace aapt
	/*
	* Copyright (C) 2015 The Android Open Source Project
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	#include "StringPool.h"

	#include <string>

	#include "androidfw/StringPiece.h"

	#include "test/Test.h"
	#include "util/Util.h"

	using android::StringPiece;
	using android::StringPiece16;

	namespace aapt {

	TEST(StringPoolTest, InsertOneString) {
	StringPool pool;

	StringPool::Ref ref = pool.MakeRef("wut");
	EXPECT_EQ(*ref, "wut");
	}

	TEST(StringPoolTest, InsertTwoUniqueStrings) {
	StringPool pool;

	StringPool::Ref ref = pool.MakeRef("wut");
	StringPool::Ref ref2 = pool.MakeRef("hey");

	EXPECT_EQ(*ref, "wut");
	EXPECT_EQ(*ref2, "hey");
	}

	TEST(StringPoolTest, DoNotInsertNewDuplicateString) {
	StringPool pool;

	StringPool::Ref ref = pool.MakeRef("wut");
	StringPool::Ref ref2 = pool.MakeRef("wut");

	EXPECT_EQ(*ref, "wut");
	EXPECT_EQ(*ref2, "wut");
	EXPECT_EQ(1u, pool.size());
	}

	TEST(StringPoolTest, MaintainInsertionOrderIndex) {
	StringPool pool;

	StringPool::Ref ref = pool.MakeRef("z");
	StringPool::Ref ref2 = pool.MakeRef("a");
	StringPool::Ref ref3 = pool.MakeRef("m");

	EXPECT_EQ(0u, ref.index());
	EXPECT_EQ(1u, ref2.index());
	EXPECT_EQ(2u, ref3.index());
	}

	TEST(StringPoolTest, PruneStringsWithNoReferences) {
	StringPool pool;

	StringPool::Ref refA = pool.MakeRef("foo");
	{
	StringPool::Ref ref = pool.MakeRef("wut");
	EXPECT_EQ(*ref, "wut");
	EXPECT_EQ(2u, pool.size());
	}
	StringPool::Ref refB = pool.MakeRef("bar");

	EXPECT_EQ(3u, pool.size());
	pool.Prune();
	EXPECT_EQ(2u, pool.size());
	StringPool::const_iterator iter = begin(pool);
	EXPECT_EQ((*iter)->value, "foo");
	EXPECT_LT((*iter)->index, 2u);
	++iter;
	EXPECT_EQ((*iter)->value, "bar");
	EXPECT_LT((*iter)->index, 2u);
	}

	TEST(StringPoolTest, SortAndMaintainIndexesInReferences) {
	StringPool pool;

	StringPool::Ref ref = pool.MakeRef("z");
	StringPool::StyleRef ref2 = pool.MakeRef(StyleString{{"a"}});
	StringPool::Ref ref3 = pool.MakeRef("m");

	EXPECT_EQ(*ref, "z");
	EXPECT_EQ(0u, ref.index());

	EXPECT_EQ(*(ref2->str), "a");
	EXPECT_EQ(1u, ref2.index());

	EXPECT_EQ(*ref3, "m");
	EXPECT_EQ(2u, ref3.index());

	pool.Sort([](const StringPool::Entry& a, const StringPool::Entry& b) -> bool {
	return a.value < b.value;
	});

	EXPECT_EQ(*ref, "z");
	EXPECT_EQ(2u, ref.index());

	EXPECT_EQ(*(ref2->str), "a");
	EXPECT_EQ(0u, ref2.index());

	EXPECT_EQ(*ref3, "m");
	EXPECT_EQ(1u, ref3.index());
	}

	TEST(StringPoolTest, SortAndStillDedupe) {
	StringPool pool;

	StringPool::Ref ref = pool.MakeRef("z");
	StringPool::Ref ref2 = pool.MakeRef("a");
	StringPool::Ref ref3 = pool.MakeRef("m");

	pool.Sort([](const StringPool::Entry& a, const StringPool::Entry& b) -> bool {
	return a.value < b.value;
	});

	StringPool::Ref ref4 = pool.MakeRef("z");
	StringPool::Ref ref5 = pool.MakeRef("a");
	StringPool::Ref ref6 = pool.MakeRef("m");

	EXPECT_EQ(ref4.index(), ref.index());
	EXPECT_EQ(ref5.index(), ref2.index());
	EXPECT_EQ(ref6.index(), ref3.index());
	}

	TEST(StringPoolTest, AddStyles) {
	StringPool pool;

	StyleString str{{"android"}, {Span{{"b"}, 2, 6}}};

	StringPool::StyleRef ref = pool.MakeRef(str);

	EXPECT_EQ(0u, ref.index());
	EXPECT_EQ(std::string("android"), *(ref->str));
	ASSERT_EQ(1u, ref->spans.size());

	const StringPool::Span& span = ref->spans.front();
	EXPECT_EQ(*(span.name), "b");
	EXPECT_EQ(2u, span.first_char);
	EXPECT_EQ(6u, span.last_char);
	}

	TEST(StringPoolTest, DoNotDedupeStyleWithSameStringAsNonStyle) {
	StringPool pool;

	StringPool::Ref ref = pool.MakeRef("android");

	StyleString str{{"android"}};
	StringPool::StyleRef styleRef = pool.MakeRef(str);

	EXPECT_NE(ref.index(), styleRef.index());
	}

	TEST(StringPoolTest, FlattenEmptyStringPoolUtf8) {
	using namespace android; // For NO_ERROR on Windows.

	StringPool pool;
	BigBuffer buffer(1024);
	StringPool::FlattenUtf8(&buffer, pool);

	std::unique_ptr<uint8_t[]> data = util::Copy(buffer);
	ResStringPool test;
	ASSERT_EQ(test.setTo(data.get(), buffer.size()), NO_ERROR);
	}

	TEST(StringPoolTest, FlattenOddCharactersUtf16) {
	using namespace android; // For NO_ERROR on Windows.

	StringPool pool;
	pool.MakeRef("\u093f");
	BigBuffer buffer(1024);
	StringPool::FlattenUtf16(&buffer, pool);

	std::unique_ptr<uint8_t[]> data = util::Copy(buffer);
	ResStringPool test;
	ASSERT_EQ(test.setTo(data.get(), buffer.size()), NO_ERROR);
	size_t len = 0;
	const char16_t* str = test.stringAt(0, &len);
	EXPECT_EQ(1u, len);
	EXPECT_EQ(u'\u093f', *str);
	EXPECT_EQ(0u, str[1]);
	}

	constexpr const char* sLongString =
	"バッテリーを長持ちさせるため、バッテリーセーバーは端末のパフォーマンスを抑"
	"え、バイブレーション、位置情報サービス、大半のバックグラウンドデータを制限"
	"します。メール、SMSや、同期を使 "
	"用するその他のアプリは、起動しても更新されないことがあります。バッテリーセ"
	"ーバーは端末の充電中は自動的にOFFになります。";

	TEST(StringPoolTest, Flatten) {
	using namespace android; // For NO_ERROR on Windows.

	StringPool pool;

	StringPool::Ref ref1 = pool.MakeRef("hello");
	StringPool::Ref ref2 = pool.MakeRef("goodbye");
	StringPool::Ref ref3 = pool.MakeRef(sLongString);
	StringPool::Ref ref4 = pool.MakeRef("");
	StringPool::StyleRef ref5 = pool.MakeRef(
	StyleString{{"style"}, {Span{{"b"}, 0, 1}, Span{{"i"}, 2, 3}}});

	EXPECT_EQ(0u, ref1.index());
	EXPECT_EQ(1u, ref2.index());
	EXPECT_EQ(2u, ref3.index());
	EXPECT_EQ(3u, ref4.index());
	EXPECT_EQ(4u, ref5.index());

	BigBuffer buffers[2] = {BigBuffer(1024), BigBuffer(1024)};
	StringPool::FlattenUtf8(&buffers[0], pool);
	StringPool::FlattenUtf16(&buffers[1], pool);

	// Test both UTF-8 and UTF-16 buffers.
	for (const BigBuffer& buffer : buffers) {
	std::unique_ptr<uint8_t[]> data = util::Copy(buffer);

	ResStringPool test;
	ASSERT_EQ(test.setTo(data.get(), buffer.size()), NO_ERROR);

	EXPECT_EQ(std::string("hello"), util::GetString(test, 0));
	EXPECT_EQ(StringPiece16(u"hello"), util::GetString16(test, 0));

	EXPECT_EQ(std::string("goodbye"), util::GetString(test, 1));
	EXPECT_EQ(StringPiece16(u"goodbye"), util::GetString16(test, 1));

	EXPECT_EQ(StringPiece(sLongString), util::GetString(test, 2));
	EXPECT_EQ(util::Utf8ToUtf16(sLongString), util::GetString16(test, 2).to_string());

	size_t len;
	EXPECT_TRUE(test.stringAt(3, &len) != nullptr \|\|
	test.string8At(3, &len) != nullptr);

	EXPECT_EQ(std::string("style"), util::GetString(test, 4));
	EXPECT_EQ(StringPiece16(u"style"), util::GetString16(test, 4));

	const ResStringPool_span* span = test.styleAt(4);
	ASSERT_NE(nullptr, span);
	EXPECT_EQ(std::string("b"), util::GetString(test, span->name.index));
	EXPECT_EQ(StringPiece16(u"b"), util::GetString16(test, span->name.index));
	EXPECT_EQ(0u, span->firstChar);
	EXPECT_EQ(1u, span->lastChar);
	span++;

	ASSERT_NE(ResStringPool_span::END, span->name.index);
	EXPECT_EQ(std::string("i"), util::GetString(test, span->name.index));
	EXPECT_EQ(StringPiece16(u"i"), util::GetString16(test, span->name.index));
	EXPECT_EQ(2u, span->firstChar);
	EXPECT_EQ(3u, span->lastChar);
	span++;

	EXPECT_EQ(ResStringPool_span::END, span->name.index);
	}
	}

	} // namespace aapt