fixed broken tests.

This commit is contained in:
kay.one 2013-02-15 20:16:33 -08:00
parent eb90040dd6
commit 1adc2892db
1 changed files with 11 additions and 16 deletions

View File

@ -35,25 +35,20 @@ namespace NzbDrone.Core.Datastore
modelBase.Id = _indexProvider.Next(obj.GetType()); modelBase.Id = _indexProvider.Next(obj.GetType());
} }
foreach (var propertyInfo in GetPotentialProperties(obj.GetType())) var list = obj as IEnumerable;
{
var propValue = propertyInfo.GetValue(obj, null);
var list = propValue as IEnumerable;
if (list != null) if (list != null)
{ {
foreach (var item in list) foreach (var item in list)
{ {
EnsureIds(item, context); EnsureIds(item, context);
} }
return;
} }
else
foreach (var propertyInfo in GetPotentialProperties(obj.GetType()))
{ {
var propValue = propertyInfo.GetValue(obj, null);
EnsureIds(propValue, context); EnsureIds(propValue, context);
}
} }
} }