@@ -262,7 +262,7 @@ def test_sends_op_down(self, mocker, stage, op):
262262
263263
264264@pytest .mark .describe (
265- "EnsureDesiredPropertiesStage - OCCURRENCE: ConnectedEvent received (ensure_desired_properties enabled)"
265+ "EnsureDesiredPropertiesStage - .handle_pipeline_event() -- Called with ConnectedEvent (ensure_desired_properties enabled)"
266266)
267267class TestEnsureDesiredPropertiesStageWhenConnectedEventReceivedWithEnsureDesiredPropertiesEnabled (
268268 EnsureDesiredPropertiesStageTestConfig , StageHandlePipelineEventTestBase
@@ -332,7 +332,7 @@ def test_no_last_version_seen_pending(self, mocker, stage, event):
332332
333333
334334@pytest .mark .describe (
335- "EnsureDesiredPropertiesStage - OCCURRENCE: ConnectedEvent received (ensure_desired_properties disabled)"
335+ "EnsureDesiredPropertiesStage - .handle_pipeline_event() -- Called with ConnectedEvent (ensure_desired_properties disabled)"
336336)
337337class TestEnsureDesiredPropertiesStageWhenConnectedEventReceivedWithEnsureDesiredPropertiesDisabled (
338338 EnsureDesiredPropertiesStageTestConfig , StageHandlePipelineEventTestBase
@@ -362,7 +362,7 @@ def test_no_get_twin_op(self, stage, event):
362362
363363
364364@pytest .mark .describe (
365- "EnsureDesiredPropertiesStage - OCCURRENCE: TwinDesiredPropertiesPatchEvent received (ensure_desired_properties enabled)"
365+ "EnsureDesiredPropertiesStage - .handle_pipeline_event() -- Called with TwinDesiredPropertiesPatchEvent (ensure_desired_properties enabled)"
366366)
367367class TestEnsureDesiredPropertiesStageWhenTwinDesiredPropertiesPatchEventReceivedWithEnsureDesiredPropertiesEnabled (
368368 EnsureDesiredPropertiesStageTestConfig , StageHandlePipelineEventTestBase
@@ -404,7 +404,7 @@ def test_sends_event_up(self, mocker, stage, event, version):
404404
405405
406406@pytest .mark .describe (
407- "EnsureDesiredPropertiesStage - OCCURRENCE: TwinDesiredPropertiesPatchEvent received (ensure_desired_properties disabled)"
407+ "EnsureDesiredPropertiesStage - .handle_pipeline_event() -- Called with TwinDesiredPropertiesPatchEvent (ensure_desired_properties disabled)"
408408)
409409class TestEnsureDesiredPropertiesStageWhenTwinDesiredPropertiesPatchEventReceivedWithEnsureDesiredPropertiesDisabled (
410410 EnsureDesiredPropertiesStageTestConfig , StageHandlePipelineEventTestBase
@@ -427,7 +427,7 @@ def version(self, mocker):
427427 def event (self , version ):
428428 return pipeline_events_iothub .TwinDesiredPropertiesPatchEvent (patch = {"$version" : version })
429429
430- @pytest .mark .it ("Saves the `$version` attribute of the patch into `last_version_seen`" )
430+ @pytest .mark .it ("Does not change `last_version_seen`" )
431431 def test_doesnt_save_the_last_version_seen (self , mocker , stage , event , version ):
432432 assert stage .nucleus .pipeline_configuration .ensure_desired_properties is False
433433 stage .last_version_seen = mocker .MagicMock ()
@@ -446,6 +446,72 @@ def test_sends_event_up(self, mocker, stage, event, version):
446446 assert stage .send_event_up .call_args == mocker .call (event )
447447
448448
449+ @pytest .mark .describe (
450+ "EnsureDesiredPropertiesStage - .handle_pipeline_event() -- Called with other arbitrary event (ensure_desired_properties enabled)"
451+ )
452+ class TestEnsureDesiredPropertiesStageWhenArbitraryEventReceivedWithEnsureDesiredPropertiesEnabled (
453+ EnsureDesiredPropertiesStageTestConfig , StageHandlePipelineEventTestBase
454+ ):
455+ @pytest .fixture
456+ def stage (self , mocker , cls_type , init_kwargs , nucleus ):
457+ stage = cls_type (** init_kwargs )
458+ stage .nucleus = nucleus
459+ stage .nucleus .pipeline_configuration .ensure_desired_properties = True
460+ stage .send_op_down = mocker .MagicMock ()
461+ stage .send_event_up = mocker .MagicMock ()
462+ mocker .spy (stage , "report_background_exception" )
463+ return stage
464+
465+ @pytest .fixture
466+ def version (self , mocker ):
467+ return mocker .MagicMock ()
468+
469+ @pytest .fixture
470+ def event (self , arbitrary_event ):
471+ return arbitrary_event
472+
473+ @pytest .mark .it ("Sends the event to the previous stage" )
474+ def test_sends_event_up (self , mocker , stage , event ):
475+ assert stage .nucleus .pipeline_configuration .ensure_desired_properties
476+ stage .handle_pipeline_event (event )
477+
478+ assert stage .send_event_up .call_count == 1
479+ assert stage .send_event_up .call_args == mocker .call (event )
480+
481+
482+ @pytest .mark .describe (
483+ "EnsureDesiredPropertiesStage - .handle_pipeline_event() -- Called with other arbitrary event (ensure_desired_properties disabled)"
484+ )
485+ class TestEnsureDesiredPropertiesStageWhenArbitraryEventReceivedWithEnsureDesiredPropertiesDisabled (
486+ EnsureDesiredPropertiesStageTestConfig , StageHandlePipelineEventTestBase
487+ ):
488+ @pytest .fixture
489+ def stage (self , mocker , cls_type , init_kwargs , nucleus ):
490+ stage = cls_type (** init_kwargs )
491+ stage .nucleus = nucleus
492+ stage .nucleus .pipeline_configuration .ensure_desired_properties = False
493+ stage .send_op_down = mocker .MagicMock ()
494+ stage .send_event_up = mocker .MagicMock ()
495+ mocker .spy (stage , "report_background_exception" )
496+ return stage
497+
498+ @pytest .fixture
499+ def version (self , mocker ):
500+ return mocker .MagicMock ()
501+
502+ @pytest .fixture
503+ def event (self , arbitrary_event ):
504+ return arbitrary_event
505+
506+ @pytest .mark .it ("Sends the event to the previous stage" )
507+ def test_sends_event_up (self , mocker , stage , event ):
508+ assert stage .nucleus .pipeline_configuration .ensure_desired_properties is False
509+ stage .handle_pipeline_event (event )
510+
511+ assert stage .send_event_up .call_count == 1
512+ assert stage .send_event_up .call_args == mocker .call (event )
513+
514+
449515@pytest .mark .describe (
450516 "EnsureDesiredPropertiesStage - OCCURRENCE: GetTwinOperation that was sent down by this stage completes (ensure_desired_properties enabled)"
451517)
@@ -602,72 +668,6 @@ def test_does_not_change_last_version_seen_if_same_version(
602668 assert stage .last_version_seen == new_version
603669
604670
605- @pytest .mark .describe (
606- "EnsureDesiredPropertiesStage - OCCURRENCE: ArbitraryEvent received (ensure_desired_properties enabled)"
607- )
608- class TestEnsureDesiredPropertiesStageWhenArbitraryEventReceivedWithEnsureDesiredPropertiesEnabled (
609- EnsureDesiredPropertiesStageTestConfig , StageHandlePipelineEventTestBase
610- ):
611- @pytest .fixture
612- def stage (self , mocker , cls_type , init_kwargs , nucleus ):
613- stage = cls_type (** init_kwargs )
614- stage .nucleus = nucleus
615- stage .nucleus .pipeline_configuration .ensure_desired_properties = True
616- stage .send_op_down = mocker .MagicMock ()
617- stage .send_event_up = mocker .MagicMock ()
618- mocker .spy (stage , "report_background_exception" )
619- return stage
620-
621- @pytest .fixture
622- def version (self , mocker ):
623- return mocker .MagicMock ()
624-
625- @pytest .fixture
626- def event (self , arbitrary_event ):
627- return arbitrary_event
628-
629- @pytest .mark .it ("Sends the event to the previous stage" )
630- def test_sends_event_up (self , mocker , stage , event ):
631- assert stage .nucleus .pipeline_configuration .ensure_desired_properties
632- stage .handle_pipeline_event (event )
633-
634- assert stage .send_event_up .call_count == 1
635- assert stage .send_event_up .call_args == mocker .call (event )
636-
637-
638- @pytest .mark .describe (
639- "EnsureDesiredPropertiesStage - OCCURRENCE: ArbitraryEvent received (ensure_desired_properties disabled)"
640- )
641- class TestEnsureDesiredPropertiesStageWhenArbitraryEventReceivedWithEnsureDesiredPropertiesDisabled (
642- EnsureDesiredPropertiesStageTestConfig , StageHandlePipelineEventTestBase
643- ):
644- @pytest .fixture
645- def stage (self , mocker , cls_type , init_kwargs , nucleus ):
646- stage = cls_type (** init_kwargs )
647- stage .nucleus = nucleus
648- stage .nucleus .pipeline_configuration .ensure_desired_properties = False
649- stage .send_op_down = mocker .MagicMock ()
650- stage .send_event_up = mocker .MagicMock ()
651- mocker .spy (stage , "report_background_exception" )
652- return stage
653-
654- @pytest .fixture
655- def version (self , mocker ):
656- return mocker .MagicMock ()
657-
658- @pytest .fixture
659- def event (self , arbitrary_event ):
660- return arbitrary_event
661-
662- @pytest .mark .it ("Sends the event to the previous stage" )
663- def test_sends_event_up (self , mocker , stage , event ):
664- assert stage .nucleus .pipeline_configuration .ensure_desired_properties is False
665- stage .handle_pipeline_event (event )
666-
667- assert stage .send_event_up .call_count == 1
668- assert stage .send_event_up .call_args == mocker .call (event )
669-
670-
671671###############################
672672# TWIN REQUEST RESPONSE STAGE #
673673###############################
0 commit comments