// ------------------------------------------------------------
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
// ------------------------------------------------------------

namespace Dapr.Actors.Test
{
    using System;
    using System.Threading;
    using System.Threading.Tasks;
    using Dapr.Actors.Runtime;

    /// <summary>
    /// Interface for test actor.
    /// </summary>
    public interface ITestActor : IActor
    {
        /// <summary>
        /// GetCount method for TestActor.
        /// </summary>
        /// <param name="cancellationToken">Cancellation token to cancel the operation.</param>
        /// <returns>The current count as stored in actor.</returns>
        Task<int> GetCountAsync(CancellationToken cancellationToken);

        /// <summary>
        /// SetCount method for test actor.
        /// </summary>
        /// <param name="count">Count to set for the actor.</param>
        /// <param name="cancellationToken">Cancellation token to cancel the operation.</param>
        /// <returns>Task.</returns>
        Task SetCountAsync(int count, CancellationToken cancellationToken);
    }

    /// <summary>
    /// Test Actor Class.
    /// </summary>
    internal class TestActor : Actor,  ITestActor
    {
        public TestActor(ActorHost host, IActorStateManager stateManager = null, IDaprInteractor daprInteractor = null)
            : base(host)
        {
            if (stateManager != null)
            {
                this.StateManager = stateManager;
            }

            if(daprInteractor != null)
            {
                this.DaprInteractor = daprInteractor;
            }
        }

        /// <inheritdoc/>
        public Task<int> GetCountAsync(CancellationToken cancellationToken)
        {
            return Task.FromResult(5);
        }

        /// <inheritdoc/>
        public Task SetCountAsync(int count, CancellationToken cancellationToken)
        {
            return Task.CompletedTask;
        }

        public Task SaveTestState()
        {
            return this.SaveStateAsync();
        }

        public Task ResetTestStateAsync()
        {
            return this.ResetStateAsync();
        }

        public void TimerCallbackNonTaskReturnType()
        {
        }

        public Task TimerCallbackTwoArguments(int i, int j)
        {
            Console.WriteLine(i + j);
            return default;
        }

        private Task TimerCallbackPrivate()
        {
            return default;
        }

        protected Task TimerCallbackProtected()
        {
            return default;
        }

        internal Task TimerCallbackInternal()
        {
            return default;
        }

        public Task TimerCallbackPublicWithNoArguments()
        {
            return default;
        }

        public Task TimerCallbackPublicWithOneArgument(int i)
        {
            return default;
        }

        public Task TimerCallbackOverloaded()
        {
            return default;
        }

        public Task TimerCallbackOverloaded(int i)
        {
            return default;
        }

        public static Task TimerCallbackStatic()
        {
            return default;
        }
    }
}