Add ws explicitly to ADS

mantidproject · Dec 6, 2024 · 97012f9 · 97012f9
1 parent fd918ca
commit 97012f9
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/workspacemanager_presenter_ads_test.py b/tests/workspacemanager_presenter_ads_test.py
@@ -31,7 +31,8 @@ def test_ensure_that_the_ads_observer_calls_clear_handle(self):
           presenter.delete_handle, presenter.clear_handle, presenter.rename_handle
         )
 
-        CreateSampleWorkspace(OutputWorkspace="ws", StoreInADS=True)
+        workspace = CreateSampleWorkspace(OutputWorkspace="ws", StoreInADS=True)
+        AnalysisDataService.addOrReplace("ws", workspace)
         AnalysisDataService.clear(True)
 
         presenter.clear_handle.assert_called_once()
@@ -44,8 +45,8 @@ def test_ensure_that_the_ads_observer_calls_delete_handle(self):
             presenter.delete_handle, presenter.clear_handle, presenter.rename_handle
         )
 
-        CreateSampleWorkspace(OutputWorkspace="ws", StoreInADS=True)
-        print(AnalysisDataService.getObjectNames())
+        workspace = CreateSampleWorkspace(OutputWorkspace="ws", StoreInADS=True)
+        AnalysisDataService.addOrReplace("ws", workspace)
         AnalysisDataService.remove("ws")
 
         presenter.delete_handle.assert_called_once_with("ws")
@@ -58,7 +59,8 @@ def test_ensure_that_the_ads_observer_calls_rename_handle(self):
             presenter.delete_handle, presenter.clear_handle, presenter.rename_handle
         )
 
-        CreateSampleWorkspace(OutputWorkspace="ws", StoreInADS=True)
+        workspace = CreateSampleWorkspace(OutputWorkspace="ws", StoreInADS=True)
+        AnalysisDataService.addOrReplace("ws", workspace)
         RenameWorkspace(InputWorkspace="ws", OutputWorkspace="ws1")
 
         presenter.rename_handle.assert_called_once_with("ws", "ws1")