#pragma once

#include "AssetLoading/IAssetLoadingManager.h"

#include <map>
#include <memory>

class MockAssetLoadingManager final : public IAssetLoadingManager
{
    Zone& m_zone;
    std::unique_ptr<Gdt> m_mock_gdt;
    std::unique_ptr<AssetLoadingContext> m_context;
    std::map<std::string, std::unique_ptr<XAssetInfoGeneric>> m_added_assets;
    std::multimap<std::string, std::unique_ptr<XAssetInfoGeneric>> m_available_dependencies;

public:
    MockAssetLoadingManager(Zone& zone, ISearchPath& searchPath);

    _NODISCARD AssetLoadingContext* GetAssetLoadingContext() const override;
    XAssetInfoGeneric* AddAsset(std::unique_ptr<XAssetInfoGeneric> xAssetInfo) override;
    XAssetInfoGeneric* LoadDependency(asset_type_t assetType, const std::string& assetName) override;
    IndirectAssetReference LoadIndirectAssetReference(asset_type_t assetType, const std::string& assetName) override;

    void MockAddAvailableDependency(asset_type_t assetType, std::string assetName, void* asset);
    XAssetInfoGeneric* MockGetAddedAsset(const std::string& assetName);
};