MSSQL - Create Comments

Author: abhinavminhas

QueryDB.Core.Tests/MSSQLTests.cs

@@ -316,19 +316,31 @@ public void Test_MSSQL_ExecuteDDL_Queries()
         {
             var createTableSql = Queries.MSSQLQueries.TestDB.DDL.Create_Table;
             var alterTableSql = Queries.MSSQLQueries.TestDB.DDL.Alter_Table;
+            var commentTableSql = Queries.MSSQLQueries.TestDB.DDL.Comment_Table;
+            var commentTableColumnSql = Queries.MSSQLQueries.TestDB.DDL.Comment_Table_Column;
             var truncateTableSql = Queries.MSSQLQueries.TestDB.DDL.Truncate_Table;
             var renameTableSql = Queries.MSSQLQueries.TestDB.DDL.Rename_Table;
             var dropTableSql = Queries.MSSQLQueries.TestDB.DDL.Drop_Table;
             var dDLExecutionCheckSql = Queries.MSSQLQueries.TestDB.DDL.DDL_Execute_check;
+            var dDLTableCommentCheckSql = Queries.MSSQLQueries.TestDB.DDL.DDL_Table_Comment_check;
+            var dDLTableColumnCommentCheckSql = Queries.MSSQLQueries.TestDB.DDL.DDL_Table_Column_Comment_check;
 
             var dbContext = new DBContext(DB.MSSQL, MSSQLConnectionString);
             dbContext.ExecuteDDL(createTableSql);
             dbContext.ExecuteDDL(alterTableSql);
+            dbContext.ExecuteDDL(commentTableSql);
+            dbContext.ExecuteDDL(commentTableColumnSql);
             dbContext.ExecuteDDL(truncateTableSql);
 
             var tableCount = dbContext
                 .FetchData(string.Format(dDLExecutionCheckSql, "dbo", "Employee"));
             Assert.AreEqual("1", tableCount[0].ReferenceData["Table_Count"]);
+            var tableComment = dbContext
+                .FetchData(string.Format(dDLTableCommentCheckSql, "dbo", "Employee"));
+            Assert.AreEqual("This table stores employee records", tableComment[0].ReferenceData["Table_Comment"]);
+            var tableColumnComment = dbContext
+                .FetchData(string.Format(dDLTableColumnCommentCheckSql, "dbo", "Employee"));
+            Assert.AreEqual("This column stores employee middle name", tableColumnComment[0].ReferenceData["Table_Column_Comment"]);
 
             dbContext.ExecuteDDL(renameTableSql);
 

QueryDB.Core.Tests/Queries.cs

@@ -25,10 +25,14 @@ internal static class DDL
                 {
                     internal static string Create_Table = @"CREATE TABLE Employee (EmployeeID INT PRIMARY KEY, FirstName NVARCHAR(50), LastName NVARCHAR(50))";
                     internal static string Alter_Table = @"ALTER TABLE Employee ADD MiddleName VARCHAR(50)";
+                    internal static string Comment_Table = @"EXEC sp_addextendedproperty @name = N'MS_Description',  @value = N'This table stores employee records', @level0type = N'SCHEMA', @level0name = 'dbo', @level1type = N'TABLE', @level1name = 'Employee'";
+                    internal static string Comment_Table_Column = @"EXEC sp_addextendedproperty @name = N'MS_Description', @value = N'This column stores employee middle name', @level0type = N'SCHEMA', @level0name = 'dbo', @level1type = N'TABLE', @level1name = 'Employee', @level2type = N'COLUMN', @level2name = 'MiddleName'";
                     internal static string Truncate_Table = @"TRUNCATE TABLE Employee";
                     internal static string Rename_Table = @"EXEC SP_RENAME Employee, Employees";
                     internal static string Drop_Table = @"DROP TABLE Employees";
                     internal static string DDL_Execute_check = @"SELECT COUNT(*) AS Table_Count FROM Information_Schema.Tables WHERE LOWER(Table_Schema) = LOWER('{0}') AND LOWER(Table_Name) = LOWER('{1}')";
+                    internal static string DDL_Table_Comment_check = @"SELECT value AS Table_Comment FROM fn_listextendedproperty(NULL, 'SCHEMA', '{0}', 'TABLE', '{1}', NULL, NULL)";
+                    internal static string DDL_Table_Column_Comment_check = @"SELECT value AS Table_Column_Comment FROM fn_listextendedproperty(NULL, 'SCHEMA', '{0}', 'TABLE', '{1}', 'COLUMN', 'MiddleName')";
                 }
             }
         }