public void MWQMPlanController__MWQMPlanAddOrModify_Test() { foreach (CultureInfo culture in setupData.cultureListGood) { SetupTest(contactModelListGood[0], culture, controllerAction); using (TransactionScope ts = new TransactionScope()) { RetAddMWQMPlan retAddMWQMPlan = AddMWQMPlanAndSubElements((culture.TwoLetterISOLanguageName == "fr" ? LanguageEnum.fr : LanguageEnum.en)); PartialViewResult partialViewResult = controller._MWQMPlanAddOrModify(retAddMWQMPlan.TVItemModelNB.TVItemID, retAddMWQMPlan.MWQMPlanModel.MWQMPlanID) as PartialViewResult; Assert.IsNotNull(partialViewResult); MWQMPlanController mwqmPlanController = (MWQMPlanController)partialViewResult.ViewBag.MWQMPlanController; Assert.IsNotNull(mwqmPlanController); int ProvinceTVItemIDRet = (int)partialViewResult.ViewBag.ProvinceTVItemID; Assert.AreEqual(retAddMWQMPlan.TVItemModelNB.TVItemID, ProvinceTVItemIDRet); int MWQMPlanIDRet = (int)partialViewResult.ViewBag.MWQMPlanID; Assert.AreEqual(retAddMWQMPlan.MWQMPlanModel.MWQMPlanID, MWQMPlanIDRet); MWQMPlanModel MWQMPlanModelRet = (MWQMPlanModel)partialViewResult.ViewBag.MWQMPlanModel; Assert.IsNotNull(MWQMPlanModelRet); Assert.AreEqual(retAddMWQMPlan.MWQMPlanModel.MWQMPlanID, MWQMPlanModelRet.MWQMPlanID); List <MWQMPlanSubsectorModel> MWQMPlanSubsectorModelListRet = (List <MWQMPlanSubsectorModel>)partialViewResult.ViewBag.MWQMPlanSubsectorModelList; Assert.IsTrue(MWQMPlanSubsectorModelListRet.Count == 1); List <ContactModel> AdminContactModelListRet = (List <ContactModel>)partialViewResult.ViewBag.AdminContactModelList; Assert.IsNull(AdminContactModelListRet); // because IsMWQMPlanner == true ContactModel ContactModelRet = (ContactModel)partialViewResult.ViewBag.ContactModel; Assert.IsNotNull(ContactModelRet); Assert.AreEqual(contactModelListGood[0].ContactTVItemID, ContactModelRet.ContactTVItemID); List <TVItemModel> TVItemModelSubsectorListRet = (List <TVItemModel>)partialViewResult.ViewBag.TVItemModelSubsectorList; Assert.IsTrue(TVItemModelSubsectorListRet.Count > 0); List <SampleTypeEnumTextOrdered> SampleTypeEnumTextOrderedListRet = (List <SampleTypeEnumTextOrdered>)partialViewResult.ViewBag.SampleTypeEnumTextOrderedList; Assert.IsTrue(SampleTypeEnumTextOrderedListRet.Count > 0); SampleTypeEnumTextOrdered SampleTypeEnumTextOrderedFirstRet = (SampleTypeEnumTextOrdered)partialViewResult.ViewBag.SampleTypeEnumTextOrderedFirst; Assert.IsNotNull(SampleTypeEnumTextOrderedFirstRet); Assert.AreEqual(SampleTypeEnum.Routine, SampleTypeEnumTextOrderedFirstRet.SampleType); bool IsMWQMPlannerRet = (bool)partialViewResult.ViewBag.IsMWQMPlanner; Assert.IsTrue(IsMWQMPlannerRet); } } }
public PartialViewResult _SamplingPlanAddOrModify(int ProvinceTVItemID, int SamplingPlanID) { ViewBag.SamplingPlanController = _SamplingPlanController; ViewBag.ProvinceTVItemID = ProvinceTVItemID; ViewBag.SamplingPlanID = SamplingPlanID; ViewBag.SamplingPlanModel = null; ViewBag.SamplingPlanSubsectorModelList = null; ViewBag.AdminContactModelList = null; ViewBag.ContactModel = null; ViewBag.TVItemModelSubsectorList = null; ViewBag.SampleTypeEnumTextOrderedList = null; ViewBag.SamplingPlanTypeEnumTextOrderedList = null; ViewBag.LabSheetTypeEnumTextOrderedList = null; ViewBag.SampleTypeEnumTextOrderedFirst = null; ViewBag.IsSamplingPlanner = null; List <SampleTypeEnumTextOrdered> sampleTypeEnumTextOrderedList = _BaseEnumService.GetSampleTypeEnumTextOrderedList(); ViewBag.SampleTypeEnumTextOrderedList = sampleTypeEnumTextOrderedList; SampleTypeEnumTextOrdered sampleTypeEnumTextOrderedFirst = sampleTypeEnumTextOrderedList.Where(c => c.SampleType == SampleTypeEnum.Routine).FirstOrDefault(); ViewBag.SampleTypeEnumTextOrderedFirst = sampleTypeEnumTextOrderedFirst; List <SamplingPlanTypeEnumTextOrdered> samplingPlanTypeEnumTextOrderedList = _BaseEnumService.GetSamplingPlanTypeEnumTextOrderedList(); ViewBag.SamplingPlanTypeEnumTextOrderedList = samplingPlanTypeEnumTextOrderedList; List <LabSheetTypeEnumTextOrdered> labSheetTypeEnumTextOrderedList = _BaseEnumService.GetLabSheetTypeEnumTextOrderedList(); ViewBag.LabSheetTypeEnumTextOrderedList = labSheetTypeEnumTextOrderedList; ContactModel contactModel = _ContactService.GetContactLoggedInDB(); ViewBag.ContactModel = contactModel; bool IsSamplingPlanner = false; if (contactModel.SamplingPlanner_ProvincesTVItemID.Contains(ProvinceTVItemID.ToString())) { IsSamplingPlanner = true; } else { List <ContactModel> adminContactModelList = _ContactService.GetAdminContactModelListDB(); ViewBag.AdminContactModelList = adminContactModelList; } ViewBag.IsSamplingPlanner = IsSamplingPlanner; if (IsSamplingPlanner) { ViewBag.ContactModel = contactModel; if (contactModel.SamplingPlanner_ProvincesTVItemID.Contains(ProvinceTVItemID.ToString())) { List <SamplingPlanSubsectorModel> SamplingPlanSubsectorModelList = _SamplingPlanSubsectorService.GetSamplingPlanSubsectorModelListWithSamplingPlanIDDB(SamplingPlanID); ViewBag.SamplingPlanSubsectorModelList = SamplingPlanSubsectorModelList; List <TVItemModel> tvItemModelSubsectorList = _TVItemService.GetChildrenTVItemModelListWithTVItemIDAndTVTypeDB(ProvinceTVItemID, TVTypeEnum.Subsector); ViewBag.TVItemModelSubsectorList = tvItemModelSubsectorList; } if (SamplingPlanID != 0) { SamplingPlanModel SamplingPlanModel = _SamplingPlanService.GetSamplingPlanModelWithSamplingPlanIDDB(SamplingPlanID); ViewBag.SamplingPlanModel = SamplingPlanModel; } } return(PartialView()); }