// Copyright © 2017-2020 Trust Wallet.

//

// This file is part of Trust. The full Trust copyright notice, including

// terms governing use, modification, and redistribution, is contained in the

// file LICENSE at the root of the source code distribution tree.

#include "Base32.h"

#include "HexCoding.h"

#include <gtest/gtest.h>

using namespace TW;

using namespace TW::Base32;

void TestBase32Encode(const char* decoded_hex, const char* expected_encoded_in, const char* alphabet_in = nullptr)

{

auto decoded = parse_hex(std::string(decoded_hex));

auto encoded = encode(decoded, alphabet_in);

ASSERT_EQ(std::string(expected_encoded_in), encoded);

}

void TestBase32Decode(const char* encoded_in, const char* expected_decoded_hex, const char* alphabet_in = nullptr)

{

Data decoded;

bool res = decode(std::string(encoded_in), decoded, alphabet_in);

ASSERT_TRUE(res);

auto decoded_hex = hex(decoded);

ASSERT_EQ(std::string(expected_decoded_hex), decoded_hex);

}

TEST(Base32, Encode) {

TestBase32Encode("01", "AE");

TestBase32Encode("010203", "AEBAG");

TestBase32Encode("", "");

TestBase32Encode(

"48450c2745890def7da06fc2551f912a14f9fc581c12db6e4d6f73f2fd0b2ad50df3d396",

"JBCQYJ2FREG667NAN7BFKH4RFIKPT7CYDQJNW3SNN5Z7F7ILFLKQ346TSY");

TestBase32Encode(

"3dd160d60673bd9b13adc25dad5d988d0d9f4ccdbe95a2122f9ef28b3ce4e89693074620",

"HXIWBVQGOO6ZWE5NYJO22XMYRUGZ6TGNX2K2EERPT3ZIWPHE5CLJGB2GEA");

TestBase32Encode(

"7a263717bf89dd55cf7f533e315c6224d635edc0e88206fd70e733bf2cc516be84e910f7",

"PITDOF57RHOVLT37KM7DCXDCETLDL3OA5CBAN7LQ44Z36LGFC27IJ2IQ64");

}

TEST(Base32, Decode) {

TestBase32Decode("AE", "01");

TestBase32Decode("AEBAG", "010203");

TestBase32Decode("", "");

TestBase32Decode(

"JBCQYJ2FREG667NAN7BFKH4RFIKPT7CYDQJNW3SNN5Z7F7ILFLKQ346TSY",

"48450c2745890def7da06fc2551f912a14f9fc581c12db6e4d6f73f2fd0b2ad50df3d396");

TestBase32Decode(

"HXIWBVQGOO6ZWE5NYJO22XMYRUGZ6TGNX2K2EERPT3ZIWPHE5CLJGB2GEA",

"3dd160d60673bd9b13adc25dad5d988d0d9f4ccdbe95a2122f9ef28b3ce4e89693074620");

TestBase32Decode(

"PITDOF57RHOVLT37KM7DCXDCETLDL3OA5CBAN7LQ44Z36LGFC27IJ2IQ64",

"7a263717bf89dd55cf7f533e315c6224d635edc0e88206fd70e733bf2cc516be84e910f7");

}

TEST(Base32, EncodeNimiq) {

const char* BASE32_ALPHABET_NIMIQ = "0123456789ABCDEFGHJKLMNPQRSTUVXY";

TestBase32Encode("01", "04", BASE32_ALPHABET_NIMIQ);

TestBase32Encode("010203", "04106", BASE32_ALPHABET_NIMIQ);

}

TEST(Base32, DecodeInvalid) {

Data decoded;

ASSERT_FALSE(decode("+-", decoded)); // invalid characters

ASSERT_FALSE(decode("A", decoded)); // invalid odd length

ASSERT_FALSE(decode("ABC", decoded)); // invalid odd length

}