// Copyright (c) 2011 The Chromium OS 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 <brillo/data_encoding.h>

#include <algorithm>
#include <numeric>

#include <gtest/gtest.h>

namespace brillo {
namespace data_encoding {

TEST(data_encoding, UrlEncoding) {
  std::string test = "\"http://sample/path/0014.html \"";
  std::string encoded = UrlEncode(test.c_str());
  EXPECT_EQ("%22http%3A%2F%2Fsample%2Fpath%2F0014.html+%22", encoded);
  EXPECT_EQ(test, UrlDecode(encoded.c_str()));

  test = "\"http://sample/path/0014.html \"";
  encoded = UrlEncode(test.c_str(), false);
  EXPECT_EQ("%22http%3A%2F%2Fsample%2Fpath%2F0014.html%20%22", encoded);
  EXPECT_EQ(test, UrlDecode(encoded.c_str()));
}

TEST(data_encoding, WebParamsEncoding) {
  std::string encoded =
      WebParamsEncode({{"q", "test"}, {"path", "/usr/bin"}, {"#", "%"}});
  EXPECT_EQ("q=test&path=%2Fusr%2Fbin&%23=%25", encoded);

  auto params = WebParamsDecode(encoded);
  EXPECT_EQ(3, params.size());
  EXPECT_EQ("q", params[0].first);
  EXPECT_EQ("test", params[0].second);
  EXPECT_EQ("path", params[1].first);
  EXPECT_EQ("/usr/bin", params[1].second);
  EXPECT_EQ("#", params[2].first);
  EXPECT_EQ("%", params[2].second);
}

TEST(data_encoding, Base64Encode) {
  const std::string text1 = "hello world";
  const std::string encoded1 = "aGVsbG8gd29ybGQ=";

  const std::string text2 =
      "Lorem ipsum dolor sit amet, facilisis erat nec aliquam, scelerisque "
      "molestie commodo. Viverra tincidunt integer erat ipsum, integer "
      "molestie, arcu in, sit mauris ac a sed sit etiam.";
  const std::string encoded2 =
      "TG9yZW0gaXBzdW0gZG9sb3Igc2l0IGFtZXQsIGZhY2lsaXNpcyBlcmF0IG5lYyBhbGlxdWF"
      "tLCBzY2VsZXJpc3F1ZSBtb2xlc3RpZSBjb21tb2RvLiBWaXZlcnJhIHRpbmNpZHVudCBpbn"
      "RlZ2VyIGVyYXQgaXBzdW0sIGludGVnZXIgbW9sZXN0aWUsIGFyY3UgaW4sIHNpdCBtYXVya"
      "XMgYWMgYSBzZWQgc2l0IGV0aWFtLg==";

  brillo::Blob data3(256);
  std::iota(data3.begin(), data3.end(), 0);  // Fills the buffer with 0x00-0xFF.
  const std::string encoded3 =
      "AAECAwQFBgcICQoLDA0ODxAREhMUFRYXGBkaGxwdHh8gISIjJCUmJygpKissLS4vMDEyMzQ"
      "1Njc4OTo7PD0+P0BBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWltcXV5fYGFiY2RlZmdoaW"
      "prbG1ub3BxcnN0dXZ3eHl6e3x9fn+AgYKDhIWGh4iJiouMjY6PkJGSk5SVlpeYmZqbnJ2en"
      "6ChoqOkpaanqKmqq6ytrq+wsbKztLW2t7i5uru8vb6/wMHCw8TFxsfIycrLzM3Oz9DR0tPU"
      "1dbX2Nna29zd3t/g4eLj5OXm5+jp6uvs7e7v8PHy8/T19vf4+fr7/P3+/w==";

  EXPECT_EQ(encoded1, Base64Encode(text1));
  EXPECT_EQ(encoded2, Base64Encode(text2));
  EXPECT_EQ(encoded3, Base64Encode(data3));
}

TEST(data_encoding, Base64EncodeWrapLines) {
  const std::string text1 = "hello world";
  const std::string encoded1 = "aGVsbG8gd29ybGQ=\n";

  const std::string text2 =
      "Lorem ipsum dolor sit amet, facilisis erat nec aliquam, scelerisque "
      "molestie commodo. Viverra tincidunt integer erat ipsum, integer "
      "molestie, arcu in, sit mauris ac a sed sit etiam.";
  const std::string encoded2 =
      "TG9yZW0gaXBzdW0gZG9sb3Igc2l0IGFtZXQsIGZhY2lsaXNpcyBlcmF0IG5lYyBh\n"
      "bGlxdWFtLCBzY2VsZXJpc3F1ZSBtb2xlc3RpZSBjb21tb2RvLiBWaXZlcnJhIHRp\n"
      "bmNpZHVudCBpbnRlZ2VyIGVyYXQgaXBzdW0sIGludGVnZXIgbW9sZXN0aWUsIGFy\n"
      "Y3UgaW4sIHNpdCBtYXVyaXMgYWMgYSBzZWQgc2l0IGV0aWFtLg==\n";

  brillo::Blob data3(256);
  std::iota(data3.begin(), data3.end(), 0);  // Fills the buffer with 0x00-0xFF.
  const std::string encoded3 =
      "AAECAwQFBgcICQoLDA0ODxAREhMUFRYXGBkaGxwdHh8gISIjJCUmJygpKissLS4v\n"
      "MDEyMzQ1Njc4OTo7PD0+P0BBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWltcXV5f\n"
      "YGFiY2RlZmdoaWprbG1ub3BxcnN0dXZ3eHl6e3x9fn+AgYKDhIWGh4iJiouMjY6P\n"
      "kJGSk5SVlpeYmZqbnJ2en6ChoqOkpaanqKmqq6ytrq+wsbKztLW2t7i5uru8vb6/\n"
      "wMHCw8TFxsfIycrLzM3Oz9DR0tPU1dbX2Nna29zd3t/g4eLj5OXm5+jp6uvs7e7v\n"
      "8PHy8/T19vf4+fr7/P3+/w==\n";

  EXPECT_EQ(encoded1, Base64EncodeWrapLines(text1));
  EXPECT_EQ(encoded2, Base64EncodeWrapLines(text2));
  EXPECT_EQ(encoded3, Base64EncodeWrapLines(data3));
}

TEST(data_encoding, Base64Decode) {
  const std::string encoded1 = "dGVzdCBzdHJpbmc=";

  const std::string encoded2 =
      "TG9yZW0gaXBzdW0gZG9sb3Igc2l0IGFtZXQsIGZhY2lsaXNpcyBlcmF0IG5lYyBh\n"
      "bGlxdWFtLCBzY2VsZXJpc3F1ZSBtb2xlc3RpZSBjb21tb2RvLiBWaXZlcnJhIHRp\r\n"
      "bmNpZHVudCBpbnRlZ2VyIGVyYXQgaXBzdW0sIGludGVnZXIgbW9sZXN0aWUsIGFy\r"
      "Y3UgaW4sIHNpdCBtYXVyaXMgYWMgYSBzZWQgc2l0IGV0aWFt\n"
      "Lg==\n\n\n";
  const std::string decoded2 =
      "Lorem ipsum dolor sit amet, facilisis erat nec aliquam, scelerisque "
      "molestie commodo. Viverra tincidunt integer erat ipsum, integer "
      "molestie, arcu in, sit mauris ac a sed sit etiam.";

  const std::string encoded3 =
      "AAECAwQFBgcICQoLDA0ODxAREhMUFRYXGBkaGxwdHh8gISIjJCUmJygpKissLS4vMDEyMzQ"
      "1Njc4OTo7PD0+P0BBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWltcXV5fYGFiY2RlZmdoaW"
      "prbG1ub3BxcnN0dXZ3eHl6e3x9fn+AgYKDhIWGh4iJiouMjY6PkJGSk5SVlpeYmZqbnJ2en"
      "6ChoqOkpaanqKmqq6ytrq+wsbKztLW2t7i5uru8vb6/wMHCw8TFxsfIycrLzM3Oz9DR0tPU"
      "1dbX2Nna29zd3t/g4eLj5OXm5+jp6uvs7e7v8PHy8/T19vf4+fr7/P3+/w==";
  brillo::Blob decoded3(256);
  std::iota(decoded3.begin(), decoded3.end(), 0);  // Fill with 0x00..0xFF.

  std::string decoded;
  EXPECT_TRUE(Base64Decode(encoded1, &decoded));
  EXPECT_EQ("test string", decoded);

  EXPECT_TRUE(Base64Decode(encoded2, &decoded));
  EXPECT_EQ(decoded2, decoded);

  brillo::Blob decoded_blob;
  EXPECT_TRUE(Base64Decode(encoded3, &decoded_blob));
  EXPECT_EQ(decoded3, decoded_blob);

  EXPECT_FALSE(Base64Decode("A", &decoded_blob));
  EXPECT_TRUE(decoded_blob.empty());

  EXPECT_TRUE(Base64Decode("/w==", &decoded_blob));
  EXPECT_EQ((brillo::Blob{0xFF}), decoded_blob);

  EXPECT_TRUE(Base64Decode("//8=", &decoded_blob));
  EXPECT_EQ((brillo::Blob{0xFF, 0xFF}), decoded_blob);

  EXPECT_FALSE(Base64Decode("AAECAwQFB,cI", &decoded_blob));
  EXPECT_TRUE(decoded_blob.empty());
}

}  // namespace data_encoding
}  // namespace brillo