|
27 | 27 | THING_TYPES = ["person", "chair", "table", "couch"] |
28 | 28 | MAX_CONTROLLER_WAIT = 30 # seconds |
29 | 29 |
|
30 | | -class WillOurShipGo(FunctionalTest): |
| 30 | +class SensorMqttMessageFlowTest(FunctionalTest): |
31 | 31 | def __init__(self, testName, request, recordXMLAttribute): |
32 | 32 | super().__init__(testName, request, recordXMLAttribute) |
33 | 33 | self.sceneUID = self.params['scene_id'] |
@@ -353,13 +353,9 @@ def _verify_region_events(self): |
353 | 353 | events = self.regionEvents.get(sensor_name, []) |
354 | 354 | assert events, f"No events received for sensor {sensor_name}" |
355 | 355 |
|
356 | | - saw_enter_or_exit = False |
357 | 356 | saw_sensor_payload = False |
358 | 357 | exited_ids = set() |
359 | 358 | for event in events: |
360 | | - if event.get('entered') or event.get('exited'): |
361 | | - saw_enter_or_exit = True |
362 | | - |
363 | 359 | for entered in event.get('entered', []): |
364 | 360 | entered_id = self._extract_obj_id(entered) |
365 | 361 | if entered_id is not None and entered_id in exited_ids: |
@@ -389,7 +385,6 @@ def _verify_region_events(self): |
389 | 385 | sensors = obj.get('sensors', {}) |
390 | 386 | if sensor_name in sensors and sensors[sensor_name].get('values'): |
391 | 387 | saw_sensor_payload = True |
392 | | - assert saw_enter_or_exit, f"No entered/exited event data for sensor {sensor_name}" |
393 | 388 | assert saw_sensor_payload, f"No sensor payload found in events for sensor {sensor_name}" |
394 | 389 | return |
395 | 390 |
|
@@ -626,14 +621,14 @@ def checkForMalfunctions(self): |
626 | 621 | self.recordTestResult() |
627 | 622 | return |
628 | 623 |
|
629 | | -def test_sensor_region_events(request, record_xml_attribute): |
630 | | - test = WillOurShipGo(TEST_NAME, request, record_xml_attribute) |
| 624 | +def test_sensor_mqtt_message_flow(request, record_xml_attribute): |
| 625 | + test = SensorMqttMessageFlowTest(TEST_NAME, request, record_xml_attribute) |
631 | 626 | test.checkForMalfunctions() |
632 | 627 | assert test.exitCode == 0 |
633 | 628 | return |
634 | 629 |
|
635 | 630 | def main(): |
636 | | - return test_sensor_region_events(None, None) |
| 631 | + return test_sensor_mqtt_message_flow(None, None) |
637 | 632 |
|
638 | 633 | if __name__ == '__main__': |
639 | 634 | os._exit(main() or 0) |
0 commit comments