/*
 * 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 "util/Util.h"

#include <string>

#include "test/Test.h"

using ::android::StringPiece;
using ::testing::Eq;
using ::testing::Ne;
using ::testing::SizeIs;

namespace aapt {

TEST(UtilTest, TrimOnlyWhitespace) {
  const StringPiece trimmed = util::TrimWhitespace("\n        ");
  EXPECT_TRUE(trimmed.empty());
  EXPECT_THAT(trimmed, SizeIs(0u));
}

TEST(UtilTest, StringEndsWith) {
  EXPECT_TRUE(util::EndsWith("hello.xml", ".xml"));
}

TEST(UtilTest, StringStartsWith) {
  EXPECT_TRUE(util::StartsWith("hello.xml", "he"));
}

TEST(UtilTest, StringBuilderSplitEscapeSequence) {
  EXPECT_THAT(util::StringBuilder().Append("this is a new\\").Append("nline.").ToString(),
              Eq("this is a new\nline."));
}

TEST(UtilTest, StringBuilderWhitespaceRemoval) {
  EXPECT_THAT(util::StringBuilder().Append("    hey guys ").Append(" this is so cool ").ToString(),
              Eq("hey guys this is so cool"));
  EXPECT_THAT(
      util::StringBuilder().Append(" \" wow,  so many \t ").Append("spaces. \"what? ").ToString(),
      Eq(" wow,  so many \t spaces. what?"));
  EXPECT_THAT(util::StringBuilder().Append("  where \t ").Append(" \nis the pie?").ToString(),
              Eq("where is the pie?"));
}

TEST(UtilTest, StringBuilderEscaping) {
  EXPECT_THAT(util::StringBuilder()
                  .Append("    hey guys\\n ")
                  .Append(" this \\t is so\\\\ cool ")
                  .ToString(),
              Eq("hey guys\n this \t is so\\ cool"));
  EXPECT_THAT(util::StringBuilder().Append("\\@\\?\\#\\\\\\'").ToString(), Eq("@?#\\\'"));
}

TEST(UtilTest, StringBuilderMisplacedQuote) {
  util::StringBuilder builder;
  EXPECT_FALSE(builder.Append("they're coming!"));
}

TEST(UtilTest, StringBuilderUnicodeCodes) {
  EXPECT_THAT(util::StringBuilder().Append("\\u00AF\\u0AF0 woah").ToString(),
              Eq("\u00AF\u0AF0 woah"));
  EXPECT_FALSE(util::StringBuilder().Append("\\u00 yo"));
}

TEST(UtilTest, StringBuilderPreserveSpaces) {
  EXPECT_THAT(util::StringBuilder(true /*preserve_spaces*/).Append("\"").ToString(), Eq("\""));
}

TEST(UtilTest, TokenizeInput) {
  auto tokenizer = util::Tokenize(StringPiece("this| is|the|end"), '|');
  auto iter = tokenizer.begin();
  ASSERT_THAT(*iter, Eq("this"));
  ++iter;
  ASSERT_THAT(*iter, Eq(" is"));
  ++iter;
  ASSERT_THAT(*iter, Eq("the"));
  ++iter;
  ASSERT_THAT(*iter, Eq("end"));
  ++iter;
  ASSERT_THAT(iter, Eq(tokenizer.end()));
}

TEST(UtilTest, TokenizeEmptyString) {
  auto tokenizer = util::Tokenize(StringPiece(""), '|');
  auto iter = tokenizer.begin();
  ASSERT_THAT(iter, Ne(tokenizer.end()));
  ASSERT_THAT(*iter, Eq(StringPiece()));
  ++iter;
  ASSERT_THAT(iter, Eq(tokenizer.end()));
}

TEST(UtilTest, TokenizeAtEnd) {
  auto tokenizer = util::Tokenize(StringPiece("one."), '.');
  auto iter = tokenizer.begin();
  ASSERT_THAT(*iter, Eq("one"));
  ++iter;
  ASSERT_THAT(iter, Ne(tokenizer.end()));
  ASSERT_THAT(*iter, Eq(StringPiece()));
}

