public void Do_not_send_hearbeat_to_amazon_swf_when_no_interval_is_set()
{
var heartbeatApi = new TestHeartbeatSwfApi(() => false);
var hearbeat = new ActivityHeartbeat();
hearbeat.StartHeartbeatIfEnabled(heartbeatApi, "token");
Assert.IsFalse(heartbeatApi.Wait(HeartbeatIntervel * 100));
Assert.That(heartbeatApi.HearbeatRecorded, Is.False);
}
public async Task Heartbeat_started_when_it_it_enabled_on_activity_programmatically()
{
var hearbeatApi = new TestHeartbeatSwfApi(() => false);
var activity = new ActivityWithHeartbeatEnabledProgrammatically("details", TimeSpan.FromSeconds(5));
activity.SetSwfApi(hearbeatApi);
await activity.ExecuteAsync(_activityArgs);
Assert.IsTrue(hearbeatApi.Wait(WaitPeriod));
Assert.That(hearbeatApi.Details, Is.EqualTo("details"));
}
public async Task Does_not_send_hearbeat_to_amazon_swf_when_not_enabled()
{
var hearbeatApi = new TestHeartbeatSwfApi(() => false);
var activity = new ActivityWithoutHearbeat("details", TimeSpan.FromSeconds(5));
activity.SetSwfApi(hearbeatApi);
await activity.ExecuteAsync(_activityArgs);
Assert.IsFalse(hearbeatApi.Wait(HeartbeatInterval * 100));
Assert.IsFalse(hearbeatApi.HearbeatRecorded);
}
public async Task Cancellation_token_is_set_when_heartbeat_returns_cancellation_request()
{
var hearbeatApi = new TestHeartbeatSwfApi(() => true, setEventOnCalledTimes: 2);
var activity = new ActivityWithCancellationToken(executionTime: TimeSpan.FromSeconds(1));
activity.SetSwfApi(hearbeatApi);
await activity.ExecuteAsync(_activityArgs);
Assert.IsTrue(hearbeatApi.Wait(WaitPeriod));
Assert.That(activity.CancellationRequested, Is.True);
}
private void StartHearbeat(TestHeartbeatSwfApi heartbeatApi)
{
try
{
_activityHearbeat.StartHeartbeatIfEnabled(heartbeatApi, "token");
Assert.IsTrue(heartbeatApi.Wait(WaitPeriod));
}
finally
{
_activityHearbeat.StopHeartbeat();
}
}