thehub/testing/common/TestFloweeSession.cpp

/*
 * This file is part of the Flowee project
 * Copyright (C) 2011-2015 The Bitcoin Core developers
 * Copyright (C) 2017-2021 Tom Zander <tom@flowee.org>
 *
 * 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, either version 3 of the License, or
 * (at your option) any later version.
 *
 * 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 "TestFloweeSession.h"
#include "script/sigcache.h"
#include "MockApplication.h"
#include <utxo/UnspentOutputDatabase.h>
#include <BlocksDB.h>
#include <interfaces/validationinterface.h>
#include <UiInterface.h>
#include <util.h>
#include <utiltime.h>
#include <main.h>

#include <boost/filesystem.hpp>

CClientUIInterface uiInterface; // Declared but not defined in UiInterface.h

UnspentOutputDatabase *g_utxo = nullptr;

void TestFloweeSession::init()
{
    MockApplication::doStartThreads();
    bv.reset(new MockBlockValidation());
    ClearDatadirCache();
    pathTemp = boost::filesystem::temp_directory_path() / strprintf("test_flowee_%lu_%i", (unsigned long)GetTime(), (int)(GetRand(100000)));
    boost::filesystem::create_directories(pathTemp / "regtest/blocks/index");
    boost::filesystem::create_directories(pathTemp / "blocks/index");
    mapArgs["-datadir"] = pathTemp.string();
    Blocks::DB::createTestInstance(1<<20);
    UnspentOutputDatabase::setSmallLimits();
    g_utxo = new UnspentOutputDatabase(Application::instance()->ioContext(), GetDataDir(true) / "unspent");

    // in case the executable already shut them down, lets restart them.
    bv->initSingletons();
    bv->appendGenesis();
    MockApplication::setValidationEngine(bv.get());
    RegisterNodeSignals(GetNodeSignals());
}

TestFloweeSession::TestFloweeSession(const std::string& chainName) : TestFloweeEnvPlusNet(chainName)
{
    InitSignatureCache();
}

void TestFloweeSession::cleanup()
{
    MockApplication::setValidationEngine(nullptr);
    bv->waitValidationFinished();
    bv->shutdown();
    Blocks::Index::unload();
    Application::instance()->stopThreads();
    Application::instance()->joinAll();

    UnregisterNodeSignals(GetNodeSignals());
    ValidationNotifier().removeAll();
    UnloadBlockIndex();
    delete g_utxo;
    g_utxo = nullptr;
    boost::filesystem::remove_all(pathTemp);
}

TestFloweeSession::~TestFloweeSession()
{
}