public void setInvoiceRemarkTest() { IVHelper target = new IVHelper(); // TODO: Initialize to an appropriate value //Insert remark ABiRemark remark = new ABiRemark() { InvoiceID = 99999999, RoleID = 4, Remark = "测试用的", RemarkPerson = "Unit Test", RoleName = "供应商", Username = "******", DateCreated = DateTime.Now }; // TODO: Initialize to an appropriate value ABiRemark actual; actual = target.setInvoiceRemark(remark); Assert.IsNotNull(actual); //Update remark actual.Remark = "测试更新备注"; actual.RemarkPerson = "Unit单元测试"; actual = target.setInvoiceRemark(remark); var obj = target.getInvoiceRemarks(99999999); Assert.AreEqual(1, obj.Count); Assert.AreEqual(obj[0].Username, "sgptp"); var dbContext = new VPDAL.ABInvVPEntities(); dbContext.ExecuteStoreCommand(String.Format("delete from ABi_Remark where ABi_Invoice_ID = {0}", 99999999)); obj = target.getInvoiceRemarks(99999999); Assert.IsTrue(obj == null || obj.Count == 0); }
/// <summary> /// Add or update the remark info. /// </summary> /// <param name="remark"></param> /// <returns></returns> public ABiRemark setInvoiceRemark(ABiRemark remark) { if (remark.ID > 0) { //update the remark var obj = dbContent.ABi_Remark.FirstOrDefault(o => o.ABi_Remark_ID == remark.ID); var users = new Users.UsersHelper().getUsers(remark.Username); if (obj != null && users != null && users.Count > 0) { obj.ABi_Invoice_ID = remark.InvoiceID; obj.ABi_Admin_User_ID = users[0].ABi_Admin_Users_ID; obj.DateModified = DateTime.Now; obj.Remark = remark.Remark; obj.RemarkPerson = remark.RemarkPerson; obj.ABi_Admin_Role_ID = remark.RoleID; if (remark.RoleID == 3) { var ivObj = dbContent.ABi_Invoice.FirstOrDefault(o => o.ABi_Invoice_ID == remark.InvoiceID); if (ivObj != null) { if (ivObj.InvoiceStatus == "1") { ivObj.InvoiceStatus = "2"; } } } dbContent.SaveChanges(); //remark.ID = obj.ABi_Remark_ID; } } else { //insert the remark var users = new Users.UsersHelper().getUsers(remark.Username); if (users != null && users.Count > 0) { var r = new VPDAL.ABi_Remark() { ABi_Invoice_ID = remark.InvoiceID, ABi_Admin_User_ID = users[0].ABi_Admin_Users_ID, DateCreated = DateTime.Now, Remark = remark.Remark, RemarkPerson = remark.RemarkPerson, DateModified = DateTime.Now, ABi_Admin_Role_ID = remark.RoleID }; dbContent.AddToABi_Remark(r); dbContent.SaveChanges(); remark.ID = r.ABi_Remark_ID; } } return remark; }