From f4418241a6ee1d7689c585460a9cbbe6fb27724e Mon Sep 17 00:00:00 2001
From: LambertW <26795806@qq.com>
Date: Fri, 15 Mar 2019 21:37:42 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B5=8B=E8=AF=95=E5=AE=8C=E6=AF=95=EF=BC=8C?=
 =?UTF-8?q?=E6=B8=85=E7=A9=BASql=20Server=E6=B5=8B=E8=AF=95=E8=A1=A8?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../DataContext/SqlServer/SqlServerFixture.cs | 29 +++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/FreeSql.Tests/DataContext/SqlServer/SqlServerFixture.cs b/FreeSql.Tests/DataContext/SqlServer/SqlServerFixture.cs
index 1b3845c7..ea5dba1a 100644
--- a/FreeSql.Tests/DataContext/SqlServer/SqlServerFixture.cs
+++ b/FreeSql.Tests/DataContext/SqlServer/SqlServerFixture.cs
@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.Linq;
 using System.Text;
 
 namespace FreeSql.Tests.DataContext.SqlServer
@@ -9,8 +10,8 @@ namespace FreeSql.Tests.DataContext.SqlServer
 		public SqlServerFixture()
 		{
 			sqlServerLazy = new Lazy<IFreeSql>(() => new FreeSql.FreeSqlBuilder()
-			  .UseConnectionString(FreeSql.DataType.SqlServer, "Data Source=.;Integrated Security=True;Initial Catalog=freesqlTest;Pooling=true;Max Pool Size=10")
-			  //.UseConnectionString(FreeSql.DataType.SqlServer, "Data Source=(localdb)\\mssqllocaldb;Integrated Security=True;Initial Catalog=cms;Pooling=true;Max Pool Size=10")
+			  //.UseConnectionString(FreeSql.DataType.SqlServer, "Data Source=.;Integrated Security=True;Initial Catalog=freesqlTest;Pooling=true;Max Pool Size=10")
+			  .UseConnectionString(FreeSql.DataType.SqlServer, "Data Source=(localdb)\\mssqllocaldb;Integrated Security=True;Initial Catalog=cms;Pooling=true;Max Pool Size=10")
 			  .UseAutoSyncStructure(true)
 			  .UseLazyLoading(true)
 			  .Build());
@@ -21,6 +22,30 @@ namespace FreeSql.Tests.DataContext.SqlServer
 		public void Dispose()
 		{
 			// ... clean up test data from the database ...
+			var dataTables = SqlServer.DbFirst.GetTablesByDatabase(SqlServer.DbFirst.GetDatabases()[0]);
+			if (dataTables.Any(item => item.Name == "TopicAddField" && item.Schema == "dbo2"))
+			{
+				SqlServer.Ado.ExecuteNonQuery("TRUNCATE TABLE dbo2.TopicAddField ");
+				SqlServer.Ado.ExecuteNonQuery("DROP TABLE dbo2.TopicAddField");
+				SqlServer.Ado.ExecuteNonQuery("DROP SCHEMA dbo2");
+			}
+
+			var tempTables = new string[] { "cccccdddwww", "song", "tb_alltype", "tb_topic", "tb_topic22",
+				"tb_topic22211", "tb_topic111333", "TestTypeInfo", "TestTypeInfo333", "TestTypeParentInfo",
+				"TestTypeParentInfo23123", "xxdkdkdk1222", "xxx"};
+			foreach (var tempTable in tempTables)
+			{
+				DeleteTmpTable(dataTables, tempTable);
+			}
+		}
+
+		private void DeleteTmpTable(List<DatabaseModel.DbTableInfo> dbTables, string deleteTableName, string schemaName = "dbo")
+		{
+			if (dbTables.Any(item => item.Name.ToLower() == deleteTableName.ToLower() && item.Schema.ToLower() == schemaName.ToLower()))
+			{
+				SqlServer.Ado.ExecuteNonQuery($"TRUNCATE TABLE {schemaName.ToLower()}.{deleteTableName}");
+				SqlServer.Ado.ExecuteNonQuery($"DROP TABLE {schemaName.ToLower()}.{deleteTableName}");
+			}
 		}
 
 		private Lazy<IFreeSql> sqlServerLazy;