Лучший способ модульного тестирования DbProviderFactory

Question

Я пытаюсь создать Unit Test (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;
    }

Любая помощь будет оценена.

Answer 1

Вы можете попробовать использовать заглушки один раз. Я написал простой тестовый пример с использованием заглушки. Работает как положено. Вот код:

    [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() для модульного тестирования различных сценариев.