public ActionResult Create(UnitTemplateViewModel unitTemplateViewModel)
        {
            if (ModelState.IsValid)
            {
                var unitTemplate = Mapper.Map <UnitTemplate>(unitTemplateViewModel);
                Data.UnitTemplates.Add(unitTemplate);
                Data.SaveChanges();
                return(RedirectToAction("Index"));
            }

            return(View(unitTemplateViewModel));
        }
        public ActionResult Edit(UnitTemplateViewModel unitTemplateViewModel)
        {
            if (ModelState.IsValid)
            {
                UnitTemplate unitTemplate = this.Mapper.Map <UnitTemplate>(unitTemplateViewModel);
                if (unitTemplateViewModel.ImageFromView == null)
                {
                    unitTemplate.Image = (byte[])TempData["image"];
                }

                if (TempData.ContainsKey("image"))
                {
                    TempData.Remove("image");
                }

                this.Data.UnitTemplates.Update(unitTemplate);
                this.Data.SaveChanges();
                return(RedirectToAction("Index"));
            }

            ViewBag.ImageId = new SelectList(this.Data.UnitTemplates.All(), "Id", "Name", unitTemplateViewModel.Id);
            return(View(unitTemplateViewModel));
        }