Skip to content

Commit 0481226

Browse files
sjh37sjh37
committed
Update tests to reflect new DbSet methods.
1 parent 7a732f0 commit 0481226

File tree

5 files changed

+101
-2
lines changed

5 files changed

+101
-2
lines changed

‎Generator.Tests.Unit/TestComparison/Northwind_PostgreSQL_EfCore3_FkLegacy.cs‎

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -735,6 +735,11 @@ public override EntityEntry<TEntity> Add(TEntity entity)
735735
return null;
736736
}
737737

738+
public override ValueTask<EntityEntry<TEntity>> AddAsync(TEntity entity, CancellationToken cancellationToken = default)
739+
{
740+
return new ValueTask<EntityEntry<TEntity>>(Task<EntityEntry<TEntity>>.Factory.StartNew(() => Add(entity)));
741+
}
742+
738743
public override void AddRange(params TEntity[] entities)
739744
{
740745
if (entities == null) throw new ArgumentNullException("entities");
@@ -753,12 +758,29 @@ public override Task AddRangeAsync(params TEntity[] entities)
753758
return Task.Factory.StartNew(() => AddRange(entities));
754759
}
755760

761+
public override Task AddRangeAsync(IEnumerable<TEntity> entities, CancellationToken cancellationToken = default)
762+
{
763+
if (entities == null) throw new ArgumentNullException("entities");
764+
return Task.Factory.StartNew(() => AddRange(entities));
765+
}
766+
767+
public override void AttachRange(IEnumerable<TEntity> entities)
768+
{
769+
AddRange(entities.ToArray());
770+
}
771+
756772
public override void AttachRange(params TEntity[] entities)
757773
{
758774
if (entities == null) throw new ArgumentNullException("entities");
759775
AddRange(entities);
760776
}
761777

778+
public override EntityEntry<TEntity> Remove(TEntity entity)
779+
{
780+
_data.Remove(entity);
781+
return null;
782+
}
783+
762784
public override void RemoveRange(params TEntity[] entities)
763785
{
764786
if (entities == null) throw new ArgumentNullException("entities");
@@ -771,6 +793,17 @@ public override void RemoveRange(IEnumerable<TEntity> entities)
771793
RemoveRange(entities.ToArray());
772794
}
773795

796+
public override EntityEntry<TEntity> Update(TEntity entity)
797+
{
798+
_data.Remove(entity);
799+
_data.Add(entity);
800+
return null;
801+
}
802+
803+
public override void UpdateRange(IEnumerable<TEntity> entities) {
804+
UpdateRange(entities.ToArray());
805+
}
806+
774807
public override void UpdateRange(params TEntity[] entities)
775808
{
776809
if (entities == null) throw new ArgumentNullException("entities");

‎Generator.Tests.Unit/TestComparison/Northwind_SqlCe_EfCore2_FkLegacy.cs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -455,7 +455,7 @@ public override void AddRange(IEnumerable<TEntity> entities)
455455
AddRange(entities.ToArray());
456456
}
457457

458-
public override Task AddRangeAsync(params TEntity[] entities)
458+
public override Task AddRangeAsync(IEnumerable<TEntity> entities, CancellationToken cancellationToken = default)
459459
{
460460
if (entities == null) throw new ArgumentNullException("entities");
461461
return Task.Factory.StartNew(() => AddRange(entities));

‎Generator.Tests.Unit/TestComparison/Northwind_SqlCe_EfCore3_FkLegacy.cs‎

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -448,6 +448,11 @@ public override EntityEntry<TEntity> Add(TEntity entity)
448448
return null;
449449
}
450450

451+
public override ValueTask<EntityEntry<TEntity>> AddAsync(TEntity entity, CancellationToken cancellationToken = default)
452+
{
453+
return new ValueTask<EntityEntry<TEntity>>(Task<EntityEntry<TEntity>>.Factory.StartNew(() => Add(entity)));
454+
}
455+
451456
public override void AddRange(params TEntity[] entities)
452457
{
453458
if (entities == null) throw new ArgumentNullException("entities");
@@ -466,12 +471,29 @@ public override Task AddRangeAsync(params TEntity[] entities)
466471
return Task.Factory.StartNew(() => AddRange(entities));
467472
}
468473

