如何模拟IDataReader来测试将SqlDataReader转换为System.DataView的方法

时间:2016-02-04 12:19:52

标签: c# unit-testing mocking moq idatareader

我是Moq的新手,我正在努力编写单元测试以测试将SqlDataAdapter转换为System.DataView的方法。这是我的方法:

private DataView ResolveDataReader(IDataReader dataReader)
{
    DataTable table = new DataTable();

    for (int count = 0; count < dataReader.FieldCount; count++)
    {
        DataColumn col = new DataColumn(dataReader.GetName(count), 
                                        dataReader.GetFieldType(count));
        table.Columns.Add(col);
    }

    while (dataReader.Read())
    {
        DataRow dr = table.NewRow();
        for (int i = 0; i < dataReader.FieldCount; i++)
        {
            dr[i] = dataReader.GetValue(dataReader.GetOrdinal(dataReader.GetName(i)));
        }
        table.Rows.Add(dr);
    }

    return table.DefaultView;
}

我试图创建以创建类似的东西:

var dataReaderMock = new Mock<IDataReader>();
var records = new Mock<IDataRecord>();
dataReaderMock.Setup(x => x.FieldCount).Returns(2);
dataReaderMock.Setup(x => x.Read()).Returns(() => records);

我想传递一些数据并验证它已被转换。

感谢。

2 个答案:

答案 0 :(得分:8)

你的嘲讽是正确的,但dataReaderMock.Setup(x => x.Read()).Returns(() => records);是你出错的地方,因为.Read返回一个布尔,而不是记录本身,这些记录是IDataReader的你的方法。

安排嘲笑

var dataReader = new Mock<IDataReader>();
dataReader.Setup(m => m.FieldCount).Returns(2); // the number of columns in the faked data

dataReader.Setup(m => m.GetName(0)).Returns("First"); // the first column name
dataReader.Setup(m => m.GetName(1)).Returns("Second"); // the second column name

dataReader.Setup(m => m.GetFieldType(0)).Returns(typeof(string)); // the data type of the first column
dataReader.Setup(m => m.GetFieldType(1)).Returns(typeof(string)); // the data type of the second column

您可以排列要品尝的列以在您的系统中模拟更多真实数据,类型等..只需确保第一次计数,GetName的数量和数量GetFieldType是同步的。

要安排.Read(),我们可以使用SetupSequence:

dataReader.SetupSequence(m => m.Read())
    .Returns(true) // Read the first row
    .Returns(true) // Read the second row
    .Returns(false); // Done reading

要在测试中使用它,您可以将其提取到方法中:

private const string Column1 = "First";
private const string Column2 = "Second";
private const string ExpectedValue1 = "Value1";
private const string ExpectedValue2 = "Value1";

private static Mock<IDataReader> CreateDataReader()
{
    var dataReader = new Mock<IDataReader>();

    dataReader.Setup(m => m.FieldCount).Returns(2);
    dataReader.Setup(m => m.GetName(0)).Returns(Column1);
    dataReader.Setup(m => m.GetName(1)).Returns(Column2);

    dataReader.Setup(m => m.GetFieldType(0)).Returns(typeof(string));
    dataReader.Setup(m => m.GetFieldType(1)).Returns(typeof(string));

    dataReader.Setup(m => m.GetOrdinal("First")).Returns(0);
    dataReader.Setup(m => m.GetValue(0)).Returns(ExpectedValue1);
    dataReader.Setup(m => m.GetValue(1)).Returns(ExpectedValue2);

    dataReader.SetupSequence(m => m.Read())
        .Returns(true)
        .Returns(true)
        .Returns(false);
    return dataReader;
}

(或者,您可以在Setup上安排此操作,如果这对您的测试类更有意义 - 在这种情况下,dataReader模拟将是一个字段,而不是返回值)

示例测试。然后它可以像:

一样使用
[Test]
public void ResovleDataReader_RowCount()
{
    var dataReader = CreateDateReader();
    var view = ResolveDataReader(dataReader.Object);
    Assert.AreEqual(2, view.Count);
}

[Test]
public void ResolveDataReader_NamesColumn1()
{
    var dataReader = CreateDataReader();
    var view = ResolveDataReader(dataReader.Object);
    Assert.AreEqual(Column1, view.Table.Columns[0].ColumnName);
}

[Test]
public void ResolveDataReader_PopulatesColumn1()
{
    var dataReader = CreateDataReader();
    var view = ResolveDataReader(dataReader.Object);
    Assert.AreEqual(ExpectedValue1, view.Table.Rows[0][0]);
}

// Etc..

(我使用过NUnit,但它与测试方法的不同属性和不同的断言语法类似,适用于不同的测试框架)

顺便说一句,我通过将ResolveDataReader更改为internal并设置InternalsVisibleTo来实现上述目标,但我认为您已经拥有了这个私有方法的网关就你试图测试而言。

答案 1 :(得分:0)

我的班级设置public static class DataReaderMock { public static void SetupDataReader(this Mock<IDataReader> mock, ICollection<string> columns, object[,] values) { if (columns.Count != values.GetLength(1)) { throw new ArgumentException($"The number of named columns must be identical to the number of columns in the 2d values array: {columns.Count} compared to {values.GetLength(1)}"); } mock.Setup(reader => reader.FieldCount).Returns(columns.Count); var setupSequence = mock.SetupSequence(reader => reader.Read()); var callbacks = new List<Action<object[]>> { vals => vals.Populate(columns.Cast<object>().ToList()) }; for (var row = 0; row < values.GetLength(0); row++) { var currentRow = row; // for closure callbacks.Add(vals => vals.Populate(values, currentRow)); setupSequence.Returns(true); } setupSequence.Returns(false); mock.Setup(reader => reader.GetValues(It.IsAny<object[]>())).CallbackSequence(callbacks.ToArray()); } private static void Populate<T>(this IList<T> target, IList<T> source) { for (var i = 0; i < target.Count; i++) { target[i] = source[i]; } } private static void Populate<T>(this IList<T> target, T[,] sourceTable, int row) { for (var i = 0; i < sourceTable.GetLength(1); i++) { target[i] = sourceTable[row, i]; } } private static void CallbackSequence<T, TResult, TArg>(this ISetup<T, TResult> setup, params Action<TArg>[] callbacks) where T : class { var queue = new ConcurrentQueue<Action<TArg>>(callbacks); setup.Callback((TArg arg) => { Action<TArg> callback; if (!queue.TryDequeue(out callback)) { Assert.Fail("More callbacks were invoked than defined in sequence"); } callback(arg); }); } } mock:

const int ItemsCount = 1000;
var dataReaderMock = new Mock<IDataReader>();
var values = new object[ItemsCount, 2];
for (var i = 0; i < ItemsCount; i++)
{
    values[i, 0] = i + 1;
    values[i, 1] = (i + 1).ToString();
}
dataReaderMock.SetupDataReader(new List<string> {"Col1", "Col2"}, values);

用法:

    @Override
protected void onCreate(Bundle savedInstanceState) {
    super.onCreate(savedInstanceState);
    setContentView(R.layout.activity_one);
    String title = getIntent().getStringExtra("title");
    String message = getIntent().getStringExtra("notification message");
    setTitle(title);
    TextView desc = (TextView) findViewById(R.id.desc1);
    desc.setText(message);
}