// ------------------------------------------------------------------------
// Copyright 2021 The Dapr Authors
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//     http://www.apache.org/licenses/LICENSE-2.0
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// ------------------------------------------------------------------------

namespace Dapr.Workflow.Test
{
    using Moq;
    using System.Threading.Tasks;
    using Xunit;
    using Xunit.Sdk;

    /// <summary>
    /// Contains tests for WorkflowActivityContext.
    /// </summary>
    public class WorkflowActivityTest
    {
        private IWorkflowActivity workflowActivity;

        private Mock<WorkflowActivityContext> workflowActivityContextMock;

        public WorkflowActivityTest()
        {
            this.workflowActivity = new TestDaprWorkflowActivity();
            this.workflowActivityContextMock = new Mock<WorkflowActivityContext>();
        }

        [Fact]
        public async Task RunAsync_ShouldReturnCorrectContextInstanceId()
        {
            this.workflowActivityContextMock.Setup((x) => x.InstanceId).Returns("instanceId");

            string result = (string) (await this.workflowActivity.RunAsync(this.workflowActivityContextMock.Object, "input"))!;

            Assert.Equal("instanceId", result);
        }


        public class TestDaprWorkflowActivity : WorkflowActivity<string, string>
        {
            public override Task<string> RunAsync(WorkflowActivityContext context, string input)
            {
                return Task.FromResult(context.InstanceId);
            }
        }
    }
}