public void It_Returns_The_Reference_Objects() { ProviderInfo info = new ProviderInfo(); var entity1 = new EntityImpl("T1"); var entity2 = new EntityImpl("T2"); var entity3 = new EntityImpl("T3"); var rel1 = new ReferenceImpl { EntitySet = info.MappingSet.EntitySet }; var rel2 = new ReferenceImpl { EntitySet = info.MappingSet.EntitySet }; info.MappingSet.EntitySet.AddEntity(entity1); info.MappingSet.EntitySet.AddEntity(entity2); info.MappingSet.EntitySet.AddEntity(entity3); rel1.AddThisTo(entity1, entity2); rel2.AddThisTo(entity3, entity2); IEnumerable <IScriptBaseObject> objects = info.GetAllObjectsOfType(typeof(Reference)); Assert.That(objects, Has.Length(2)); Assert.That(objects.Contains(rel1)); Assert.That(objects.Contains(rel2)); }