2020using Xtensive . Orm . Providers ;
2121using Xtensive . Orm . Rse . Providers ;
2222using Xtensive . Orm . Tests . Storage . Prefetch . Model ;
23+ using GraphContainerDictionary = System . Collections . Generic . Dictionary < ( Xtensive . Orm . Key key , Xtensive . Orm . Model . TypeInfo type ) , Xtensive . Orm . Internals . Prefetch . GraphContainer > ;
2324
2425namespace Xtensive . Orm . Tests . Storage . Prefetch
2526{
@@ -103,9 +104,9 @@ public void ReferencedEntityHasBeenFullyLoadedBeforeTaskActivationTest()
103104
104105 prefetchManager . InvokePrefetch ( order0Key , null , new PrefetchFieldDescriptor ( EmployeeField , true , true ) ) ;
105106 prefetchManager . InvokePrefetch ( order1Key , null , new PrefetchFieldDescriptor ( EmployeeField , true , true ) ) ;
106- var graphContainers = ( HashSet < GraphContainer > ) GraphContainersField . GetValue ( prefetchManager ) ;
107+ var graphContainers = ( GraphContainerDictionary ) GraphContainersField . GetValue ( prefetchManager ) ;
107108 Assert . AreEqual ( 2 , graphContainers . Count ) ;
108- Func < Key , ReferencedEntityContainer > taskSelector = containerKey => graphContainers
109+ Func < Key , ReferencedEntityContainer > taskSelector = containerKey => graphContainers . Values
109110 . Where ( container => container . Key == containerKey )
110111 . SelectMany ( container => container . ReferencedEntityContainers ) . Single ( ) ;
111112 var entityContainer0 = taskSelector . Invoke ( order0Key ) ;
@@ -180,7 +181,7 @@ public void PrefetchEmptyEntitySetTest()
180181 var prefetchManager = ( PrefetchManager ) PrefetchProcessorField . GetValue ( session . Handler ) ;
181182 session . Handler . FetchEntityState ( orderKey ) ;
182183 prefetchManager . InvokePrefetch ( orderKey , null , new PrefetchFieldDescriptor ( DetailsField , null ) ) ;
183- var graphContainers = ( HashSet < GraphContainer > ) GraphContainersField . GetValue ( prefetchManager ) ;
184+ var graphContainers = ( GraphContainerDictionary ) GraphContainersField . GetValue ( prefetchManager ) ;
184185 Assert . AreEqual ( 1 , graphContainers . Count ) ;
185186 prefetchManager . ExecuteTasks ( true ) ;
186187 EntitySetState actualState ;
@@ -202,8 +203,8 @@ public void PrefetchReferencedEntityWhenTypeSpecifiedForOwnerIsInvalidTest()
202203 prefetchManager . InvokePrefetch ( keyWithoutType , Domain . Model . Types [ typeof ( PersonalProduct ) ] ,
203204 new PrefetchFieldDescriptor ( Domain . Model . Types [ typeof ( PersonalProduct ) ] . Fields [ "Employee" ] ,
204205 true , true ) ) ;
205- var graphContainers = ( HashSet < GraphContainer > ) GraphContainersField . GetValue ( prefetchManager ) ;
206- var referencedEntityContainer = graphContainers
206+ var graphContainers = ( GraphContainerDictionary ) GraphContainersField . GetValue ( prefetchManager ) ;
207+ var referencedEntityContainer = graphContainers . Values
207208 . Where ( container => container . ReferencedEntityContainers != null ) . Single ( )
208209 . ReferencedEntityContainers . Single ( ) ;
209210 prefetchManager . ExecuteTasks ( true ) ;
@@ -218,10 +219,10 @@ public void DeletingOfTasksAtTransactionCommitOrRollbackTest()
218219
219220 using ( var session = Domain . OpenSession ( ) ) {
220221 var prefetchManager = ( PrefetchManager ) PrefetchProcessorField . GetValue ( session . Handler ) ;
221- HashSet < GraphContainer > graphContainers ;
222+ GraphContainerDictionary graphContainers ;
222223 using ( var tx = session . OpenTransaction ( ) ) {
223224 prefetchManager . InvokePrefetch ( orderKey , null , new PrefetchFieldDescriptor ( CustomerField ) ) ;
224- graphContainers = ( HashSet < GraphContainer > ) GraphContainersField . GetValue ( prefetchManager ) ;
225+ graphContainers = ( GraphContainerDictionary ) GraphContainersField . GetValue ( prefetchManager ) ;
225226 Assert . AreEqual ( 1 , graphContainers . Count ) ;
226227 tx . Complete ( ) ;
227228 }
@@ -251,7 +252,7 @@ public void TasksAreExecutedAutomaticallyWhenCountLimitIsReachedTest()
251252 using ( var session = Domain . OpenSession ( ) )
252253 using ( var tx = session . OpenTransaction ( ) ) {
253254 var prefetchManager = ( PrefetchManager ) PrefetchProcessorField . GetValue ( session . Handler ) ;
254- var graphContainers = ( HashSet < GraphContainer > ) GraphContainersField . GetValue ( prefetchManager ) ;
255+ var graphContainers = ( GraphContainerDictionary ) GraphContainersField . GetValue ( prefetchManager ) ;
255256 var idField = BookType . Fields [ "Id" ] ;
256257 for ( var i = 1 ; i < keys . Count ; i ++ ) {
257258 prefetchManager . InvokePrefetch ( keys [ i - 1 ] , null , new PrefetchFieldDescriptor ( idField ) ) ;
0 commit comments