import { IMock, It, Mock, Times } from 'typemoq'
import { StartedByProxy } from './started-by-proxy'
import { Persistence } from '../persistence'
import { WorkflowHandlerFn } from './workflow-handler-fn'
import { TestCommand, TestWorkflowData } from '../../test'
import { Logger } from '@node-ts/logger-core'
import { WorkflowStatus } from '../workflow-data'
import { MessageAttributes } from '@node-ts/bus-messages'

describe('StartedByProxy', () => {
  let persistence: IMock<Persistence>
  let handler: IMock<WorkflowHandlerFn<TestCommand, TestWorkflowData>>
  let logger: IMock<Logger>
  let sut: StartedByProxy<TestCommand, TestWorkflowData>

  beforeEach(() => {
    persistence = Mock.ofType<Persistence>()
    logger = Mock.ofType<Logger>()

    handler = Mock.ofType<WorkflowHandlerFn<TestCommand, TestWorkflowData>>()
    handler.setup(x => x.name).returns(() => 'handler-name')

    sut = new StartedByProxy<TestCommand, TestWorkflowData>(
      TestWorkflowData,
      handler.object,
      persistence.object,
      logger.object
    )
  })

  describe('when handling messages', () => {
    let command: TestCommand
    const messageOptions = new MessageAttributes()
    let dataOutput: Partial<TestWorkflowData>

    beforeEach(async () => {
      command = new TestCommand('abc')

      dataOutput = { property1: command.property1 }
      handler
        // tslint:disable-next-line:no-unsafe-any
        .setup(x => x(command, It.isAny(), messageOptions))
        .returns(async () => dataOutput)

      await sut.handle(command, messageOptions)
    })

    it('should dispatch the message to the handler', () => {
      handler.verify(
        x => x(
          command,
          It.is((data: TestWorkflowData) => !!data && data.$version === 0 && data.$status === WorkflowStatus.Running),
          messageOptions
        ),
        Times.once())
    })

    it('should save the workflow data', () => {
      persistence.verify(
        async x => x.saveWorkflowData(It.isObjectWith<TestWorkflowData>({
          ...dataOutput,
          $version: 0
        })),
        Times.once()
      )
    })
  })

  describe('when the workflow is completed', () => {
    let command: TestCommand
    const messageOptions = new MessageAttributes()

    beforeEach(async () => {
      command = new TestCommand('abc')

      let dataOutput: TestWorkflowData
      handler
        // tslint:disable-next-line:no-unsafe-any
        .setup(x => x(command, It.isAny(), messageOptions))
        .callback(() => {
          dataOutput = { ...dataOutput, $status: WorkflowStatus.Complete }
        })
        .returns(async () => dataOutput)

      await sut.handle(command, messageOptions)
    })

    it('should mark the workflow as complete', () => {
      persistence.verify(
        async x =>
          x.saveWorkflowData(It.isObjectWith<TestWorkflowData>({ $status: WorkflowStatus.Complete })),
        Times.once()
      )
    })
  })
})