public void ColumnDropper_NoMatchingColumnAtRuntime()
{
var dropper = new ColumnDropper
{
ColumnNameToDrop = "DoesNotExist"
};
var cts = new GracefulCancellationTokenSource();
var toProcess = new DataTable();
toProcess.Columns.Add("Column1");
var ex = Assert.Throws <InvalidOperationException>(() => dropper.ProcessPipelineData(toProcess, new ThrowImmediatelyDataLoadEventListener(), cts.Token));
Assert.IsTrue(ex.Message.Contains("does not exist in the supplied data table"));
}
public void ColumnDropper_Successful()
{
var dropper = new ColumnDropper
{
ColumnNameToDrop = "ToDrop"
};
var cts = new GracefulCancellationTokenSource();
var toProcess = new DataTable();
toProcess.Columns.Add("ToDrop");
var processed = dropper.ProcessPipelineData(toProcess, new ThrowImmediatelyDataLoadEventListener(), cts.Token);
Assert.AreEqual(0, processed.Columns.Count);
Assert.AreEqual(false, processed.Columns.Contains("ToDrop"));
}
