def stage(self, mocker, init_kwargs, fake_auth_provider): stage = pipeline_stages_iothub.UseAuthProviderStage(**init_kwargs) stage.send_op_down = mocker.MagicMock() stage.send_event_up = mocker.MagicMock() # Attach an auth provider set_auth_op = pipeline_ops_iothub.SetAuthProviderOperation( auth_provider=fake_auth_provider, callback=mocker.MagicMock()) stage.run_op(set_auth_op) assert stage.auth_provider is fake_auth_provider stage.send_op_down.reset_mock() stage.send_event_up.reset_mock() return stage
def op(self, mocker, fake_auth_provider): return pipeline_ops_iothub.SetAuthProviderOperation( auth_provider=fake_auth_provider, callback=mocker.MagicMock() )