2024-11-13 18:18:28 +08:00

480 lines
19 KiB
C#

using FreeSql.DataAnnotations;
using System;
using System.Collections.Generic;
using System.Linq;
using Xunit;
namespace FreeSql.Tests.GBase
{
public class GBaseInsertOrUpdateTest
{
IFreeSql fsql => g.gbase;
[Fact]
public void InsertOrUpdate_OnlyPrimary()
{
fsql.Delete<tbiou01>().Where("1=1").ExecuteAffrows();
var iou = fsql.InsertOrUpdate<tbiou01>().SetSource(new tbiou01 { id = 1 });
var sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou01 t1
USING (SELECT 1 as id FROM dual ) t2 ON (t1.id = t2.id)
WHEN NOT MATCHED THEN
insert (id)
values (t2.id)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou01>().SetSource(new tbiou01 { id = 1 });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou01 t1
USING (SELECT 1 as id FROM dual ) t2 ON (t1.id = t2.id)
WHEN NOT MATCHED THEN
insert (id)
values (t2.id)", sql);
Assert.Equal(0, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou01>().SetSource(new tbiou01 { id = 2 });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou01 t1
USING (SELECT 2 as id FROM dual ) t2 ON (t1.id = t2.id)
WHEN NOT MATCHED THEN
insert (id)
values (t2.id)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou01>().SetSource(new[] { new tbiou01 { id = 1 }, new tbiou01 { id = 2 }, new tbiou01 { id = 3 }, new tbiou01 { id = 4 } });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou01 t1
USING (SELECT 1 as id FROM dual
UNION ALL
SELECT 2 FROM dual
UNION ALL
SELECT 3 FROM dual
UNION ALL
SELECT 4 FROM dual ) t2 ON (t1.id = t2.id)
WHEN NOT MATCHED THEN
insert (id)
values (t2.id)", sql);
Assert.Equal(2, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou01>().SetSource(new[] { new tbiou01 { id = 1 }, new tbiou01 { id = 2 }, new tbiou01 { id = 3 }, new tbiou01 { id = 4 } });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou01 t1
USING (SELECT 1 as id FROM dual
UNION ALL
SELECT 2 FROM dual
UNION ALL
SELECT 3 FROM dual
UNION ALL
SELECT 4 FROM dual ) t2 ON (t1.id = t2.id)
WHEN NOT MATCHED THEN
insert (id)
values (t2.id)", sql);
Assert.Equal(0, iou.ExecuteAffrows());
}
class tbiou01
{
public int id { get; set; }
}
[Fact]
public void InsertOrUpdate_OnePrimary()
{
fsql.Delete<tbiou02>().Where("1=1").ExecuteAffrows();
var iou = fsql.InsertOrUpdate<tbiou02>().SetSource(new tbiou02 { id = 1, name = "01" });
var sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou02 t1
USING (SELECT 1 as id, '01' as name FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id, name)
values (t2.id, t2.name)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou02>().SetSource(new tbiou02 { id = 1, name = "011" });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou02 t1
USING (SELECT 1 as id, '011' as name FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id, name)
values (t2.id, t2.name)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou02>().SetSource(new tbiou02 { id = 2, name = "02" });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou02 t1
USING (SELECT 2 as id, '02' as name FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id, name)
values (t2.id, t2.name)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou02>().SetSource(new[] { new tbiou02 { id = 1, name = "01" }, new tbiou02 { id = 2, name = "02" }, new tbiou02 { id = 3, name = "03" }, new tbiou02 { id = 4, name = "04" } });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou02 t1
USING (SELECT 1 as id, '01' as name FROM dual
UNION ALL
SELECT 2, '02' FROM dual
UNION ALL
SELECT 3, '03' FROM dual
UNION ALL
SELECT 4, '04' FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id, name)
values (t2.id, t2.name)", sql);
Assert.Equal(4, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou02>().SetSource(new[] { new tbiou02 { id = 1, name = "001" }, new tbiou02 { id = 2, name = "002" }, new tbiou02 { id = 3, name = "003" }, new tbiou02 { id = 4, name = "004" } });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou02 t1
USING (SELECT 1 as id, '001' as name FROM dual
UNION ALL
SELECT 2, '002' FROM dual
UNION ALL
SELECT 3, '003' FROM dual
UNION ALL
SELECT 4, '004' FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id, name)
values (t2.id, t2.name)", sql);
Assert.Equal(4, iou.ExecuteAffrows());
var lst = fsql.Select<tbiou02>().Where(a => new[] { 1, 2, 3, 4 }.Contains(a.id)).ToList();
Assert.Equal(4, lst.Where(a => a.name == "00" + a.id).Count());
}
class tbiou02
{
public int id { get; set; }
public string name { get; set; }
}
[Fact]
public void InsertOrUpdate_OnePrimaryAndIdentity()
{
fsql.Delete<tbiou022>().Where("1=1").ExecuteAffrows();
var iou = fsql.InsertOrUpdate<tbiou022>().SetSource(new tbiou022 { id = 1, name = "01" });
var sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou022 t1
USING (SELECT 1 as id, '01' as name FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id, name)
values (t2.id, t2.name)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou022>().SetSource(new tbiou022 { id = 1, name = "011" });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou022 t1
USING (SELECT 1 as id, '011' as name FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id, name)
values (t2.id, t2.name)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou022>().SetSource(new tbiou022 { id = 2, name = "02" });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou022 t1
USING (SELECT 2 as id, '02' as name FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id, name)
values (t2.id, t2.name)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou022>().SetSource(new[] { new tbiou022 { id = 1, name = "01" }, new tbiou022 { id = 2, name = "02" }, new tbiou022 { id = 3, name = "03" }, new tbiou022 { id = 4, name = "04" } });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou022 t1
USING (SELECT 1 as id, '01' as name FROM dual
UNION ALL
SELECT 2, '02' FROM dual
UNION ALL
SELECT 3, '03' FROM dual
UNION ALL
SELECT 4, '04' FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id, name)
values (t2.id, t2.name)", sql);
Assert.Equal(4, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou022>().SetSource(new[] { new tbiou022 { id = 1, name = "001" }, new tbiou022 { id = 2, name = "002" }, new tbiou022 { id = 3, name = "003" }, new tbiou022 { id = 4, name = "004" } });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou022 t1
USING (SELECT 1 as id, '001' as name FROM dual
UNION ALL
SELECT 2, '002' FROM dual
UNION ALL
SELECT 3, '003' FROM dual
UNION ALL
SELECT 4, '004' FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id, name)
values (t2.id, t2.name)", sql);
Assert.Equal(4, iou.ExecuteAffrows());
var lst = fsql.Select<tbiou022>().Where(a => new[] { 1, 2, 3, 4 }.Contains(a.id)).ToList();
//Assert.Equal(4, lst.Where(a => a.name == "00" + a.id).Count());
//--no primary
iou = fsql.InsertOrUpdate<tbiou022>().SetSource(new tbiou022 { name = "01" });
sql = iou.ToSql();
Assert.Equal(@"INSERT INTO tbiou022(name) VALUES('01')", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou022>().SetSource(new tbiou022 { name = "011" });
sql = iou.ToSql();
Assert.Equal(@"INSERT INTO tbiou022(name) VALUES('011')", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou022>().SetSource(new tbiou022 { name = "02" });
sql = iou.ToSql();
Assert.Equal(@"INSERT INTO tbiou022(name) VALUES('02')", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou022>().SetSource(new[] { new tbiou022 { name = "01" }, new tbiou022 { name = "02" }, new tbiou022 { name = "03" }, new tbiou022 { name = "04" } });
sql = iou.ToSql();
Assert.Equal(@"INSERT INTO tbiou022(name)
SELECT * FROM (
SELECT '01' FROM dual
UNION ALL
SELECT '02' FROM dual
UNION ALL
SELECT '03' FROM dual
UNION ALL
SELECT '04' FROM dual
) ftbtmp", sql);
Assert.Equal(4, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou022>().SetSource(new[] { new tbiou022 { name = "001" }, new tbiou022 { name = "002" }, new tbiou022 { name = "003" }, new tbiou022 { name = "004" } });
sql = iou.ToSql();
Assert.Equal(@"INSERT INTO tbiou022(name)
SELECT * FROM (
SELECT '001' FROM dual
UNION ALL
SELECT '002' FROM dual
UNION ALL
SELECT '003' FROM dual
UNION ALL
SELECT '004' FROM dual
) ftbtmp", sql);
Assert.Equal(4, iou.ExecuteAffrows());
//--no primary and yes
iou = fsql.InsertOrUpdate<tbiou022>().SetSource(new[] { new tbiou022 { id = 1, name = "100001" }, new tbiou022 { name = "00001" }, new tbiou022 { id = 2, name = "100002" }, new tbiou022 { name = "00002" }, new tbiou022 { id = 3, name = "100003" }, new tbiou022 { name = "00003" }, new tbiou022 { id = 4, name = "100004" }, new tbiou022 { name = "00004" } });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou022 t1
USING (SELECT 1 as id, '100001' as name FROM dual
UNION ALL
SELECT 2, '100002' FROM dual
UNION ALL
SELECT 3, '100003' FROM dual
UNION ALL
SELECT 4, '100004' FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id, name)
values (t2.id, t2.name)
;
INSERT INTO tbiou022(name)
SELECT * FROM (
SELECT '00001' FROM dual
UNION ALL
SELECT '00002' FROM dual
UNION ALL
SELECT '00003' FROM dual
UNION ALL
SELECT '00004' FROM dual
) ftbtmp", sql);
Assert.Equal(8, iou.ExecuteAffrows());
lst = fsql.Select<tbiou022>().Where(a => new[] { 1, 2, 3, 4 }.Contains(a.id)).ToList();
//Assert.Equal(4, lst.Where(a => a.name == "10000" + a.id).Count());
}
class tbiou022
{
[Column(IsIdentity = true)]
public int id { get; set; }
public string name { get; set; }
}
[Fact]
public void InsertOrUpdate_TwoPrimary()
{
fsql.Delete<tbiou03>().Where("1=1").ExecuteAffrows();
var iou = fsql.InsertOrUpdate<tbiou03>().SetSource(new tbiou03 { id1 = 1, id2 = "01", name = "01" });
var sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou03 t1
USING (SELECT 1 as id1, '01' as id2, '01' as name FROM dual ) t2 ON (t1.id1 = t2.id1 AND t1.id2 = t2.id2)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id1, id2, name)
values (t2.id1, t2.id2, t2.name)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou03>().SetSource(new tbiou03 { id1 = 1, id2 = "01", name = "011" });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou03 t1
USING (SELECT 1 as id1, '01' as id2, '011' as name FROM dual ) t2 ON (t1.id1 = t2.id1 AND t1.id2 = t2.id2)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id1, id2, name)
values (t2.id1, t2.id2, t2.name)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou03>().SetSource(new tbiou03 { id1 = 2, id2 = "02", name = "02" });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou03 t1
USING (SELECT 2 as id1, '02' as id2, '02' as name FROM dual ) t2 ON (t1.id1 = t2.id1 AND t1.id2 = t2.id2)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id1, id2, name)
values (t2.id1, t2.id2, t2.name)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou03>().SetSource(new[] { new tbiou03 { id1 = 1, id2 = "01", name = "01" }, new tbiou03 { id1 = 2, id2 = "02", name = "02" }, new tbiou03 { id1 = 3, id2 = "03", name = "03" }, new tbiou03 { id1 = 4, id2 = "04", name = "04" } });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou03 t1
USING (SELECT 1 as id1, '01' as id2, '01' as name FROM dual
UNION ALL
SELECT 2, '02', '02' FROM dual
UNION ALL
SELECT 3, '03', '03' FROM dual
UNION ALL
SELECT 4, '04', '04' FROM dual ) t2 ON (t1.id1 = t2.id1 AND t1.id2 = t2.id2)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id1, id2, name)
values (t2.id1, t2.id2, t2.name)", sql);
Assert.Equal(4, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou03>().SetSource(new[] { new tbiou03 { id1 = 1, id2 = "01", name = "001" }, new tbiou03 { id1 = 2, id2 = "02", name = "002" }, new tbiou03 { id1 = 3, id2 = "03", name = "003" }, new tbiou03 { id1 = 4, id2 = "04", name = "004" } });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou03 t1
USING (SELECT 1 as id1, '01' as id2, '001' as name FROM dual
UNION ALL
SELECT 2, '02', '002' FROM dual
UNION ALL
SELECT 3, '03', '003' FROM dual
UNION ALL
SELECT 4, '04', '004' FROM dual ) t2 ON (t1.id1 = t2.id1 AND t1.id2 = t2.id2)
WHEN MATCHED THEN
update set t1.name = t2.name
WHEN NOT MATCHED THEN
insert (id1, id2, name)
values (t2.id1, t2.id2, t2.name)", sql);
Assert.Equal(4, iou.ExecuteAffrows());
var lst = fsql.Select<tbiou03>().Where(a => a.id1 == 1 && a.id2 == "01" || a.id1 == 2 && a.id2 == "02" || a.id1 == 3 && a.id2 == "03" || a.id1 == 4 && a.id2 == "04").ToList();
Assert.Equal(4, lst.Where(a => a.name == "00" + a.id1).Count());
}
class tbiou03
{
[Column(IsPrimary = true)]
public int id1 { get; set; }
[Column(IsPrimary = true)]
public string id2 { get; set; }
public string name { get; set; }
}
[Fact]
public void InsertOrUpdate_OnePrimaryAndVersionAndCanUpdate()
{
fsql.Delete<tbiou04>().Where("1=1").ExecuteAffrows();
var iou = fsql.InsertOrUpdate<tbiou04>().SetSource(new tbiou04 { id = 1, name = "01" });
var sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou04 t1
USING (SELECT 1 as id, '01' as name, 0 as version, current as CreateTime FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name, t1.version = t1.version + 1
WHEN NOT MATCHED THEN
insert (id, name, version, CreateTime)
values (t2.id, t2.name, t2.version, t2.CreateTime)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou04>().SetSource(new tbiou04 { id = 1, name = "011" });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou04 t1
USING (SELECT 1 as id, '011' as name, 0 as version, current as CreateTime FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name, t1.version = t1.version + 1
WHEN NOT MATCHED THEN
insert (id, name, version, CreateTime)
values (t2.id, t2.name, t2.version, t2.CreateTime)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou04>().SetSource(new tbiou04 { id = 2, name = "02" });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou04 t1
USING (SELECT 2 as id, '02' as name, 0 as version, current as CreateTime FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name, t1.version = t1.version + 1
WHEN NOT MATCHED THEN
insert (id, name, version, CreateTime)
values (t2.id, t2.name, t2.version, t2.CreateTime)", sql);
Assert.Equal(1, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou04>().SetSource(new[] { new tbiou04 { id = 1, name = "01" }, new tbiou04 { id = 2, name = "02" }, new tbiou04 { id = 3, name = "03" }, new tbiou04 { id = 4, name = "04" } });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou04 t1
USING (SELECT 1 as id, '01' as name, 0 as version, current as CreateTime FROM dual
UNION ALL
SELECT 2, '02', 0, current FROM dual
UNION ALL
SELECT 3, '03', 0, current FROM dual
UNION ALL
SELECT 4, '04', 0, current FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name, t1.version = t1.version + 1
WHEN NOT MATCHED THEN
insert (id, name, version, CreateTime)
values (t2.id, t2.name, t2.version, t2.CreateTime)", sql);
Assert.Equal(4, iou.ExecuteAffrows());
iou = fsql.InsertOrUpdate<tbiou04>().SetSource(new[] { new tbiou04 { id = 1, name = "001" }, new tbiou04 { id = 2, name = "002" }, new tbiou04 { id = 3, name = "003" }, new tbiou04 { id = 4, name = "004" } });
sql = iou.ToSql();
Assert.Equal(@"MERGE INTO tbiou04 t1
USING (SELECT 1 as id, '001' as name, 0 as version, current as CreateTime FROM dual
UNION ALL
SELECT 2, '002', 0, current FROM dual
UNION ALL
SELECT 3, '003', 0, current FROM dual
UNION ALL
SELECT 4, '004', 0, current FROM dual ) t2 ON (t1.id = t2.id)
WHEN MATCHED THEN
update set t1.name = t2.name, t1.version = t1.version + 1
WHEN NOT MATCHED THEN
insert (id, name, version, CreateTime)
values (t2.id, t2.name, t2.version, t2.CreateTime)", sql);
Assert.Equal(4, iou.ExecuteAffrows());
var lst = fsql.Select<tbiou04>().Where(a => new[] { 1, 2, 3, 4 }.Contains(a.id)).ToList();
Assert.Equal(4, lst.Where(a => a.name == "00" + a.id).Count());
}
class tbiou04
{
public int id { get; set; }
public string name { get; set; }
[Column(IsVersion = true)]
public int version { get; set; }
[Column(CanUpdate = false, ServerTime = DateTimeKind.Local)]
public DateTime CreateTime { get; set; }
}
}
}