Fixed propagation of the token

Author: sakno

src/DotNext.Threading/Threading/Leases/LeaseConsumer.cs

@@ -210,10 +210,15 @@ public async Task<TResult> ExecuteAsync<TResult>(Func<CancellationToken, Task<TR
         {
             throw new AggregateException(exception, e);
         }
-        catch (OperationCanceledException e) when (e.CausedBy(cts, leaseToken))
+        catch (OperationCanceledException e) when (e.CancellationToken == leaseToken ||
+                                                   e.CancellationToken == cts?.Token && cts.CancellationOrigin == leaseToken)
         {
             throw new TimeoutException(ExceptionMessages.LeaseExpired, e);
         }
+        catch (OperationCanceledException e) when (cts is not null)
+        {
+            throw new OperationCanceledException(e.Message, e, cts.CancellationOrigin);
+        }
         finally
         {
             cts?.Dispose();