defmodule Commanded.Aggregates.AggregateConcurrencyTest do
  use Commanded.MockEventStoreCase

  alias Commanded.Aggregates.{Aggregate, ExecutionContext}
  alias Commanded.EventStore.RecordedEvent
  alias Commanded.ExampleDomain.{BankAccount, OpenAccountHandler, DepositMoneyHandler}
  alias Commanded.ExampleDomain.BankAccount.Commands.{OpenAccount, DepositMoney}
  alias Commanded.ExampleDomain.BankAccount.Events.MoneyDeposited

  setup do
    expect(MockEventStore, :subscribe_to_all_streams, fn _handler_name, handler, _subscribe_from ->
      {:ok, handler}
    end)

    expect(MockEventStore, :subscribe, fn _aggregate_uuid -> :ok end)

    :ok
  end

  describe "concurrency error" do
    setup [
      :open_account
    ]

    test "should retry command", context do
      %{account_number: account_number} = context

      command = %DepositMoney{
        account_number: account_number,
        transfer_uuid: UUID.uuid4(),
        amount: 100
      }

      context = %ExecutionContext{
        command: command,
        handler: DepositMoneyHandler,
        function: :handle,
        retry_attempts: 5
      }

      # fail to append once
      expect(MockEventStore, :append_to_stream, fn ^account_number, 1, _event_data ->
        {:error, :wrong_expected_version}
      end)

      # return "missing" event
      expect(MockEventStore, :stream_forward, fn ^account_number, 2, _batch_size ->
        [
          %RecordedEvent{
            event_id: UUID.uuid4(),
            event_number: 2,
            stream_id: account_number,
            stream_version: 2,
            event_type: "Elixir.Commanded.ExampleDomain.BankAccount.Events.MoneyDeposited",
            data: %MoneyDeposited{
              account_number: account_number,
              transfer_uuid: UUID.uuid4(),
              amount: 500,
              balance: 1_500
            },
            metadata: %{}
          }
        ]
      end)

      # succeed on second attempt
      expect(MockEventStore, :append_to_stream, fn ^account_number, 2, event_data ->
        {:ok, 2 + length(event_data)}
      end)

      assert {:ok, 3, _events} = Aggregate.execute(BankAccount, account_number, context)

      assert Aggregate.aggregate_version(BankAccount, account_number) == 3

      assert Aggregate.aggregate_state(BankAccount, account_number) == %BankAccount{
               account_number: account_number,
               balance: 1_600,
               state: :active
             }
    end

    test "should error after too many attempts", context do
      %{account_number: account_number} = context

      # fail to append to stream
      expect(MockEventStore, :append_to_stream, 6, fn ^account_number, 1, _event_data ->
        {:error, :wrong_expected_version}
      end)

      expect(MockEventStore, :stream_forward, 6, fn ^account_number, 2, _batch_size -> [] end)

      command = %DepositMoney{
        account_number: account_number,
        transfer_uuid: UUID.uuid4(),
        amount: 100
      }

      context = %ExecutionContext{
        command: command,
        handler: DepositMoneyHandler,
        function: :handle,
        retry_attempts: 5
      }

      assert {:error, :too_many_attempts} =
               Aggregate.execute(BankAccount, account_number, context)
    end

    defp open_account(_context) do
      account_number = UUID.uuid4()

      expect(MockEventStore, :stream_forward, fn ^account_number, 1, _batch_size ->
        []
      end)

      expect(MockEventStore, :append_to_stream, fn ^account_number, 0, event_data ->
        {:ok, length(event_data)}
      end)

      {:ok, ^account_number} =
        Commanded.Aggregates.Supervisor.open_aggregate(BankAccount, account_number)

      command = %OpenAccount{account_number: account_number, initial_balance: 1_000}

      context = %ExecutionContext{
        command: command,
        handler: OpenAccountHandler,
        function: :handle,
        retry_attempts: 1
      }

      {:ok, 1, _events} = Aggregate.execute(BankAccount, account_number, context)

      [
        account_number: account_number
      ]
    end
  end
end