Branch data     Line data    Source code

#       1                 :            : // Copyright (c) 2018-2021 The Bitcoin Core developers
#       2                 :            : // Distributed under the MIT software license, see the accompanying
#       3                 :            : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
#       4                 :            : 
#       5                 :            : #include <boost/test/unit_test.hpp>
#       6                 :            : 
#       7                 :            : #include <fs.h>
#       8                 :            : #include <test/util/setup_common.h>
#       9                 :            : #include <wallet/bdb.h>
#      10                 :            : 
#      11                 :            : #include <fstream>
#      12                 :            : #include <memory>
#      13                 :            : #include <string>
#      14                 :            : 
#      15                 :            : namespace wallet {
#      16                 :            : BOOST_FIXTURE_TEST_SUITE(db_tests, BasicTestingSetup)
#      17                 :            : 
#      18                 :            : static std::shared_ptr<BerkeleyEnvironment> GetWalletEnv(const fs::path& path, std::string& database_filename)
#      19                 :          9 : {
#      20                 :          9 :     fs::path data_file = BDBDataFile(path);
#      21                 :          9 :     database_filename = fs::PathToString(data_file.filename());
#      22                 :          9 :     return GetBerkeleyEnv(data_file.parent_path(), false);
#      23                 :          9 : }
#      24                 :            : 
#      25                 :            : BOOST_AUTO_TEST_CASE(getwalletenv_file)
#      26                 :          1 : {
#      27                 :          1 :     std::string test_name = "test_name.dat";
#      28                 :          1 :     const fs::path datadir = m_args.GetDataDirNet();
#      29                 :          1 :     fs::path file_path = datadir / test_name;
#      30                 :          1 :     std::ofstream f{file_path};
#      31                 :          1 :     f.close();
#      32                 :            : 
#      33                 :          1 :     std::string filename;
#      34                 :          1 :     std::shared_ptr<BerkeleyEnvironment> env = GetWalletEnv(file_path, filename);
#      35                 :          1 :     BOOST_CHECK_EQUAL(filename, test_name);
#      36                 :          1 :     BOOST_CHECK_EQUAL(env->Directory(), datadir);
#      37                 :          1 : }
#      38                 :            : 
#      39                 :            : BOOST_AUTO_TEST_CASE(getwalletenv_directory)
#      40                 :          1 : {
#      41                 :          1 :     std::string expected_name = "wallet.dat";
#      42                 :          1 :     const fs::path datadir = m_args.GetDataDirNet();
#      43                 :            : 
#      44                 :          1 :     std::string filename;
#      45                 :          1 :     std::shared_ptr<BerkeleyEnvironment> env = GetWalletEnv(datadir, filename);
#      46                 :          1 :     BOOST_CHECK_EQUAL(filename, expected_name);
#      47                 :          1 :     BOOST_CHECK_EQUAL(env->Directory(), datadir);
#      48                 :          1 : }
#      49                 :            : 
#      50                 :            : BOOST_AUTO_TEST_CASE(getwalletenv_g_dbenvs_multiple)
#      51                 :          1 : {
#      52                 :          1 :     fs::path datadir = m_args.GetDataDirNet() / "1";
#      53                 :          1 :     fs::path datadir_2 = m_args.GetDataDirNet() / "2";
#      54                 :          1 :     std::string filename;
#      55                 :            : 
#      56                 :          1 :     std::shared_ptr<BerkeleyEnvironment> env_1 = GetWalletEnv(datadir, filename);
#      57                 :          1 :     std::shared_ptr<BerkeleyEnvironment> env_2 = GetWalletEnv(datadir, filename);
#      58                 :          1 :     std::shared_ptr<BerkeleyEnvironment> env_3 = GetWalletEnv(datadir_2, filename);
#      59                 :            : 
#      60                 :          1 :     BOOST_CHECK(env_1 == env_2);
#      61                 :          1 :     BOOST_CHECK(env_2 != env_3);
#      62                 :          1 : }
#      63                 :            : 
#      64                 :            : BOOST_AUTO_TEST_CASE(getwalletenv_g_dbenvs_free_instance)
#      65                 :          1 : {
#      66                 :          1 :     fs::path datadir = gArgs.GetDataDirNet() / "1";
#      67                 :          1 :     fs::path datadir_2 = gArgs.GetDataDirNet() / "2";
#      68                 :          1 :     std::string filename;
#      69                 :            : 
#      70                 :          1 :     std::shared_ptr <BerkeleyEnvironment> env_1_a = GetWalletEnv(datadir, filename);
#      71                 :          1 :     std::shared_ptr <BerkeleyEnvironment> env_2_a = GetWalletEnv(datadir_2, filename);
#      72                 :          1 :     env_1_a.reset();
#      73                 :            : 
#      74                 :          1 :     std::shared_ptr<BerkeleyEnvironment> env_1_b = GetWalletEnv(datadir, filename);
#      75                 :          1 :     std::shared_ptr<BerkeleyEnvironment> env_2_b = GetWalletEnv(datadir_2, filename);
#      76                 :            : 
#      77                 :          1 :     BOOST_CHECK(env_1_a != env_1_b);
#      78                 :          1 :     BOOST_CHECK(env_2_a == env_2_b);
#      79                 :          1 : }
#      80                 :            : 
#      81                 :            : BOOST_AUTO_TEST_SUITE_END()
#      82                 :            : } // namespace wallet