由于您使用的是Linq to Sql,因此这里是测试使用NUnit和Moq提到的方案的样本。我不知道您的DataContext的确切详细信息以及其中可用的内容。根据您的需求进行编辑。
您将需要使用自定义类包装DataContext,而不能使用Moq模拟DataContext。您也不能模拟SqlException,因为它是密封的。您将需要用自己的Exception类包装它。完成这两件事并不困难。
让我们从创建测试开始:
[Test]
public void FindBy_When_something_goes_wrong_Should_handle_the_CustomSqlException()
{
var mockDataContextWrapper = new Mock<IDataContextWrapper>();
mockDataContextWrapper.Setup(x => x.Table<User>()).Throws<CustomSqlException>();
IUserResository userRespoistory = new UserRepository(mockDataContextWrapper.Object);
User user = userRepository.FindBy(1);
}
让我们实现测试,首先让我们使用存储库模式将Linq封装为Sql调用:
public interface IUserRepository
{
User FindBy(int id);
}
public class UserRepository : IUserRepository
{
public IDataContextWrapper DataContextWrapper { get; protected set; }
public UserRepository(IDataContextWrapper dataContextWrapper)
{
DataContextWrapper = dataContextWrapper;
}
public User FindBy(int id)
{
return DataContextWrapper.Table<User>().SingleOrDefault(u => u.UserID == id);
}
}
接下来像这样创建IDataContextWrapper,您可以查看有关此主题的博客文章,我的观点有所不同:
public interface IDataContextWrapper : IDisposable
{
Table<T> Table<T>() where T : class;
}
接下来创建CustomSqlException类:
public class CustomSqlException : Exception
{
public CustomSqlException()
{
}
public CustomSqlException(string message, SqlException innerException) : base(message, innerException)
{
}
}
这是IDataContextWrapper的示例实现:
public class DataContextWrapper<T> : IDataContextWrapper where T : DataContext, new()
{
private readonly T _db;
public DataContextWrapper()
{
var t = typeof(T);
_db = (T)Activator.CreateInstance(t);
}
public DataContextWrapper(string connectionString)
{
var t = typeof(T);
_db = (T)Activator.CreateInstance(t, connectionString);
}
public Table<TableName> Table<TableName>() where TableName : class
{
try
{
return (Table<TableName>) _db.GetTable(typeof (TableName));
}
catch (SqlException exception)
{
throw new CustomSqlException("Ooops...", exception);
}
}
}