Index: Application/Ringtoets/test/Application.Ringtoets.Storage.TestUtil/RingtoetsEntitiesHelper.cs =================================================================== diff -u -r64d5609bb2912cd52dc74deffdd189222e240599 -r845d2b0230eca6b2e9e94a2aec3fa36bd3c916b6 --- Application/Ringtoets/test/Application.Ringtoets.Storage.TestUtil/RingtoetsEntitiesHelper.cs (.../RingtoetsEntitiesHelper.cs) (revision 64d5609bb2912cd52dc74deffdd189222e240599) +++ Application/Ringtoets/test/Application.Ringtoets.Storage.TestUtil/RingtoetsEntitiesHelper.cs (.../RingtoetsEntitiesHelper.cs) (revision 845d2b0230eca6b2e9e94a2aec3fa36bd3c916b6) @@ -20,6 +20,8 @@ // All rights reserved. using System.Collections.ObjectModel; +using System.Data.Entity; + using Application.Ringtoets.Storage.DbContext; using Rhino.Mocks; @@ -29,30 +31,40 @@ { public static IRingtoetsEntities Create(MockRepository mockRepository) { + DbSet projectsSet = CreateEmptyTestDbSet(); + DbSet hydraylicLocationsSet = CreateEmptyTestDbSet(); + DbSet failureMechanismsSet = CreateEmptyTestDbSet(); + DbSet failureMechanismSectionsSet = CreateEmptyTestDbSet(); + DbSet failureMechanismSectionPointsSet = CreateEmptyTestDbSet(); + DbSet assessmentSectionsSet = CreateEmptyTestDbSet(); + DbSet referenceLinesSet = CreateEmptyTestDbSet(); + DbSet stochasticSoilModelsSet = CreateEmptyTestDbSet(); + DbSet stochasticSoilProfilesSet = CreateEmptyTestDbSet(); + DbSet soilProfilesSet = CreateEmptyTestDbSet(); + DbSet soilLayersSet = CreateEmptyTestDbSet(); + DbSet surfaceLinesSet = CreateEmptyTestDbSet(); + DbSet surfaceLinePointsSet = CreateEmptyTestDbSet(); + var ringtoetsEntities = mockRepository.Stub(); - var pSet = new TestDbSet(new ObservableCollection()); - var hlSet = new TestDbSet(new ObservableCollection()); - var fmSet = new TestDbSet(new ObservableCollection()); - var fmsSet = new TestDbSet(new ObservableCollection()); - var fmspSet = new TestDbSet(new ObservableCollection()); - var dasSet = new TestDbSet(new ObservableCollection()); - var rlpSet = new TestDbSet(new ObservableCollection()); - var ssmSet = new TestDbSet(new ObservableCollection()); - var sspSet = new TestDbSet(new ObservableCollection()); - var spSet = new TestDbSet(new ObservableCollection()); - var slSet = new TestDbSet(new ObservableCollection()); - ringtoetsEntities.Stub(r => r.ProjectEntities).Return(pSet); - ringtoetsEntities.Stub(r => r.HydraulicLocationEntities).Return(hlSet); - ringtoetsEntities.Stub(r => r.FailureMechanismEntities).Return(fmSet); - ringtoetsEntities.Stub(r => r.FailureMechanismSectionEntities).Return(fmsSet); - ringtoetsEntities.Stub(r => r.FailureMechanismSectionPointEntities).Return(fmspSet); - ringtoetsEntities.Stub(r => r.AssessmentSectionEntities).Return(dasSet); - ringtoetsEntities.Stub(r => r.ReferenceLinePointEntities).Return(rlpSet); - ringtoetsEntities.Stub(r => r.StochasticSoilModelEntities).Return(ssmSet); - ringtoetsEntities.Stub(r => r.StochasticSoilProfileEntities).Return(sspSet); - ringtoetsEntities.Stub(r => r.SoilProfileEntities).Return(spSet); - ringtoetsEntities.Stub(r => r.SoilLayerEntities).Return(slSet); + ringtoetsEntities.Stub(r => r.ProjectEntities).Return(projectsSet); + ringtoetsEntities.Stub(r => r.HydraulicLocationEntities).Return(hydraylicLocationsSet); + ringtoetsEntities.Stub(r => r.FailureMechanismEntities).Return(failureMechanismsSet); + ringtoetsEntities.Stub(r => r.FailureMechanismSectionEntities).Return(failureMechanismSectionsSet); + ringtoetsEntities.Stub(r => r.FailureMechanismSectionPointEntities).Return(failureMechanismSectionPointsSet); + ringtoetsEntities.Stub(r => r.AssessmentSectionEntities).Return(assessmentSectionsSet); + ringtoetsEntities.Stub(r => r.ReferenceLinePointEntities).Return(referenceLinesSet); + ringtoetsEntities.Stub(r => r.StochasticSoilModelEntities).Return(stochasticSoilModelsSet); + ringtoetsEntities.Stub(r => r.StochasticSoilProfileEntities).Return(stochasticSoilProfilesSet); + ringtoetsEntities.Stub(r => r.SoilProfileEntities).Return(soilProfilesSet); + ringtoetsEntities.Stub(r => r.SoilLayerEntities).Return(soilLayersSet); + ringtoetsEntities.Stub(r => r.SurfaceLineEntities).Return(surfaceLinesSet); + ringtoetsEntities.Stub(r => r.SurfaceLinePointEntities).Return(surfaceLinePointsSet); return ringtoetsEntities; } + + private static DbSet CreateEmptyTestDbSet() where T : class + { + return new TestDbSet(new ObservableCollection()); + } } } \ No newline at end of file