def test_update_handler_publishes_int_update(pv_value, pv_caproto_type,
pv_numpy_type):
producer = FakeProducer()
context = FakeContext()
pv_source_name = "source_name"
update_handler = CAUpdateHandler(producer, context, pv_source_name,
"output_topic", "f142") # type: ignore
metadata = (0, 0, TimeStamp(4, 0))
context.call_monitor_callback_with_fake_pv_update(
ReadNotifyResponse(
np.array([pv_value]).astype(pv_numpy_type),
pv_caproto_type,
1,
1,
1,
metadata=metadata,
))
assert producer.published_payload is not None
pv_update_output = deserialise_f142(producer.published_payload)
assert pv_update_output.value == pv_value
assert pv_update_output.source_name == pv_source_name
update_handler.stop()
def test_update_handler_publishes_periodic_update():
producer = FakeProducer()
context = FakeContext()
pv_value = 42
pv_caproto_type = ChannelType.TIME_INT
pv_numpy_type = np.int32
pv_source_name = "source_name"
update_period_ms = 10
update_handler = CAUpdateHandler(producer, context, pv_source_name,
"output_topic", "f142",
update_period_ms) # type: ignore
metadata = (0, 0, TimeStamp(4, 0))
context.call_monitor_callback_with_fake_pv_update(
ReadNotifyResponse(
np.array([pv_value]).astype(pv_numpy_type),
pv_caproto_type,
1,
1,
1,
metadata=metadata,
))
assert producer.published_payload is not None
pv_update_output = deserialise_f142(producer.published_payload)
assert pv_update_output.value == pv_value
assert pv_update_output.source_name == pv_source_name
sleep(0.05)
assert (
producer.messages_published > 1
), "Expected more than the 1 message from triggered update due to periodic updates being active"
update_handler.stop()
def test_update_handler_publishes_alarm_update():
producer = FakeProducer()
context = FakeContext()
pv_value = 42
pv_caproto_type = ChannelType.TIME_INT
pv_numpy_type = np.int32
pv_source_name = "source_name"
alarm_status = 6 # AlarmStatus.LOW
alarm_severity = 1 # AlarmSeverity.MINOR
update_handler = CAUpdateHandler(producer, context, pv_source_name,
"output_topic", "f142") # type: ignore
metadata = (alarm_status, alarm_severity, TimeStamp(4, 0))
context.call_monitor_callback_with_fake_pv_update(
ReadNotifyResponse(
np.array([pv_value]).astype(pv_numpy_type),
pv_caproto_type,
1,
1,
1,
metadata=metadata,
))
assert producer.published_payload is not None
pv_update_output = deserialise_f142(producer.published_payload)
assert pv_update_output.value == pv_value
assert pv_update_output.source_name == pv_source_name
assert pv_update_output.alarm_status == AlarmStatus.LOW
assert pv_update_output.alarm_severity == AlarmSeverity.MINOR
update_handler.stop()
def test_update_handler_publishes_enum_update():
producer = FakeProducer()
context = FakeContext()
pv_caproto_type = ChannelType.TIME_ENUM
pv_source_name = "source_name"
update_handler = CAUpdateHandler(producer, context, pv_source_name,
"output_topic", "f142") # type: ignore
# Nothing gets published when ENUM type update is received, the handler will resubscribe using STRING
# type as the string is more useful to forwarder to the filewriter than the enum int
metadata = (0, 0, TimeStamp(4, 0))
context.call_monitor_callback_with_fake_pv_update(
ReadNotifyResponse(
np.array([0]),
pv_caproto_type,
1,
1,
1,
metadata=metadata,
))
# Second update, with STRING type
enum_string_value = "ENUM_STRING"
context.call_monitor_callback_with_fake_pv_update(
ReadNotifyResponse(
[enum_string_value.encode("utf8")],
ChannelType.TIME_STRING,
1,
1,
1,
metadata=metadata,
))
assert (
producer.messages_published == 1
), "Only expected a single message with string payload, not the original enum update"
assert producer.published_payload is not None
pv_update_output = deserialise_f142(producer.published_payload)
assert pv_update_output.value == enum_string_value
assert pv_update_output.source_name == pv_source_name
update_handler.stop()
def test_update_handler_does_not_include_alarm_details_if_unchanged_in_subsequent_updates(
):
producer = FakeProducer()
context = FakeContext()
pv_value = 42
pv_caproto_type = ChannelType.TIME_INT
pv_numpy_type = np.int32
pv_source_name = "source_name"
alarm_status = 6 # AlarmStatus.LOW
alarm_severity = 1 # AlarmSeverity.MINOR
update_handler = CAUpdateHandler(producer, context, pv_source_name,
"output_topic", "f142") # type: ignore
metadata = (alarm_status, alarm_severity, TimeStamp(4, 0))
context.call_monitor_callback_with_fake_pv_update(
ReadNotifyResponse(
np.array([pv_value]).astype(pv_numpy_type),
pv_caproto_type,
1,
1,
1,
metadata=metadata,
))
# Second update, with unchanged alarm
context.call_monitor_callback_with_fake_pv_update(
ReadNotifyResponse(
np.array([pv_value]).astype(pv_numpy_type),
pv_caproto_type,
1,
1,
1,
metadata=metadata,
))
assert producer.messages_published == 2
pv_update_output = deserialise_f142(producer.published_payload)
assert pv_update_output.alarm_status == AlarmStatus.NO_CHANGE
assert pv_update_output.alarm_severity == AlarmSeverity.NO_CHANGE
update_handler.stop()