474+
public override Task AddRangeAsync(IEnumerable<TEntity> entities, CancellationToken cancellationToken = default)
475+
{
476+
if (entities == null) throw new ArgumentNullException("entities");
477+
return Task.Factory.StartNew(() => AddRange(entities));
478+
}
479+
480+
public override void AttachRange(IEnumerable<TEntity> entities)
481+
{
482+
AddRange(entities.ToArray());
483+
}
484+
469485
public override void AttachRange(params TEntity[] entities)
470486
{
471487
if (entities == null) throw new ArgumentNullException("entities");
472488
AddRange(entities);
473489
}
474490

491+
public override EntityEntry<TEntity> Remove(TEntity entity)
492+
{
493+
_data.Remove(entity);
494+
return null;
495+
}
496+
475497
public override void RemoveRange(params TEntity[] entities)
476498
{
477499
if (entities == null) throw new ArgumentNullException("entities");
@@ -484,6 +506,17 @@ public override void RemoveRange(IEnumerable<TEntity> entities)
484506
RemoveRange(entities.ToArray());
485507
}
486508

509+
public override EntityEntry<TEntity> Update(TEntity entity)
510+
{
511+
_data.Remove(entity);
512+
_data.Add(entity);
513+
return null;
514+
}
515+
516+
public override void UpdateRange(IEnumerable<TEntity> entities) {
517+
UpdateRange(entities.ToArray());
518+
}
519+
487520
public override void UpdateRange(params TEntity[] entities)
488521
{
489522
if (entities == null) throw new ArgumentNullException("entities");

‎Generator.Tests.Unit/TestComparison/Northwind_SqlServer_EfCore2_FkLegacy.cs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1017,7 +1017,7 @@ public override void AddRange(IEnumerable<TEntity> entities)
10171017
AddRange(entities.ToArray());
10181018
}
10191019

1020-
public override Task AddRangeAsync(params TEntity[] entities)
1020+
public override Task AddRangeAsync(IEnumerable<TEntity> entities, CancellationToken cancellationToken = default)
10211021
{
10221022
if (entities == null) throw new ArgumentNullException("entities");
10231023
return Task.Factory.StartNew(() => AddRange(entities));

‎Generator.Tests.Unit/TestComparison/Northwind_SqlServer_EfCore3_FkLegacy.cs‎

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1015,6 +1015,11 @@ public override EntityEntry<TEntity> Add(TEntity entity)
10151015
return null;
10161016
}
10171017

1018+
public override ValueTask<EntityEntry<TEntity>> AddAsync(TEntity entity, CancellationToken cancellationToken = default)
1019+
{
1020+
return new ValueTask<EntityEntry<TEntity>>(Task<EntityEntry<TEntity>>.Factory.StartNew(() => Add(entity)));
1021+
}
1022+
10181023
public override void AddRange(params TEntity[] entities)
10191024
{
10201025
if (entities == null) throw new ArgumentNullException("entities");
@@ -1033,12 +1038,29 @@ public override Task AddRangeAsync(params TEntity[] entities)
10331038
return Task.Factory.StartNew(() => AddRange(entities));
10341039
}
10351040

1041+
public override Task AddRangeAsync(IEnumerable<TEntity> entities, CancellationToken cancellationToken = default)
1042+
{
1043+
if (entities == null) throw new ArgumentNullException("entities");
1044+
return Task.Factory.StartNew(() => AddRange(entities));
1045+
}
1046+
1047+
public override void AttachRange(IEnumerable<TEntity> entities)
1048+
{
1049+
AddRange(entities.ToArray());
1050+
}
1051+
10361052
public override void AttachRange(params TEntity[] entities)
10371053
{
10381054
if (entities == null) throw new ArgumentNullException("entities");
10391055
AddRange(entities);
10401056
}
10411057

1058+
public override EntityEntry<TEntity> Remove(TEntity entity)
1059+
{
1060+
_data.Remove(entity);
1061+
return null;
1062+
}
1063+
10421064
public override void RemoveRange(params TEntity[] entities)
10431065
{
10441066
if (entities == null) throw new ArgumentNullException("entities");
@@ -1051,6 +1073,17 @@ public override void RemoveRange(IEnumerable<TEntity> entities)
10511073
RemoveRange(entities.ToArray());
10521074
}
10531075

1076+
public override EntityEntry<TEntity> Update(TEntity entity)
1077+
{
1078+
_data.Remove(entity);
1079+
_data.Add(entity);
1080+
return null;
1081+
}
1082+
1083+
public override void UpdateRange(IEnumerable<TEntity> entities) {
1084+
UpdateRange(entities.ToArray());
1085+
}
1086+
10541087
public override void UpdateRange(params TEntity[] entities)
10551088
{
10561089
if (entities == null) throw new ArgumentNullException("entities");

0 commit comments

Comments
 (0)