keychain_test_helper.h

/*
 *
 * keychain_test_helper
 * ledger-core
 *
 * Created by El Khalil Bellakrid on 16/05/2018.
 *
 * The MIT License (MIT)
 *
 * Copyright (c) 2018 Ledger
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *
 */

#ifndef LEDGER_CORE_KEYCHAIN_TEST_HELPER_H
#define LEDGER_CORE_KEYCHAIN_TEST_HELPER_H

#include <src/wallet/currencies.hpp>
#include <api/EthereumLikeNetworkParameters.hpp>
#include <api/Currency.hpp>
#include "../BaseFixture.h"


struct KeychainTestData {

    ledger::core::api::BitcoinLikeNetworkParameters btc_parameters;
    ledger::core::api::EthereumLikeNetworkParameters eth_parameters;
    ledger::core::api::Currency currency;
    std::string xpub;
    std::string derivationPath;

    KeychainTestData() = default;

    KeychainTestData(ledger::core::api::BitcoinLikeNetworkParameters parameters_,
                      ledger::core::api::Currency currency_,
                      std::string xpub_,
                      std::string derivationPath_):
            btc_parameters(std::move(parameters_)),
            currency(std::move(currency_)),
            xpub(std::move(xpub_)),
            derivationPath(std::move(derivationPath_))
    {
        currency.bitcoinLikeNetworkParameters = btc_parameters;
    }

    KeychainTestData(ledger::core::api::EthereumLikeNetworkParameters parameters_,
                     ledger::core::api::Currency currency_,
                     std::string xpub_,
                     std::string derivationPath_):
            eth_parameters(std::move(parameters_)),
            currency(std::move(currency_)),
            xpub(std::move(xpub_)),
            derivationPath(std::move(derivationPath_))
    {
        currency.ethereumLikeNetworkParameters = eth_parameters;
    }

    KeychainTestData(const KeychainTestData &data) {
        this->btc_parameters = data.btc_parameters;
        this->eth_parameters = data.eth_parameters;
        this->currency = data.currency;
        this->xpub = data.xpub;
        this->derivationPath = data.derivationPath;
    }

    KeychainTestData& operator=(const KeychainTestData &data) {
        this->btc_parameters = data.btc_parameters;
        this->eth_parameters = data.eth_parameters;
        this->currency = data.currency;
        this->xpub = data.xpub;
        this->derivationPath = data.derivationPath;
        return *this;
    }
};

extern KeychainTestData BTC_DATA;
extern KeychainTestData BTC_TESTNET_DATA;
extern KeychainTestData BCH_DATA;
extern KeychainTestData BTG_DATA;
extern KeychainTestData ZCASH_DATA;
extern KeychainTestData ZENCASH_DATA;
extern KeychainTestData LTC_DATA_LEGACY;
extern KeychainTestData LTC_DATA_SEGWIT;
extern KeychainTestData PEERCOIN_DATA;
extern KeychainTestData DIGIBYTE_DATA;
extern KeychainTestData HCASH_DATA;
extern KeychainTestData QTUM_DATA;
extern KeychainTestData XST_DATA;
extern KeychainTestData VTC_DATA_LEGACY;
extern KeychainTestData VTC_DATA_SEGWIT;
extern KeychainTestData VIA_DATA_LEGACY;
extern KeychainTestData VIA_DATA_SEGWIT;
extern KeychainTestData DASH_DATA;
extern KeychainTestData DOGE_DATA;
extern KeychainTestData STRATIS_DATA;
extern KeychainTestData KOMODO_DATA;
extern KeychainTestData POSWALLET_DATA;
extern KeychainTestData PIVX_DATA;
extern KeychainTestData CLUBCOIN_DATA;
extern KeychainTestData DECRED_DATA;
extern KeychainTestData ETHEREUM_DATA;

template <class Keychain>
class KeychainFixture : public BaseFixture {
public:
    void testKeychain(const KeychainTestData &data, std::function<void (Keychain&)> f) {
        auto backend = std::make_shared<ledger::core::PreferencesBackend>(
                "/preferences/tests.db",
                dispatcher->getMainExecutionContext(),
                resolver
        );
        auto configuration = std::make_shared<DynamicObject>();
        dispatcher->getMainExecutionContext()->execute(ledger::qt::make_runnable([=]() {
            Keychain keychain(
                    configuration,
                    data.currency,
                    0,
                    ledger::core::BitcoinLikeExtendedPublicKey::fromBase58(data.currency,
                                                                           data.xpub,
                                                                           optional<std::string>(data.derivationPath),
                                                                           configuration),
                    backend->getPreferences("keychain")
            );
            f(keychain);
            dispatcher->stop();
        }));
        dispatcher->waitUntilStopped();
    };
};

#endif //LEDGER_CORE_KEYCHAIN_TEST_HELPER_H