![](/img/trans.png)
[英]System.InvalidCastException in Unit Test with Oracle objects using DbProviderFactory
[英]Best way to unit test DbProviderFactory
我正在嘗試為我們擁有的舊代碼創建單元測試(NUnit)。
我們使用DbProviderFactory來檢查表是否存在於數據庫中,然后再對該表執行任何操作。
我知道這看起來更像是集成測試,但是無論您如何稱呼它,我都需要對其進行測試。 我寧願使用自包含的東西,也不依賴數據庫。
我試圖使用excel作為數據源,但是要在excel中選擇表,它應該在表名后有$,這在我的情況下不起作用,因為我不想修改代碼以適應單元測試。
如何對以下代碼進行單元測試?
static bool TableDoesNotExist(string tableName, string connectionString, string providerName = "System.Data.OleDb")
{
try
{
DbProviderFactory providerFactory = DbProviderFactories.GetFactory(providerName);
using (DbConnection conn = providerFactory.CreateConnection())
{
conn.ConnectionString = connectionString;
conn.Open();
DbCommand cmd = providerFactory.CreateCommand();
cmd.Connection = conn;
string tblQuery = "";
if (providerName == "System.Data.Odbc")
tblQuery = string.Format("SELECT COUNT(*) FROM SYSTABLE WHERE TABLE_NAME = '{0}'", tableName);
else
tblQuery = string.Format("SELECT COUNT(*) FROM [INFORMATION_SCHEMA.TABLES$] WHERE TABLE_NAME = '{0}'", tableName);
cmd.CommandText = tblQuery;
Console.WriteLine(cmd.CommandText);
DbDataReader dr = cmd.ExecuteReader();
DataTable dt = new DataTable();
dt.Load(dr);
if (dt.Rows.Count == 1 && Convert.ToInt32(dt.Rows[0][0]) == 0)
{
return true;
}
}
return false;
}
catch (Exception ex)
{
Console.WriteLine(ex.ToString());
}
return false;
}
任何幫助,將不勝感激。
您可以嘗試一次使用存根。 我已經使用Stub編寫了一個簡單的測試用例。 它按預期工作。 這是代碼:
[TestMethod]
public void Test01()
{
using (ShimsContext.Create())
{
var dbConnectionOpened = false;
var fakeConnection = new StubDbConnection()
{
Open01 = () => { dbConnectionOpened = true; }
};
var fakeCommand = new StubDbCommand()
{
ExecuteDbDataReaderCommandBehavior = (com) => GetFakeReader()
};
var fakeDbProviderFactory = new StubDbProviderFactory()
{
CreateConnection01 = () => fakeConnection,
CreateCommand01 = () => fakeCommand
};
ShimDbProviderFactories.GetFactoryString = (arg1) => fakeDbProviderFactory;
var val = SqlConnectionFactory.TableDoesNotExist("testTable", "conn");
Assert.IsTrue(dbConnectionOpened);
Assert.IsTrue(val);
}
}
private DbDataReader GetFakeReader()
{
const int count = 0;
var dt = new DataTable("Test-Table");
dt.Columns.Add(new DataColumn("Count"));
dt.Rows.Add(count);
return dt.CreateDataReader();
}
您可以使用GetFakeReader()方法進行各種場景的單元測試。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.