From c75bc83fc1e03de041cf42ee6598fdc2df633af3 Mon Sep 17 00:00:00 2001
From: SChernykh <sergey.v.chernykh@gmail.com>
Date: Sat, 4 Sep 2021 00:05:37 +0200
Subject: [PATCH] Added Wallet tests

---
 tests/CMakeLists.txt       |   6 ++
 tests/src/wallet_tests.cpp | 127 +++++++++++++++++++++++++++++++++++++
 2 files changed, 133 insertions(+)
 create mode 100644 tests/src/wallet_tests.cpp

diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index 15e637d..25d122a 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -53,8 +53,14 @@ set(HEADERS
 set(SOURCES
 	src/difficulty_type.cpp
 	src/tests.cpp
+	src/wallet_tests.cpp
+	../external/src/cryptonote/crypto-ops-data.c
+	../external/src/cryptonote/crypto-ops.c
+	../src/crypto.cpp
+	../src/keccak.cpp
 	../src/log.cpp
 	../src/util.cpp
+	../src/wallet.cpp
 )
 
 include_directories(../src)
diff --git a/tests/src/wallet_tests.cpp b/tests/src/wallet_tests.cpp
new file mode 100644
index 0000000..5f93434
--- /dev/null
+++ b/tests/src/wallet_tests.cpp
@@ -0,0 +1,127 @@
+/*
+ * This file is part of the Monero P2Pool <https://github.com/SChernykh/p2pool>
+ * Copyright (c) 2021 SChernykh <https://github.com/SChernykh>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, version 3.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "common.h"
+#include "wallet.h"
+#include "gtest/gtest.h"
+
+namespace p2pool {
+
+TEST(wallet, decode)
+{
+	// No data
+	{
+		Wallet w(nullptr);
+		ASSERT_EQ(w.valid(), false);
+	}
+
+	// Wrong length
+	{
+		Wallet w("456");
+		ASSERT_EQ(w.valid(), false);
+	}
+
+	// Symbol '0' is not from base-58
+	{
+		Wallet w("40ccoSmrBTPJd5yf8VYCULh4J5rHQaXP1TeC8Cnqhd5H9Y2cMwkJ9w42euLmMghKtCiQcgZEiGYW1K6Ae4biZ7w1HLSexS6");
+		ASSERT_EQ(w.valid(), false);
+	}
+
+	// Invalid checksum
+	{
+		Wallet w("49ccoSmrBTPJd5yf8VYCULh4J5rHQaXP1TeC8Cnqhd5H9Y2cMwkJ9w42euLmMghKtCiQcgZEiGYW1K6Ae4biZ7w1HLSexS7");
+		ASSERT_EQ(w.valid(), false);
+	}
+
+	// 64-bit overflow
+	{
+		Wallet w("49ccoSmrBTPzzzzzzzzzzzh4J5rHQaXP1TeC8Cnqhd5H9Y2cMwkJ9w42euLmMghKtCiQcgZEiGYW1K6Ae4biZ7w1HLSexS6");
+		ASSERT_EQ(w.valid(), false);
+	}
+
+	// Subaddress (not supported)
+	{
+		Wallet w("8BE7uo9kWR6fFekhGHKJt87pkTzzNj2ikZMNmN7DUJf81y6Zygzbsk1CFzGMbS7fB7E2qr6A6EZfLYgxUfYvdDxEHrMPMA5");
+		ASSERT_EQ(w.valid(), false);
+	}
+
+	auto check = [](NetworkType t, const char* address, const char* spendkey, const char* viewkey)
+	{
+		Wallet w(address);
+		ASSERT_EQ(w.type(), t);
+
+		char buf[log::Stream::BUF_SIZE + 1];
+		log::Stream s(buf);
+
+		s << w.spend_public_key();
+		ASSERT_EQ(memcmp(buf, spendkey, HASH_SIZE * 2), 0);
+
+		s.m_pos = 0;
+		s << w.view_public_key();
+		ASSERT_EQ(memcmp(buf, viewkey, HASH_SIZE * 2), 0);
+	};
+
+	// Correct mainnet addresses
+	check(
+		NetworkType::Mainnet, "49ccoSmrBTPJd5yf8VYCULh4J5rHQaXP1TeC8Cnqhd5H9Y2cMwkJ9w42euLmMghKtCiQcgZEiGYW1K6Ae4biZ7w1HLSexS6",
+		"d2e232e441546a695b27187692d035ef7be5c54692700c9f470dcd706753a833", "06f68970da46f709e2b4d0ffabd0d1f78ea6717786b5766c25c259111f212490"
+	);
+
+	check(
+		NetworkType::Mainnet, "45JHuqGBSqUXUyZx95H4C2J5aEL4zFjM3jpTmMTESPXPa3jmtSQWYezHX7r4A2xPQNBGsQupJqmPhRZb2QgBcEWRDQ9ywwR",
+		"60fe176eaf3cffb63df130bc25036b661b947900941052fffe6ff4b51fc4f2c5", "9387910b0a2e4f62c32621b77ddbeb3d6c0054e5ed9bc492d87bab1a1eef366d"
+	);
+
+	check(
+		NetworkType::Mainnet, "43S5vhReDY4fJs99DBZtFS8JoJVNG17iaAVAARvRT8xzSYZqnJfXfTACLrZUzoBHQKhiJZCWCpqB4Kf3c64CEagdSRXd5D7",
+		"2fc2f902659541e50753853ddb96912baf55f26bebe7d338b5c2239c437ddb98", "b814951166253543cfb0e1b8bdea58f366de824fddb8ef6f895fcf631873f6e1"
+	);
+
+	// Correct testnet addresses
+	check(
+		NetworkType::Testnet, "9x6aEN1yd2WhPMPw89LV5LLK1ZFe6N8xiAm18Ay4q1U4LKMde7MpDdPRN6GiiGCJMVTHuptGGmfj2Qfp2vcKSRSG79HJrQn",
+		"821623ac165f07f172c86980254a43737332fd89ca36d33a57dc02d8026d9173", "7c55413e672f8691a9211eac6003109d2fdf224ba72c4d8d82353427a02bc136"
+	);
+
+	check(
+		NetworkType::Testnet, "9zsJP6KFF6ZGern5UkR7gyRXHFRTba6jG8JKnfzDySeqEdwPZaD8MNYGkjyADdVpWs7rXgyeu712JdxhX2k7d9SNB4TdRdS",
+		"cb366a3b44f6aa5d94e03db06325b6929b9e75dbf19dcf2ba2d14eb2efa53651", "8789afa33dca295e301baef826cec028fa22b831822c1bdcf8a847a43a3bff59"
+	);
+
+	check(
+		NetworkType::Testnet, "A1SqL5oPjh8Km1At7mao7U1fNjWkzeSwvQ39GimMqvhBF3FUoJhx1zxL2i6XbHzzAXDhKetiwSmYQeVwG6sUgwJuEqPyjWq",
+		"da78298fb6eb8f702698bec873bad703f4a51e1377a66d89ba977ca7f43b8e53", "eeb348f70afad971c50aa062f1d1544be64ef9cdc12475e030f2d295305e6e7a"
+	);
+
+	// Correct stagenet addresses
+	check(
+		NetworkType::Stagenet, "55AJ4jJBhV6JsoqrEsAazTLrJjg9SA1SFReLUoXDudrsA9tdL9i2VkJefEbx3zrFRt6swuibPVySPGNzsNvyshrRNZbSDnD",
+		"57e0c2fef80a1d6adfa3189134009076ad0ddc4c4668709355cea98524e9fc36", "b94fafe59d5037e126557665f76cd3232504ebd82500e05bf25801d853d182bf"
+	);
+
+	check(
+		NetworkType::Stagenet, "5BQqg4HTWuN3j4NzBHTK31eTaygRXYxWRQW9dTD7qMuJSiVtskraSErXQ24FUBeifiV6NaQPmxLS559vbUT4xYUoF2fiGvH",
+		"fcd35a53cef9a1104ae556f01cee0cdff2f18f2f2f6bde8c833d5bd980fe8999", "be2b1142a046bfb5bb21e1f2a49bd1a7f46e1c18b009b218d5962f663938707c"
+	);
+
+	check(
+		NetworkType::Stagenet, "53CFYfjzcouW95hQ7AHvqS3GZ2UAAaRLKc1ymmhHATQTZxhtakpYcfjiRVzrRdxVZ5F8p61KSpPEmFu9DVRULRDkK4v1TCU",
+		"23fdd143264794ae367083791bb8fd0d8f719b27b7b858d15a2b67d6eddd60c5", "0ebafc1284ab1af7a5ff4ade682bcc54817a319a00eede591344855c420beba0"
+	);
+}
+
+}