TestCsc¶
- class lsst.ts.salobj.TestCsc(index: int, config_dir: str | pathlib.Path | None = None, check_if_duplicate: bool = False, initial_state: State = State.STANDBY, override: str = '', simulation_mode: int = 0)¶
Bases:
ConfigurableCsc
A simple CSC intended for unit testing.
Supported commands:
setScalars
andsetArrays
: output the provided data using the corresponding event and telemetry topics. Note that this violates the convention that telemetry is output at regular intervals, but it makes unit tests much easier to write.wait
: wait for the specified amount of time, and, if requested, raise an exception. One use for this is to test command timeout by specifying a long wait and waiting a shorter time for the command to finish. Another use is to test multiple simultaneous commands, sincewait
supports this.The standard state transition commands do the usual thing and output the
summaryState
event. TheexitControl
command shuts the CSC down.
- Parameters:
- index
int
Index of Test component; each unit test method should use a different index.
- config_dir
str
, optional Path to configuration files.
- check_if_duplicate
bool
, optional Check for heartbeat events from the same SAL name and index at startup (before starting the heartbeat loop)? Defaults to False in order to speed up unit tests, but
amain
sets it true.- initial_state
salobj.State
, optional The initial state of the CSC. Typically one of:
salobj.State.ENABLED
if you want the CSC immediately usable.salobj.State.STANDBY
if you want full emulation of a CSC.
- override
str
, optional Configuration override file to apply if
initial_state
isState.DISABLED
orState.ENABLED
.- simulation_mode
int
, optional Simulation mode. The only allowed value is 0.
- index
- Raises:
- ValueError
If
config_dir
is not a directory orinitial_state
is invalid.- salobj.ExpectedError
If
simulation_mode
is invalid. Note: you will only see this error if you awaitstart_task
.
Notes
Unlike a normal CSC this one does not output telemetry at regular intervals. Instead, in order to simplify unit tests, it outputs the
arrays
andscalars
telemetry topics in reponse to thesetArrays
orsetScalars
command (just like thearrays
andscalars
events). That makes it more predictable when this data will appear. Note that theheartbeat
event is output at regular intervals, as for any CSC.Also, unlike most normal configurable CSCs, this one does not need to be configured in order to be used (though self.config will be None). Thus it is safe to start this CSC in the
salobj.State.ENABLED
state.Error Codes
1: the fault command was executed
Attributes Summary
Get a tuple of the fields in an arrays struct.
Get or set the configuration directory.
Return True if the summary state is
State.DISABLED
orState.ENABLED
.Get a dict of field_name: element type.
Get a tuple of the integer fields in a struct.
Get a tuple of the fields in a scalars struct.
Get the current simulation mode.
Get the summary state as a
State
enum.Methods Summary
add_arguments
(parser)Add arguments to the parser created by
make_from_cmd_line
.add_kwargs_from_args
(args, kwargs)Add constructor keyword arguments based on parsed arguments.
amain
(index, **kwargs)Make a CSC from command-line arguments and run it.
as_dict
(data, fields)Return the specified fields from a data struct as a dict.
assert_arrays_equal
(arrays1, arrays2)Assert that two arrays data structs are equal.
assert_enabled
([action])Assert that an action that requires ENABLED state can be run.
assert_scalars_equal
(scalars1, scalars2)Assert that two scalars data structs are equal.
begin_disable
(data)Begin do_disable; called before state changes.
begin_enable
(data)Begin do_enable; called before state changes.
begin_exitControl
(data)Begin do_exitControl; called before state changes.
begin_standby
(data)Begin do_standby; called before the state changes.
begin_start
(data)Begin do_start; configure the CSC before changing state.
check_for_duplicate_heartbeat
([num_messages])Monitor heartbeat events and return True if any have a different private_origin.
close
([exception, cancel_start])Shut down, clean up resources and set done_task done.
Shut down pending tasks.
configure
(config)Configure the CSC.
do_disable
(data)Transition from
State.ENABLED
toState.DISABLED
.do_enable
(data)Transition from
State.DISABLED
toState.ENABLED
.do_exitControl
(data)Transition from
State.STANDBY
toState.OFFLINE
and quit.do_fault
(data)Execute the fault command.
do_newCommand
(data)This method is defined here to provide a test to the extra_commands features in CSC.
do_setArrays
(data)Execute the setArrays command.
do_setLogLevel
(data)Set logging level.
do_setScalars
(data)Execute the setScalars command.
do_standby
(data)Transition from
State.DISABLED
orState.FAULT
toState.STANDBY
.do_start
(data)Transition from
State.STANDBY
toState.DISABLED
.do_wait
(data)Execute the wait command by waiting for the specified duration.
end_disable
(data)End do_disable; called after state changes but before command acknowledged.
end_enable
(data)End do_enable; called after state changes but before command acknowledged.
end_exitControl
(data)End do_exitControl; called after state changes but before command acknowledged.
end_standby
(data)End do_standby; called after state changes but before command acknowledged.
end_start
(data)End do_start; called after state changes but before command acknowledged.
fault
(code, report[, traceback])Enter the fault state and output the
errorCode
event.Get the name of the configuration package, e.g.
Called when the summary state has changed.
implement_simulation_mode
(simulation_mode)Implement going into or out of simulation mode.
make_from_cmd_line
(index[, check_if_duplicate])Construct a CSC from command line arguments.
Make a random arrays data dict.
Make a random arrays data dict.
Output the logLevel event.
Read the config dir and put configurationsAvailable if changed.
read_config_files
(config_validator, ...[, ...])Read a set of configuration files and return the validated config.
set_simulation_mode
(simulation_mode)Set the simulation mode.
start
()Finish constructing the CSC.
Handle the initial state.
Attributes Documentation
- arrays_fields¶
Get a tuple of the fields in an arrays struct.
- config_dir¶
Get or set the configuration directory.
- Parameters:
- config_dir
str
,pathlib.Path
New configuration directory.
- config_dir
- Returns:
- config_dir
pathlib.Path
Absolute path to the configuration directory.
- config_dir
- Raises:
- ValueError
If the new configuration dir is not a directory.
- disabled_or_enabled¶
Return True if the summary state is
State.DISABLED
orState.ENABLED
.This is useful in
handle_summary_state
to determine if you should start or stop a telemetry loop, and connect to or disconnect from an external controller
- domain¶
- enable_cmdline_state = True¶
- field_type¶
Get a dict of field_name: element type.
- int_fields¶
Get a tuple of the integer fields in a struct.
- scalars_fields¶
Get a tuple of the fields in a scalars struct.
- simulation_mode¶
Get the current simulation mode.
0 means normal operation (no simulation).
- Raises:
- ExpectedError
If the new simulation mode is not a supported value.
- version = '?'¶
Methods Documentation
- classmethod add_arguments(parser: ArgumentParser) None ¶
Add arguments to the parser created by
make_from_cmd_line
.- Parameters:
- parser
argparse.ArgumentParser
The argument parser.
- parser
Notes
If you override this method then you should almost certainly override
add_kwargs_from_args
as well.
- classmethod add_kwargs_from_args(args: Namespace, kwargs: dict[str, Any]) None ¶
Add constructor keyword arguments based on parsed arguments.
- Parameters:
- args
argparse.Namespace
Parsed command.
- kwargs
dict
Keyword argument dict for the constructor. Update this based on
args
. The index argument will already be present if relevant.
- args
Notes
If you override this method then you should almost certainly override
add_arguments
as well.
- async classmethod amain(index: int | enum.IntEnum | bool | None, **kwargs: Any) None ¶
Make a CSC from command-line arguments and run it.
- Parameters:
- index
int
,enum.IntEnum
,True
, orNone
If the CSC is indexed, do one of the following:
Specify
True
to makeindex
a required command-line argument that accepts any nonzero index.Specify an
enum.IntEnum
class to makeindex
a required command-line argument that only accepts the enum values.Specify a non-zero integer to use that index. This is rare; if the CSC is indexed then the user should usually be allowed to specify the index.
If the CSC is not indexed, specify
None
or 0.- **kwargs
dict
, optional Additional keyword arguments for your CSC’s constructor.
- index
- as_dict(data: Any, fields: Sequence[str]) dict[str, Any] ¶
Return the specified fields from a data struct as a dict.
- assert_arrays_equal(arrays1: Any, arrays2: Any) None ¶
Assert that two arrays data structs are equal.
The types need not match; each struct can be command, event or telemetry data, or a dict of field: value.
- assert_enabled(action: str = '') None ¶
Assert that an action that requires ENABLED state can be run.
- Parameters:
- action
str
, optional Action attempted. Not needed if this is called at the beginning of a
do_...
method, since the user will know what command was called.
- action
- assert_scalars_equal(scalars1: Any, scalars2: Any) None ¶
Assert that two scalars data structs are equal.
The types need not match; each struct can be command, event or telemetry data, or a dict of field: value.
- async begin_disable(data: BaseMsgType) None ¶
Begin do_disable; called before state changes.
- Parameters:
- data
DataType
Command data
- data
- async begin_enable(data: BaseMsgType) None ¶
Begin do_enable; called before state changes.
- Parameters:
- data
DataType
Command data
- data
- async begin_exitControl(data: BaseMsgType) None ¶
Begin do_exitControl; called before state changes.
- Parameters:
- data
DataType
Command data
- data
- async begin_standby(data: BaseMsgType) None ¶
Begin do_standby; called before the state changes.
- Parameters:
- data
DataType
Command data
- data
- async begin_start(data: BaseMsgType) None ¶
Begin do_start; configure the CSC before changing state.
- Parameters:
- data
cmd_start.DataType
Command data
- data
Notes
The
override
field must be one of:The name of a config label or config file
The name and version of a config file, formatted as
<file_name>:<version>
, where the version is a git reference, such as a git tag or commit hash. This form does not support labels.
- async check_for_duplicate_heartbeat(num_messages: int = 5) int ¶
Monitor heartbeat events and return True if any have a different private_origin.
Intended for use by check_if_duplicate.
The heartbeat loop should be running before this is called. This avoids the issue that if 2 CSCs are started at the same time then neither will see the other one’s heartbeat. This code also assumes that will be true; it will likely raise asyncio.TimeoutError if not.
- async close(exception: Exception | None = None, cancel_start: bool = True) None ¶
Shut down, clean up resources and set done_task done.
May be called multiple times. The first call closes the Controller; subsequent calls wait until the Controller is closed.
Subclasses should override
close_tasks
instead ofclose
, unless you have a good reason to do otherwise.- Parameters:
- exception
Exception
, optional The exception that caused stopping, if any, in which case the
self.done_task
exception is set to this value. SpecifyNone
for a normal exit, in which case theself.done_task
result is set toNone
.- cancel_start
bool
, optional Cancel the start task? Leave this true unless calling this from the start task.
- exception
Notes
Removes the SAL log handler, calls
close_tasks
to stop all background tasks, pauses briefly to allow final SAL messages to be sent, then closes the dds domain.
- async configure(config: SimpleNamespace) None ¶
Configure the CSC.
- Parameters:
- config
object
The configuration, as described by the config schema, as a struct-like object.
- config
Notes
Called when running the
start
command, just before changing summary state fromState.STANDBY
toState.DISABLED
.
- async do_disable(data: BaseMsgType) None ¶
Transition from
State.ENABLED
toState.DISABLED
.- Parameters:
- data
cmd_disable.DataType
Command data
- data
- async do_enable(data: BaseMsgType) None ¶
Transition from
State.DISABLED
toState.ENABLED
.- Parameters:
- data
cmd_enable.DataType
Command data
- data
- async do_exitControl(data: BaseMsgType) None ¶
Transition from
State.STANDBY
toState.OFFLINE
and quit.- Parameters:
- data
cmd_exitControl.DataType
Command data
- data
- async do_fault(data: BaseMsgType) None ¶
Execute the fault command.
Change the summary state to State.FAULT
- async do_newCommand(data: BaseMsgType) None ¶
This method is defined here to provide a test to the extra_commands features in CSC.
Warning: DO NOTE REMOVE THIS COMMAND.
- async do_setArrays(data: BaseMsgType) None ¶
Execute the setArrays command.
- async do_setLogLevel(data: BaseMsgType) None ¶
Set logging level.
- Parameters:
- data
cmd_setLogLevel.DataType
Logging level.
- data
- async do_setScalars(data: BaseMsgType) None ¶
Execute the setScalars command.
- async do_standby(data: BaseMsgType) None ¶
Transition from
State.DISABLED
orState.FAULT
toState.STANDBY
.- Parameters:
- data
cmd_standby.DataType
Command data
- data
- async do_start(data: BaseMsgType) None ¶
Transition from
State.STANDBY
toState.DISABLED
.- Parameters:
- data
cmd_start.DataType
Command data
- data
- async do_wait(data: BaseMsgType) None ¶
Execute the wait command by waiting for the specified duration.
If duration is negative then wait for abs(duration) but do not acknowledge the command as “in progress”. This is useful for testing command timeout.
- async end_disable(data: BaseMsgType) None ¶
End do_disable; called after state changes but before command acknowledged.
- Parameters:
- data
DataType
Command data
- data
- async end_enable(data: BaseMsgType) None ¶
End do_enable; called after state changes but before command acknowledged.
- Parameters:
- data
DataType
Command data
- data
- async end_exitControl(data: BaseMsgType) None ¶
End do_exitControl; called after state changes but before command acknowledged.
- Parameters:
- data
DataType
Command data
- data
- async end_standby(data: BaseMsgType) None ¶
End do_standby; called after state changes but before command acknowledged.
- Parameters:
- data
DataType
Command data
- data
- async end_start(data: BaseMsgType) None ¶
End do_start; called after state changes but before command acknowledged.
- Parameters:
- data
DataType
Command data
- data
- async fault(code: int | None, report: str, traceback: str = '') None ¶
Enter the fault state and output the
errorCode
event.
- async handle_summary_state() None ¶
Called when the summary state has changed.
Override to perform tasks such as starting and stopping telemetry (example).
Notes
The versions in
BaseCsc
andConfigurableCsc
do nothing, so if you subclass one of those you do not need to callawait super().handle_summary_state()
.
- async implement_simulation_mode(simulation_mode: int) None ¶
Implement going into or out of simulation mode.
Deprecated. See simulation mode for details.
- Parameters:
- simulation_mode
int
Requested simulation mode; 0 for normal operation.
- simulation_mode
- Raises:
- ExpectedError
If
simulation_mode
is not a supported value.
- classmethod make_from_cmd_line(index: int | enum.IntEnum | bool | None, check_if_duplicate: bool = False, **kwargs: Any) BaseCsc ¶
Construct a CSC from command line arguments.
- Parameters:
- index
int
,enum.IntEnum
,True
, orNone
If the CSC is indexed, do one of the following:
Specify
True
to makeindex
a required command-line argument that accepts any nonzero index.Specify an
enum.IntEnum
class to makeindex
a required command-line argument that only accepts the enum values.Specify a non-zero integer to use that index. This is rare; if the CSC is indexed then the user should usually be allowed to specify the index.
If the CSC is not indexed, specify
None
or 0.- check_if_duplicate
bool
, optional Check for heartbeat events from the same SAL name and index at startup (before starting the heartbeat loop)? The default is False, to match the BaseCsc default. Note: handled by setting the attribute directly instead of as a constructor argument, because CSCs may not all support the constructor argument.
- **kwargs
dict
, optional Additional keyword arguments for your CSC’s constructor.
- index
- Returns:
- csc
cls
The CSC.
- csc
Notes
To add additional command-line arguments, override
add_arguments
andadd_kwargs_from_args
.
- async read_config_dir() None ¶
Read the config dir and put configurationsAvailable if changed.
Output the
configurationsAvailable
event (if changed), after updating theoverrides
andversion
fields. Also update theversion
field ofevt_configurationApplied
, in preparation for the next time the event is output.
- classmethod read_config_files(config_validator: Draft7Validator, config_dir: Path, files_to_read: list[str], git_hash: str = '') SimpleNamespace ¶
Read a set of configuration files and return the validated config.
- Parameters:
- config_validatorjsonschema validator
Schema validator for configuration.
- config_dirpathlib.Path
Path to config files.
- files_to_readList [str]
Names of files to read, with .yaml suffix. Empty names are ignored (a useful feature for BaseConfigTestCase). The files are read in order, with each later file overriding values that have been accumulated so far.
- git_hashstr, optional
Git hash to use for the files. “” if current.
- Returns:
- types.SimpleNamespace
The validated config as a simple namespace.
- Raises:
- ExpectedError
If the specified configuration files cannot be found, cannot be parsed as yaml dicts, or produce an invalid configuration (one that does not match the schema).
- async set_simulation_mode(simulation_mode: int) None ¶
Set the simulation mode.
Await implement_simulation_mode, update the simulation mode property and report the new value.
- Parameters:
- simulation_mode
int
Requested simulation mode; 0 for normal operation.
- simulation_mode
- async start() None ¶
Finish constructing the CSC.
Call
set_simulation_mode
. If this fails, setself.start_task
to the exception, callstop
, making the CSC unusable, and return.Call
handle_summary_state
Set
self.start_task
done.