public static TreatmentType CreateTreatment(Treatment.TreatmentEnum treatmentEnum)
{
switch(treatmentEnum)
{
case Treatment.TreatmentEnum.Dietary:
return new Dietary();
break;
case Treatment.TreatmentEnum.Hydration:
return new Hydration();
break;
case Treatment.TreatmentEnum.Prescription:
return new Prescription();
break;
case Treatment.TreatmentEnum.Referral:
return new Referral();
break;
case Treatment.TreatmentEnum.Surgery:
return new Surgery();
break;
}
return null;
}
public void Constructor_TreatmentEnumParam_ReturnsCorrectCourseOfActionProperty()
{
var sut = new Treatment(Treatment.TreatmentEnum.Dietary);
sut.TreatmentType.CourseOfAction.ShouldEqual("Give dietary advice.");
}
public void AddTreatment(Treatment treatment)
{
this.Treatments.Add(treatment);
}
public void Constructor_NoInputParams_IsInstanceOfDomainEntity()
{
var sut = new Treatment(Treatment.TreatmentEnum.Surgery);
sut.ShouldBeInstanceOf<DomainEntity>();
}