From 8812e360d8034615bb656d40fc7495577caaf1cc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=BD=95=E9=95=87=E6=B1=90?= <8907060@qq.com>
Date: Sat, 14 Apr 2018 16:55:16 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8DCrud=E6=9C=8D=E5=8A=A1?=
 =?UTF-8?q?=E5=8D=95=E5=85=83=E6=B5=8B=E8=AF=95BUG?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Applications/CrudServiceTest.BatchSave.cs       | 13 ++++++++++---
 .../Util.Tests/Applications/CrudServiceTest.Save.cs |  2 ++
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/test/Util.Tests/Applications/CrudServiceTest.BatchSave.cs b/test/Util.Tests/Applications/CrudServiceTest.BatchSave.cs
index 7f5aca20..369581b5 100644
--- a/test/Util.Tests/Applications/CrudServiceTest.BatchSave.cs
+++ b/test/Util.Tests/Applications/CrudServiceTest.BatchSave.cs
@@ -188,6 +188,9 @@ namespace Util.Tests.Applications {
             var deleteList = new List<DtoSample> {
                 new DtoSample{Id = id3.ToString()}
             };
+            _repository.Find( id3 ).Returns( t => new EntitySample( id3 ) );
+            _repository.Find( id4 ).Returns( t => new EntitySample( id4 ) );
+            _repository.Find( id5 ).Returns( t => new EntitySample( id5 ) );
             var result = _service.Save( addList, updateList, deleteList );
             _repository.Received( 2 ).Add( Arg.Any<EntitySample>() );
             _repository.Received( 2 ).Update( Arg.Any<EntitySample>() );
@@ -229,7 +232,8 @@ namespace Util.Tests.Applications {
         /// </summary>
         [Fact]
         public async Task TestSaveAsync_Update_1() {
-            _repository.Find( _id ).Returns( t => new EntitySample( _id ) );
+            var entity = new EntitySample( _id );
+            _repository.FindAsync( _id ).Returns( t => entity );
             var input = new List<DtoSample> {
                 new DtoSample{Id = _id.ToString(),Name = "b"}
             };
@@ -243,8 +247,8 @@ namespace Util.Tests.Applications {
         /// </summary>
         [Fact]
         public async Task TestSaveAsync_Update_2() {
-            _repository.Find( _id ).Returns( t => new EntitySample( _id ) );
-            _repository.Find( _id2 ).Returns( t => new EntitySample( _id2 ) );
+            _repository.FindAsync( _id ).Returns( t => new EntitySample( _id ) );
+            _repository.FindAsync( _id2 ).Returns( t => new EntitySample( _id2 ) );
             var input = new List<DtoSample> {
                 new DtoSample{Id = _id.ToString(),Name = "b"},
                 new DtoSample{Id = _id2.ToString(),Name = "b"}
@@ -290,6 +294,9 @@ namespace Util.Tests.Applications {
             var deleteList = new List<DtoSample> {
                 new DtoSample{Id = id3.ToString()}
             };
+            _repository.FindAsync( id3 ).Returns( t => new EntitySample( id3 ) );
+            _repository.FindAsync( id4 ).Returns( t => new EntitySample( id4 ) );
+            _repository.FindAsync( id5 ).Returns( t => new EntitySample( id5 ) );
             var result = await _service.SaveAsync( addList, updateList, deleteList );
             await _repository.Received( 2 ).AddAsync( Arg.Any<EntitySample>() );
             await _repository.Received( 2 ).UpdateAsync( Arg.Any<EntitySample>() );
diff --git a/test/Util.Tests/Applications/CrudServiceTest.Save.cs b/test/Util.Tests/Applications/CrudServiceTest.Save.cs
index a0504874..0124339c 100644
--- a/test/Util.Tests/Applications/CrudServiceTest.Save.cs
+++ b/test/Util.Tests/Applications/CrudServiceTest.Save.cs
@@ -108,6 +108,7 @@ namespace Util.Tests.Applications {
         /// </summary>
         [Fact]
         public void TestSave_Update() {
+            _repository.Find( _id ).Returns( t => new EntitySample( _id ) );
             _service.Save( new DtoSample { Id =_id.ToString(), Name = "b" } );
             _repository.DidNotReceive().Add( Arg.Any<EntitySample>() );
             _repository.Received().Update( Arg.Is<EntitySample>( t => t.Name == "b" ) );
@@ -127,6 +128,7 @@ namespace Util.Tests.Applications {
         /// </summary>
         [Fact]
         public async Task TestSaveAsync_Update() {
+            _repository.FindAsync( _id ).Returns( t => new EntitySample( _id ) );
             await _service.SaveAsync( new DtoSample { Id = _id.ToString(), Name = "b" } );
             await _repository.DidNotReceive().AddAsync( Arg.Any<EntitySample>() );
             await _repository.Received().UpdateAsync( Arg.Is<EntitySample>( t => t.Name == "b" ) );