/// <summary>
        /// Page of treatment detail
        /// </summary>
        /// <returns></returns>
        public IActionResult Index()
        {
            // Get by role
            List <Treatment> items = new List <Treatment>();

            if (User.IsInRole("doctor"))
            {
                items = treatmentRepository.GetByDoctor(GetUserId());
            }
            else if (User.IsInRole("patient"))
            {
                items = treatmentRepository.GetByPatient(GetUserId());
            }

            // Parse to viewmodel
            TreatmentViewModel vm = new TreatmentViewModel()
            {
                treatments = new List <TreatmentDetailViewModel>()
            };

            // add treatments
            foreach (Treatment treatment in items)
            {
                vm.treatments.Add(TreatmentConverter.ModelToViewModel(treatment));
            }

            // Return view with treatments
            return(View(vm.treatments));
        }
        public void GetByDoctorFalseInput()
        {
            EmptyLists();
            treatmentRepository = new TreatmentRepository(context);
            Exception ex = Assert.Throws <NullReferenceException>(() => treatmentRepository.GetByDoctor(-1));

            Assert.Equal("Het dokterId is leeg.", ex.Message);
        }
 public void GetByDoctor()
 {
     EmptyLists();
     treatmentRepository = new TreatmentRepository(context);
     Assert.Equal(3, treatmentRepository.GetByDoctor(11).Count);
 }