using System;
using System.Collections.Generic;
using Landmark.OLA.Models.Individuals;
using Landmark.OLA.Models.Individuals.Enums;

namespace Landmark.OLA.Tests.TestData
{
    public class TestIncomes
    {
        private static Income GetValidIncome()
        {
            return new Income
            {
                IncomeTypeCD = IncomeType.Salary,
                SourceName = "Salary",
                AmountPerPeriod = 1000,
                Frequency = Frequency.Monthly,
                IncomeStatus = IncomeStatus.IncomeStatus_Included
            };
        }

        public static Income GetValidEmploymentIncome()
        {
            return GetValidIncome();
        }

        public static IEnumerable<Object[]> GetValidTestIncomes()
        {
            yield return new Object[] { GetValidIncome() };
        }

        public static IEnumerable<Object[]> GetInvalidTestIncomes()
        {
            var missingSourceName = GetValidIncome();
            missingSourceName.SourceName = null;
            yield return new Object[] { missingSourceName };
        }
    }
}