Skip to content

Commit b8c0266

Browse files
authored
fix createLiteData regression (#607)
* fix createLiteData regression * fix unit tests
1 parent 7af22a2 commit b8c0266

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

src/snapred/backend/service/LiteDataService.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@ def createLiteData(
4242
liteDataMap = self._ensureLiteDataMap()
4343
runNumber = inputWorkspace.split("_")[-1].lstrip("0")
4444

45-
ffIngredients = FarmFreshIngredients(runNumber=runNumber, useLiteMode=True)
45+
state, _ = self.dataFactoryService.constructStateId(runNumber)
46+
ffIngredients = FarmFreshIngredients(runNumber=runNumber, useLiteMode=True, state=state)
4647
ingredients = self.sousChef.prepInstrumentState(ffIngredients)
4748

4849
recipeKwargs = {

tests/unit/backend/service/test_LiteDataService.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@ def test_createLiteData_calls_executeRecipe_with_correct_arguments(
3636
liteDataService.sousChef.prepInstrumentState = Mock()
3737
liteDataService.sousChef.prepInstrumentState.return_value = Mock()
3838
liteDataService.sousChef.prepInstrumentState.return_value.model_dump_json.return_value = "{}" # noqa: E501
39+
liteDataService.dataFactoryService = Mock()
40+
liteDataService.dataFactoryService.constructStateId = Mock(return_value=("state_id", "state"))
3941

4042
with TemporaryDirectory(dir=Resource.getPath("outputs"), suffix="/") as tmpdir:
4143
outputPath = Path(tmpdir, "555.nxs.h5")
@@ -75,6 +77,8 @@ def test_createLiteData_fails(self, mock_executeRecipe):
7577
liteDataService.sousChef.prepInstrumentState = Mock()
7678
liteDataService.sousChef.prepInstrumentState.return_value = Mock()
7779
liteDataService.sousChef.prepInstrumentState.return_value.model_dump_json.return_value = "{}" # noqa: E501
80+
liteDataService.dataFactoryService = Mock()
81+
liteDataService.dataFactoryService.constructStateId = Mock(return_value=("state_id", "state"))
7882

7983
inputWorkspace = "_test_liteservice_555"
8084
outputWorkspace = "_test_output_lite_"

0 commit comments

Comments
 (0)