cs-codex-dist-tests/DistTestCore/Marketplace/MarketplaceAccessFactory.cs

using DistTestCore.Codex;
using Logging;

namespace DistTestCore.Marketplace
{
    public interface IMarketplaceAccessFactory
    {
        IMarketplaceAccess CreateMarketplaceAccess(CodexAccess access);
    }

    public class MarketplaceUnavailableAccessFactory : IMarketplaceAccessFactory
    {
        public IMarketplaceAccess CreateMarketplaceAccess(CodexAccess access)
        {
            return new MarketplaceUnavailable();
        }
    }

    public class GethMarketplaceAccessFactory : IMarketplaceAccessFactory
    {
        private readonly TestLog log;
        private readonly MarketplaceNetwork marketplaceNetwork;

        public GethMarketplaceAccessFactory(TestLog log, MarketplaceNetwork marketplaceNetwork)
        {
            this.log = log;
            this.marketplaceNetwork = marketplaceNetwork;
        }

        public IMarketplaceAccess CreateMarketplaceAccess(CodexAccess access)
        {
            var companionNode = GetGethCompanionNode(access);
            return new MarketplaceAccess(log, marketplaceNetwork, companionNode, access);
        }

        private GethCompanionAccount GetGethCompanionNode(CodexAccess access)
        {
            var account = access.Container.Recipe.Additionals.Single(a => a is GethCompanionAccount);
            return (GethCompanionAccount)account;
        }
    }
}