##// END OF EJS Templates
Bound promise to CancellationToken...
Bound promise to CancellationToken Added new states to ExecutionSate enum. Added Safe.Guard() method to handle cleanup of the result of the promise

File last commit:

r205:8200ab154c8a v2
r209:a867536c68fc v2
Show More
MockPollingComponent.cs
71 lines | 1.9 KiB | text/x-csharp | CSharpLexer
/ Implab.Test / Mock / MockPollingComponent.cs
using System;
using Implab.Components;
namespace Implab.Test.Mock {
class MockPollingComponent : PollingComponent {
public MockPollingComponent(TimeSpan interval, Func<Func<ICancellationToken, IPromise>, IPromise> dispatcher, bool initialized) : base(interval, dispatcher, initialized) {
}
public Action MockInit {
get;
set;
}
public Action<Exception> MockOnError {
get;
set;
}
public Action<Exception> MockOnCancel {
get;
set;
}
public Func<IPromise> MockStart {
get;
set;
}
public Func<IPromise> MockStop {
get;
set;
}
public Func<ICancellationToken, IPromise> MockTick {
get;
set;
}
protected override IPromise OnStart() {
return MockStart != null ? Safe.Run(MockStart).Chain(base.OnStart) : Safe.Run(base.OnStart);
}
protected override IPromise OnStop() {
return MockStop != null ? Safe.Run(MockStop).Chain(base.OnStop) : Safe.Run(base.OnStop);
}
protected override void OnInitialize() {
if (MockInit != null)
MockInit();
}
protected override IPromise OnTick(ICancellationToken cancellationToken) {
return MockTick != null ? Safe.Run(() => MockTick(cancellationToken)) : Promise.Success;
}
protected override void OnTickCancel(Exception error) {
if (MockOnCancel != null)
MockOnCancel(error);
}
protected override void OnTickError(Exception error) {
if (MockOnError != null)
MockOnError(error);
}
public void CallComponentFail(Exception error) {
Fail(error);
}
}
}