fixed more tests.
This commit is contained in:
@@ -1,12 +1,13 @@
|
||||
using System.Linq;
|
||||
using NzbDrone.Core.Datastore;
|
||||
using ServiceStack.DataAnnotations;
|
||||
|
||||
namespace NzbDrone.Core.Configuration
|
||||
{
|
||||
public class Config : ModelBase
|
||||
{
|
||||
[Index(Unique = true)]
|
||||
public string Key { get; set; }
|
||||
|
||||
|
||||
public string Value { get; set; }
|
||||
}
|
||||
}
|
||||
@@ -57,7 +57,15 @@ namespace NzbDrone.Core.Datastore
|
||||
|
||||
public TModel Get(int id)
|
||||
{
|
||||
return _database.GetById<TModel>(id);
|
||||
try
|
||||
{
|
||||
return _database.GetById<TModel>(id);
|
||||
}
|
||||
catch (ArgumentNullException e)
|
||||
{
|
||||
throw new InvalidOperationException(e.Message);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public TModel Single(Expression<Func<TModel, bool>> predicate)
|
||||
@@ -87,6 +95,11 @@ namespace NzbDrone.Core.Datastore
|
||||
|
||||
public TModel Insert(TModel model)
|
||||
{
|
||||
if (model.Id != 0)
|
||||
{
|
||||
throw new InvalidOperationException("Can't insert model with existing ID");
|
||||
}
|
||||
|
||||
_database.Insert(model);
|
||||
model.Id = (int)_database.GetLastInsertId();
|
||||
return model;
|
||||
@@ -94,6 +107,11 @@ namespace NzbDrone.Core.Datastore
|
||||
|
||||
public TModel Update(TModel model)
|
||||
{
|
||||
if (model.Id == 0)
|
||||
{
|
||||
throw new InvalidOperationException("Can't update model with ID 0");
|
||||
}
|
||||
|
||||
_database.Update(model);
|
||||
return model;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user