lez-programs/apps/shared/wallet/tests/support/FakeWalletProvider.h

#pragma once

#include <utility>

#include "WalletProvider.h"

class FakeWalletProvider final : public WalletProvider {
public:
    WalletSession connectResult;
    WalletCreation createWalletResult;
    WalletSnapshot snapshotResult;
    WalletAccountCreation createAccountResult;
    WalletAccountRead readResult;
    QVector<WalletAccountRead> readResults;
    WalletSubmission submissionResult;

    int connectCalls = 0;
    int createWalletCalls = 0;
    int snapshotCalls = 0;
    int clearCalls = 0;
    int createAccountCalls = 0;
    mutable int readCalls = 0;
    int publicAccountReadCalls = 0;
    int submitCalls = 0;
    int disconnectCalls = 0;
    bool lastForceRefresh = false;
    bool lastAccountWasPublic = false;
    WalletPaths lastPaths;
    WalletTransaction lastTransaction;
    QStringList lastPublicAccountIds;
    bool deferPublicAccountReads = false;

    struct PendingPublicAccountRead {
        QStringList accountIds;
        AccountReadsCallback callback;
    };
    QVector<PendingPublicAccountRead> pendingPublicAccountReads;

    WalletSession connect(const WalletPaths& paths) override
    {
        ++connectCalls;
        lastPaths = paths;
        return connectResult;
    }

    void connectAsync(const WalletPaths& paths, SessionCallback callback) override
    {
        ++connectCalls;
        lastPaths = paths;
        callback(connectResult);
    }

    WalletCreation createWallet(const WalletPaths& paths,
                                const QString&) override
    {
        ++createWalletCalls;
        lastPaths = paths;
        return createWalletResult;
    }

    WalletSnapshot snapshot(bool forceRefresh) override
    {
        ++snapshotCalls;
        lastForceRefresh = forceRefresh;
        return snapshotResult;
    }

    void snapshotAsync(bool forceRefresh, SnapshotCallback callback) override
    {
        ++snapshotCalls;
        lastForceRefresh = forceRefresh;
        callback(snapshotResult);
    }

    void clearSnapshot() override { ++clearCalls; }

    WalletAccountCreation createAccount(bool isPublic) override
    {
        ++createAccountCalls;
        lastAccountWasPublic = isPublic;
        return createAccountResult;
    }

    WalletAccountRead readPublicAccount(const QString& accountId) const override
    {
        ++readCalls;
        WalletAccountRead result = readResult;
        result.accountId = accountId;
        return result;
    }

    void readPublicAccountsAsync(const QStringList& accountIds,
                                 AccountReadsCallback callback) override
    {
        ++publicAccountReadCalls;
        lastPublicAccountIds = accountIds;
        if (deferPublicAccountReads) {
            pendingPublicAccountReads.append({ accountIds, std::move(callback) });
            return;
        }
        callback(accountReads(accountIds));
    }

    void completePendingPublicAccountReads()
    {
        QVector<PendingPublicAccountRead> pending;
        pending.swap(pendingPublicAccountReads);
        for (PendingPublicAccountRead& read : pending)
            read.callback(accountReads(read.accountIds));
    }

    WalletSubmission submitPublicTransaction(
        const WalletTransaction& transaction) override
    {
        ++submitCalls;
        lastTransaction = transaction;
        return submissionResult;
    }

    void disconnect() override { ++disconnectCalls; }

private:
    QVector<WalletAccountRead> accountReads(const QStringList& accountIds) const
    {
        QVector<WalletAccountRead> results = readResults;
        if (results.isEmpty()) {
            results.reserve(accountIds.size());
            for (const QString& accountId : accountIds) {
                WalletAccountRead result = readResult;
                result.accountId = accountId;
                results.append(std::move(result));
            }
        }
        return results;
    }
};