(fix): Fix bug for test data function

testsigmahq · Mar 7, 2023 · aaeb6d4 · aaeb6d4
1 parent abd7d4c
commit aaeb6d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/server/src/main/java/com/testsigma/step/processors/StepProcessor.java b/server/src/main/java/com/testsigma/step/processors/StepProcessor.java
@@ -486,7 +486,7 @@ private void populateTestDataFunctionDetailsFromMap(DefaultDataGeneratorsEntity
     DefaultDataGenerator defaultDataGenerator = defaultDataGeneratorService.find(testStepDTO.getTestDataFunctionId());
     defaultDataGeneratorsEntity.setClassName(defaultDataGenerator.getFile().getClassName());
     defaultDataGeneratorsEntity.setFunctionName(defaultDataGenerator.getFunctionName());
-    Map<String, String> args = (HashMap) defaultDataGenerator.getArguments();
+    Map<String, String> args = (HashMap) testStepDTO.getTestDataFunctionArgs();
     defaultDataGeneratorsEntity.setArguments(args);
     Map<String, String> argsTypes = (HashMap) defaultDataGenerator.getArguments().get("arg_types");
     defaultDataGeneratorsEntity.setArgumentTypes(argsTypes);