From cb3c19b73167c025a62e32b0c05969ab49c0f6d4 Mon Sep 17 00:00:00 2001
From: Savorboard <yangxiaodong1214@126.com>
Date: Tue, 22 Aug 2017 22:34:25 +0800
Subject: [PATCH] refactor unit tests.

---
 .../MySqlStorageTest.cs                       | 34 +++------------
 .../SqlServerStorageTest.cs                   | 43 ++++---------------
 2 files changed, 13 insertions(+), 64 deletions(-)

diff --git a/test/DotNetCore.CAP.MySql.Test/MySqlStorageTest.cs b/test/DotNetCore.CAP.MySql.Test/MySqlStorageTest.cs
index 9286929..085246a 100644
--- a/test/DotNetCore.CAP.MySql.Test/MySqlStorageTest.cs
+++ b/test/DotNetCore.CAP.MySql.Test/MySqlStorageTest.cs
@@ -29,36 +29,12 @@ namespace DotNetCore.CAP.MySql.Test
             }
         }
 
-        [Fact]
-        public void DatabaseTable_Published_IsExists()
-        {
-            var tableName = "cap.published";
-            using (var connection = ConnectionUtil.CreateConnection(_masterDbConnectionString))
-            {
-                var sql = $"SELECT TABLE_NAME FROM `TABLES` WHERE TABLE_SCHEMA='{_dbName}' AND TABLE_NAME = '{tableName}'";
-                var result = connection.QueryFirstOrDefault<string>(sql);
-                Assert.NotNull(result);
-                Assert.Equal(tableName, result);
-            }
-        }
-
-        [Fact]
-        public void DatabaseTable_Queue_IsExists()
-        {
-            var tableName = "cap.queue";
-            using (var connection = ConnectionUtil.CreateConnection(_masterDbConnectionString))
-            {
-                var sql = $"SELECT TABLE_NAME FROM `TABLES` WHERE TABLE_SCHEMA='{_dbName}' AND TABLE_NAME = '{tableName}'";
-                var result = connection.QueryFirstOrDefault<string>(sql);
-                Assert.NotNull(result);
-                Assert.Equal(tableName, result);
-            }
-        }
-
-        [Fact]
-        public void DatabaseTable_Received_IsExists()
+        [Theory]
+        [InlineData("cap.published")]
+        [InlineData("cap.queue")]
+        [InlineData("cap.received")]
+        public void DatabaseTable_IsExists(string tableName)
         {
-            var tableName = "cap.received";
             using (var connection = ConnectionUtil.CreateConnection(_masterDbConnectionString))
             {
                 var sql = $"SELECT TABLE_NAME FROM `TABLES` WHERE TABLE_SCHEMA='{_dbName}' AND TABLE_NAME = '{tableName}'";
diff --git a/test/DotNetCore.CAP.SqlServer.Test/SqlServerStorageTest.cs b/test/DotNetCore.CAP.SqlServer.Test/SqlServerStorageTest.cs
index af1fdac..01e9aeb 100644
--- a/test/DotNetCore.CAP.SqlServer.Test/SqlServerStorageTest.cs
+++ b/test/DotNetCore.CAP.SqlServer.Test/SqlServerStorageTest.cs
@@ -23,49 +23,22 @@ SELECT 'False'";
             }
         }
 
-        [Fact]
-        public void DatabaseTable_Published_IsExists()
-        {
-            using (var connection = ConnectionUtil.CreateConnection())
-            {
-                var sql = @"
-IF OBJECT_ID(N'[Cap].[Published]',N'U') IS NOT NULL
-SELECT 'True'
-ELSE
-SELECT 'False'";
-                var result = connection.QueryFirst<bool>(sql);
-                Assert.True(result);
-            }
-        }
-
-        [Fact]
-        public void DatabaseTable_Queue_IsExists()
+        [Theory]
+        [InlineData("[Cap].[Published]")]
+        [InlineData("[Cap].[Queue]")]
+        [InlineData("[Cap].[Received]")]
+        public void DatabaseTable_IsExists(string tableName)
         {
             using (var connection = ConnectionUtil.CreateConnection())
             {
-                var sql = @"
-IF OBJECT_ID(N'[Cap].[Queue]',N'U') IS NOT NULL
-SELECT 'True'
-ELSE
-SELECT 'False'";
-                var result = connection.QueryFirst<bool>(sql);
-                Assert.True(result);
-            }
-        }
-
-        [Fact]
-        public void DatabaseTable_Received_IsExists()
-        {
-            using (var connection = ConnectionUtil.CreateConnection())
-            {
-                var sql = @"
-IF OBJECT_ID(N'[Cap].[Received]',N'U') IS NOT NULL
+                var sql = $@"
+IF OBJECT_ID(N'{tableName}',N'U') IS NOT NULL
 SELECT 'True'
 ELSE
 SELECT 'False'";
                 var result = connection.QueryFirst<bool>(sql);
                 Assert.True(result);
             }
-        }
+        } 
     }
 }