squidex/backend/tests/Squidex.Infrastructure.Tests/TestHelpers/MyDomainObject.cs

// ==========================================================================
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex UG (haftungsbeschraenkt)
//  All rights reserved. Licensed under the MIT license.
// ==========================================================================

using Microsoft.Extensions.Logging;
using Squidex.Infrastructure.Commands;
using Squidex.Infrastructure.EventSourcing;
using Squidex.Infrastructure.States;

#pragma warning disable MA0048 // File name must match type name

namespace Squidex.Infrastructure.TestHelpers;

public sealed class MyDomainObject(DomainId id, IPersistenceFactory<MyDomainState> factory) : DomainObject<MyDomainState>(id, factory, A.Dummy<ILogger>())
{
    public bool RecreateCommand { get; set; }

    public bool RecreateEvent { get; set; }

    protected override bool IsRecreation(IEvent @event)
    {
        return RecreateEvent && @event is ValueChanged;
    }

    protected override bool IsDeleted(MyDomainState snapshot)
    {
        return snapshot.IsDeleted;
    }

    protected override bool CanAccept(ICommand command)
    {
        if (command is CreateAuto create)
        {
            return create.Value != 99;
        }

        if (command is UpdateAuto update)
        {
            return update.Value != 99;
        }

        return true;
    }

    protected override bool CanAccept(ICommand command, DomainObjectState state)
    {
        static bool CanCreate(ICommand command)
        {
            return
                command is CreateAuto ||
                command is CreateCustom ||
                command is Upsert;
        }

        static bool CanUpdate(ICommand command)
        {
            return
                command is UpdateAuto ||
                command is UpdateCustom ||
                command is Delete ||
                command is DeletePermanent;
        }

        switch (state)
        {
            case DomainObjectState.Undefined:
                return CanCreate(command);
            case DomainObjectState.Empty:
                return CanCreate(command);
            case DomainObjectState.Deleted:
                return (CanCreate(command) && RecreateCommand) || command is DeletePermanent;
            default:
                return CanUpdate(command);
        }
    }

    public override Task<CommandResult> ExecuteAsync(IAggregateCommand command,
        CancellationToken ct)
    {
        switch (command)
        {
            case Upsert c:
                return Apply(c, createAuto =>
                {
                    RaiseEvent(new ValueChanged { Value = createAuto.Value });
                }, ct);

            case CreateAuto c:
                return Apply(c, createAuto =>
                {
                    RaiseEvent(new ValueChanged { Value = createAuto.Value });
                }, ct);

            case CreateCustom c:
                return ApplyReturn(c, createCustom =>
                {
                    RaiseEvent(new ValueChanged { Value = createCustom.Value });

                    return "CREATED";
                }, ct);

            case UpdateAuto c:
                return Apply(c, updateAuto =>
                {
                    RaiseEvent(new ValueChanged { Value = updateAuto.Value });
                }, ct);

            case UpdateCustom c:
                return ApplyReturn(c, updateCustom =>
                {
                    RaiseEvent(new ValueChanged { Value = updateCustom.Value });

                    return "UPDATED";
                }, ct);

            case Delete c:
                return Apply(c, delete =>
                {
                    RaiseEvent(new Deleted());
                }, ct);

            case DeletePermanent c:
                return DeletePermanent(c, delete =>
                {
                    RaiseEvent(new Deleted());
                }, ct);

            default:
                throw new NotSupportedException();
        }
    }

    protected override MyDomainState Apply(MyDomainState snapshot, Envelope<IEvent> @event)
    {
        return snapshot.Apply(@event);
    }
}

public sealed class Delete : MyCommand
{
}

public sealed class DeletePermanent : MyCommand
{
}

public sealed class Upsert : MyCommand
{
    public long Value { get; set; }
}

public sealed class CreateAuto : MyCommand
{
    public long Value { get; set; }
}

public sealed class CreateCustom : MyCommand
{
    public long Value { get; set; }
}

public sealed class UpdateAuto : MyCommand
{
    public long Value { get; set; }
}

public sealed class UpdateCustom : MyCommand
{
    public long Value { get; set; }
}