diff --git a/tests/unit_tests/subaddress.cpp b/tests/unit_tests/subaddress.cpp index 0233b74a6..c304b7347 100644 --- a/tests/unit_tests/subaddress.cpp +++ b/tests/unit_tests/subaddress.cpp @@ -44,12 +44,12 @@ class WalletSubaddress : public ::testing::Test { try { - w1.generate(wallet_name, password, recovery_key, true, false); + w1.generate(wallet_name, password, recovery_key, true, false); } catch (const std::exception& e) { - LOG_ERROR("failed to generate wallet: " << e.what()); - throw e; + LOG_ERROR("failed to generate wallet: " << e.what()); + throw e; } w1.add_subaddress_account(test_label); @@ -58,21 +58,20 @@ class WalletSubaddress : public ::testing::Test virtual void TearDown() { - boost::filesystem::wpath wallet_file(wallet_name); - boost::filesystem::wpath wallet_address_file(wallet_name + ".address.txt"); - boost::filesystem::wpath wallet_keys_file(wallet_name + ".keys"); + boost::filesystem::wpath wallet_file(wallet_name); + boost::filesystem::wpath wallet_address_file(wallet_name + ".address.txt"); + boost::filesystem::wpath wallet_keys_file(wallet_name + ".keys"); - if ( boost::filesystem::exists(wallet_file) ) - boost::filesystem::remove(wallet_file); + if ( boost::filesystem::exists(wallet_file) ) + boost::filesystem::remove(wallet_file); - if ( boost::filesystem::exists(wallet_address_file) ) - boost::filesystem::remove(wallet_address_file); + if ( boost::filesystem::exists(wallet_address_file) ) + boost::filesystem::remove(wallet_address_file); - if ( boost::filesystem::exists(wallet_keys_file) ) - boost::filesystem::remove(wallet_keys_file); + if ( boost::filesystem::exists(wallet_keys_file) ) + boost::filesystem::remove(wallet_keys_file); } - tools::wallet2 w1; std::string path_working_dir = "."; std::string path_test_wallet = "test_wallet"; @@ -86,7 +85,34 @@ class WalletSubaddress : public ::testing::Test const cryptonote::subaddress_index subaddress_index = {major_index, minor_index}; }; -TEST_F(WalletSubaddress, AddRow) +TEST_F(WalletSubaddress, GetSubaddressLabel) +{ + EXPECT_EQ(test_label, w1.get_subaddress_label(subaddress_index)); +} + +TEST_F(WalletSubaddress, AddSubaddress) +{ + std::string label = "test adding subaddress"; + w1.add_subaddress(0, label); + EXPECT_EQ(label, w1.get_subaddress_label({0, 1})); +} + +TEST_F(WalletSubaddress, OutOfBoundsIndexes) { - EXPECT_EQ(test_label, w1.get_subaddress_label(subaddress_index)); + try + { + w1.get_subaddress_label({1,0}); + } + catch(const std::exception& e) + { + EXPECT_STREQ("index_major is out of bound", e.what()); + } + try + { + w1.get_subaddress_label({0,2}); + } + catch(const std::exception& e) + { + EXPECT_STREQ("index.minor is out of bound", e.what()); + } }