TEST(UtilTest, IsJavaClassName) {
  EXPECT_TRUE(util::IsJavaClassName("android.test.Class"));
  EXPECT_TRUE(util::IsJavaClassName("android.test.Class$Inner"));
  EXPECT_TRUE(util::IsJavaClassName("android_test.test.Class"));
  EXPECT_TRUE(util::IsJavaClassName("_android_.test._Class_"));
  EXPECT_FALSE(util::IsJavaClassName("android.test.$Inner"));
  EXPECT_FALSE(util::IsJavaClassName("android.test.Inner$"));
  EXPECT_FALSE(util::IsJavaClassName(".test.Class"));
  EXPECT_FALSE(util::IsJavaClassName("android"));
}

TEST(UtilTest, IsJavaPackageName) {
  EXPECT_TRUE(util::IsJavaPackageName("android"));
  EXPECT_TRUE(util::IsJavaPackageName("android.test"));
  EXPECT_TRUE(util::IsJavaPackageName("android.test_thing"));
  EXPECT_FALSE(util::IsJavaPackageName("_android"));
  EXPECT_FALSE(util::IsJavaPackageName("android_"));
  EXPECT_FALSE(util::IsJavaPackageName("android."));
  EXPECT_FALSE(util::IsJavaPackageName(".android"));
  EXPECT_FALSE(util::IsJavaPackageName("android._test"));
  EXPECT_FALSE(util::IsJavaPackageName(".."));
}

TEST(UtilTest, FullyQualifiedClassName) {
  Maybe<std::string> res = util::GetFullyQualifiedClassName("android", ".asdf");
  AAPT_ASSERT_TRUE(res);
  EXPECT_EQ(res.value(), "android.asdf");

  res = util::GetFullyQualifiedClassName("android", ".a.b");
  AAPT_ASSERT_TRUE(res);
  EXPECT_EQ(res.value(), "android.a.b");

  res = util::GetFullyQualifiedClassName("android", "a.b");
  AAPT_ASSERT_TRUE(res);
  EXPECT_EQ(res.value(), "a.b");

  res = util::GetFullyQualifiedClassName("", "a.b");
  AAPT_ASSERT_TRUE(res);
  EXPECT_EQ(res.value(), "a.b");

  res = util::GetFullyQualifiedClassName("android", "Class");
  AAPT_ASSERT_TRUE(res);
  EXPECT_EQ(res.value(), "android.Class");

  res = util::GetFullyQualifiedClassName("", "");
  AAPT_ASSERT_FALSE(res);

  res = util::GetFullyQualifiedClassName("android", "./Apple");
  AAPT_ASSERT_FALSE(res);
}

TEST(UtilTest, ExtractResourcePathComponents) {
  StringPiece prefix, entry, suffix;
  ASSERT_TRUE(util::ExtractResFilePathParts("res/xml-sw600dp/entry.xml",
                                            &prefix, &entry, &suffix));
  EXPECT_EQ(prefix, "res/xml-sw600dp/");
  EXPECT_EQ(entry, "entry");
  EXPECT_EQ(suffix, ".xml");

  ASSERT_TRUE(util::ExtractResFilePathParts("res/xml-sw600dp/entry.9.png",
                                            &prefix, &entry, &suffix));

  EXPECT_EQ(prefix, "res/xml-sw600dp/");
  EXPECT_EQ(entry, "entry");
  EXPECT_EQ(suffix, ".9.png");

  EXPECT_FALSE(util::ExtractResFilePathParts("AndroidManifest.xml", &prefix,
                                             &entry, &suffix));
  EXPECT_FALSE(
      util::ExtractResFilePathParts("res/.xml", &prefix, &entry, &suffix));

  ASSERT_TRUE(
      util::ExtractResFilePathParts("res//.", &prefix, &entry, &suffix));
  EXPECT_EQ(prefix, "res//");
  EXPECT_EQ(entry, "");
  EXPECT_EQ(suffix, ".");
}

TEST(UtilTest, VerifyJavaStringFormat) {
  ASSERT_TRUE(util::VerifyJavaStringFormat("%09.34f"));
  ASSERT_TRUE(util::VerifyJavaStringFormat("%9$.34f %8$"));
  ASSERT_TRUE(util::VerifyJavaStringFormat("%% %%"));
  ASSERT_FALSE(util::VerifyJavaStringFormat("%09$f %f"));
  ASSERT_FALSE(util::VerifyJavaStringFormat("%09f %08s"));
}

}  // namespace aapt