fixed more tests.

This commit is contained in:
kay.one 2013-03-24 12:56:51 -07:00
parent 7603d8e1ba
commit cd6f0fc55c
4 changed files with 34 additions and 7 deletions

View File

@ -1,10 +1,12 @@
using System;
using System.Data;
using System.Linq;
using FizzWare.NBuilder;
using FluentAssertions;
using NUnit.Framework;
using NzbDrone.Core.Datastore;
using NzbDrone.Core.Test.Framework;
using ServiceStack.OrmLite;
namespace NzbDrone.Core.Test.Datastore
{
@ -29,6 +31,7 @@ namespace NzbDrone.Core.Test.Datastore
.With(c => c.Id = 0)
.Build();
Mocker.Resolve<IDbConnection>().CreateTable<BaiscType>();
}
[Test]

View File

@ -1,11 +1,13 @@
using System;
using System.Collections.Generic;
using System.Data;
using System.Linq;
using FizzWare.NBuilder;
using FluentAssertions;
using NUnit.Framework;
using NzbDrone.Core.Datastore;
using NzbDrone.Core.Test.Framework;
using ServiceStack.OrmLite;
namespace NzbDrone.Core.Test.Datastore
{
@ -21,6 +23,9 @@ namespace NzbDrone.Core.Test.Datastore
.CreateNew()
.With(s => s.Id = 0)
.Build();
Mocker.Resolve<IDbConnection>().CreateTable<BaiscType>();
}
[Test]

View File

@ -1,10 +1,11 @@
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; }

View File

@ -56,9 +56,17 @@ namespace NzbDrone.Core.Datastore
}
public TModel Get(int 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;
}