def stage(self, mocker):
stage = pipeline_stages_iothub.UseAuthProviderStage()
auth_provider = mocker.MagicMock()
auth_provider.get_current_sas_token = mocker.MagicMock(return_value=fake_sas_token)
stage.auth_provider = auth_provider
return stage
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 stage(self, mocker):
stage = pipeline_stages_iothub.UseAuthProviderStage()
mocker.spy(stage, "send_op_down")
return stage
def test_auth_provider(self, init_kwargs):
stage = pipeline_stages_iothub.UseAuthProviderStage(**init_kwargs)
assert stage.auth_provider is None
def stage(self):
return pipeline_stages_iothub.UseAuthProviderStage()
