(req: express.Request, res: express.Response, next: express.NextFunction) => {
      const draft: Draft<ResponseDraft> = res.locals.draft

      res.locals.monthlyIncomeAmount = draft.document.statementOfMeans ? CalculateMonthlyIncomeExpense.calculateTotalAmount(
        IncomeExpenseSources.fromMonthlyIncomeFormModel(draft.document.statementOfMeans.monthlyIncome).incomeExpenseSources
      ) : 0
      res.locals.monthlyExpensesAmount = draft.document.statementOfMeans ? CalculateMonthlyIncomeExpense.calculateTotalAmount(
        IncomeExpenseSources.fromMonthlyExpensesFormModel(draft.document.statementOfMeans.monthlyExpenses).incomeExpenseSources
      ) : 0
      next()
    }

  it('should calculate total amount using weekly schedule', () => {

    const incomeExpenseSource = [ { 'amount': 100, 'schedule': IncomeExpenseSchedule.WEEK } ]
    expect(CalculateMonthlyIncomeExpense
      .calculateTotalAmount(incomeExpenseSource))
      .to.equal(433.33)
  })

function calculateTotalMonthlyIncomeExpense (model: MonthlyIncome): number {
  if (!model) {
    return undefined
  }
  const incomeExpenseSources = IncomeExpenseSources.fromMonthlyIncomeFormModel(model)

  if (!isValid(incomeExpenseSources)) {
    return undefined
  }

  return CalculateMonthlyIncomeExpense.calculateTotalAmount(
    incomeExpenseSources.incomeExpenseSources
  )
}

  .post(Paths.totalIncomeOrExpensesCalculation.uri, async (req: express.Request, res: express.Response) => {
    try {
      const incomeExpenseSources: IncomeExpenseSources = IncomeExpenseSources.fromObject(req.body)

      const validator: Validator = new Validator()
      const error: ValidationError[] = await validator.validate(incomeExpenseSources)
      if (error.length > 0) {
        return res.status(HttpStatus.UNPROCESSABLE_ENTITY).send(error)
      }

      const totalMonthlyIncomeExpense = CalculateMonthlyIncomeExpense
        .calculateTotalAmount(incomeExpenseSources.incomeExpenseSources)
      return res.status(HttpStatus.OK).json({
        totalMonthlyIncomeExpense: totalMonthlyIncomeExpense
      })
    } catch (err) {
      return res.status(HttpStatus.UNPROCESSABLE_ENTITY).send(err)
    }
  })

    (req: express.Request, res: express.Response, next: express.NextFunction) => {
      const claim: Claim = res.locals.claim
      const response = claim.response as FullAdmissionResponse | PartialAdmissionResponse
      res.locals.monthlyIncomeAmount = response.statementOfMeans && response.statementOfMeans.incomes ? CalculateMonthlyIncomeExpense.calculateTotalAmount(
        response.statementOfMeans.incomes.map(income => IncomeExpenseSource.fromClaimIncome(income))
      ) : 0
      res.locals.monthlyExpensesAmount = response.statementOfMeans && response.statementOfMeans.expenses ? CalculateMonthlyIncomeExpense.calculateTotalAmount(
        response.statementOfMeans.expenses.map(expense => IncomeExpenseSource.fromClaimExpense(expense))
      ) : 0
      res.locals.statementOfMeans = response.statementOfMeans

      next()
    }