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;
}
