@@ -149,6 +149,10 @@ def __init__(self):
149149 self ._defaults ['GetMaxSettablePower' ] = {}
150150 self ._defaults ['GetMaxSettablePower' ]['return' ] = 0
151151 self ._defaults ['GetMaxSettablePower' ]['value' ] = None
152+ self ._defaults ['GetScript' ] = {}
153+ self ._defaults ['GetScript' ]['return' ] = 0
154+ self ._defaults ['GetScript' ]['actualBufferSize' ] = None
155+ self ._defaults ['GetScript' ]['script' ] = None
152156 self ._defaults ['GetSelfCalibrationDateAndTime' ] = {}
153157 self ._defaults ['GetSelfCalibrationDateAndTime' ]['return' ] = 0
154158 self ._defaults ['GetSelfCalibrationDateAndTime' ]['year' ] = None
@@ -686,6 +690,22 @@ def niRFSG_GetMaxSettablePower(self, vi, value): # noqa: N802
686690 value .contents .value = self ._defaults ['GetMaxSettablePower' ]['value' ]
687691 return self ._defaults ['GetMaxSettablePower' ]['return' ]
688692
693+ def niRFSG_GetScript (self , vi , script_name , script , buffer_size , actual_buffer_size ): # noqa: N802
694+ if self ._defaults ['GetScript' ]['return' ] != 0 :
695+ return self ._defaults ['GetScript' ]['return' ]
696+ # actual_buffer_size
697+ if self ._defaults ['GetScript' ]['actualBufferSize' ] is None :
698+ raise MockFunctionCallError ("niRFSG_GetScript" , param = 'actualBufferSize' )
699+ if actual_buffer_size is not None :
700+ actual_buffer_size .contents .value = self ._defaults ['GetScript' ]['actualBufferSize' ]
701+ # script
702+ if self ._defaults ['GetScript' ]['script' ] is None :
703+ raise MockFunctionCallError ("niRFSG_GetScript" , param = 'script' )
704+ if buffer_size .value == 0 :
705+ return len (self ._defaults ['GetScript' ]['script' ])
706+ script .value = self ._defaults ['GetScript' ]['script' ].encode ('ascii' )
707+ return self ._defaults ['GetScript' ]['return' ]
708+
689709 def niRFSG_GetSelfCalibrationDateAndTime (self , vi , module , year , month , day , hour , minute , second ): # noqa: N802
690710 if self ._defaults ['GetSelfCalibrationDateAndTime' ]['return' ] != 0 :
691711 return self ._defaults ['GetSelfCalibrationDateAndTime' ]['return' ]
@@ -1154,6 +1174,8 @@ def set_side_effects_and_return_values(self, mock_library):
11541174 mock_library .niRFSG_GetExternalCalibrationLastDateAndTime .return_value = 0
11551175 mock_library .niRFSG_GetMaxSettablePower .side_effect = MockFunctionCallError ("niRFSG_GetMaxSettablePower" )
11561176 mock_library .niRFSG_GetMaxSettablePower .return_value = 0
1177+ mock_library .niRFSG_GetScript .side_effect = MockFunctionCallError ("niRFSG_GetScript" )
1178+ mock_library .niRFSG_GetScript .return_value = 0
11571179 mock_library .niRFSG_GetSelfCalibrationDateAndTime .side_effect = MockFunctionCallError ("niRFSG_GetSelfCalibrationDateAndTime" )
11581180 mock_library .niRFSG_GetSelfCalibrationDateAndTime .return_value = 0
11591181 mock_library .niRFSG_GetSelfCalibrationTemperature .side_effect = MockFunctionCallError ("niRFSG_GetSelfCalibrationTemperature" )
0 commit comments