Integrate generate script with task service (#426)

* Support generate script for backup

* change

* update task service data contract for Generate Script

* more changes

* update test

* add comments

* Add missing files

* update stub backup operation for testing

* pr comments

* remove empty space

* Fix tests

* Add unit/integration tests and isCancelable to TaskInfo

* address pr comments

* pr comments - fix tests

* fix minor issue

* fix minor issues

* remove unused variable

test/Microsoft.SqlTools.ServiceLayer.UnitTests/TaskServices/DatabaseOperationStub.cs

@@ -23,6 +23,8 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.TaskServices
             Failed = true;
         }
 
+        public TaskScript TaskScript { get; set; }
+
         public TaskResult TaskResult { get; set; }
 
         public bool IsStopped { get; set; }
@@ -62,5 +64,21 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.TaskServices
                 };
             });
         }
+
+        public async Task<TaskResult> FunctionToScript(SqlTask sqlTask)
+        {
+            return await Task.Factory.StartNew(() =>
+            {
+                sqlTask.AddMessage("start scripting", SqlTaskStatus.InProgress, true);
+                TaskScript = sqlTask.AddScript(SqlTaskStatus.Succeeded, "script generated!");
+                sqlTask.AddMessage("done", SqlTaskStatus.Succeeded);
+
+                return new TaskResult
+                {
+                    TaskStatus = SqlTaskStatus.Succeeded,
+                    
+                };
+            });
+        }
     }
 }

test/Microsoft.SqlTools.ServiceLayer.UnitTests/TaskServices/SqlTaskTests.cs

@@ -147,5 +147,26 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.TaskServices
             operation.FailTheOperation();
             await taskToVerify;
         }
+
+        [Fact]
+        public async Task RunScriptShouldReturnScriptContent()
+        {
+            SqlTaskStatus expectedStatus = SqlTaskStatus.Succeeded;
+            DatabaseOperationStub operation = new DatabaseOperationStub();
+            operation.TaskResult = new TaskResult
+            {
+                TaskStatus = expectedStatus
+            };
+            SqlTask sqlTask = new SqlTask(new TaskMetadata(), operation.FunctionToScript, null);
+            Assert.Equal(sqlTask.TaskStatus, SqlTaskStatus.NotStarted);
+
+            Task taskToVerify = sqlTask.RunAsync().ContinueWith(task => {
+                Assert.Equal(sqlTask.TaskStatus, expectedStatus);
+                Assert.Equal(sqlTask.IsCompleted, true);
+                Assert.NotNull(operation.TaskScript);
+                Assert.True(!string.IsNullOrEmpty(operation.TaskScript.Script));
+            });
+            await taskToVerify;
+        }
     }
 }

test/Microsoft.SqlTools.ServiceLayer.UnitTests/TaskServices/TaskManagerTests.cs

@@ -86,5 +86,40 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.TaskServices
             }
 
         }
+
+        [Fact]
+        public async Task VerifyScriptTask()
+        {
+            using (SqlTaskManager manager = new SqlTaskManager())
+            {
+                DatabaseOperationStub operation = new DatabaseOperationStub();
+                operation.TaskResult = new TaskResult
+                {
+                    TaskStatus = SqlTaskStatus.Succeeded
+                };
+                SqlTask sqlTask = manager.CreateTask(taskMetaData, operation.FunctionToScript);
+
+                bool scriptAddedEventRaised = false;
+                string script = null;
+                sqlTask.ScriptAdded += (object sender, TaskEventArgs<TaskScript> e) =>
+                {
+                    scriptAddedEventRaised = true;
+                    script = e.TaskData.Script;
+                };
+
+                Assert.NotNull(sqlTask);
+                
+                Task taskToVerify = sqlTask.RunAsync().ContinueWith(task =>
+                {
+                    Assert.True(scriptAddedEventRaised);
+                    Assert.True(!string.IsNullOrEmpty(script));
+                    Assert.True(manager.HasCompletedTasks());
+                    manager.RemoveCompletedTask(sqlTask);
+                });
+                operation.Stop();
+                await taskToVerify;
+            }
+
+        }
     }
 }