mirror of https://github.com/Squidex/squidex.git
Sebastian Stehle
4 years ago
committed by
GitHub
GitHub
11 changed files with 158 additions and 401 deletions
@ -1,168 +0,0 @@ |
|||
// ==========================================================================
|
|||
// Squidex Headless CMS
|
|||
// ==========================================================================
|
|||
// Copyright (c) Squidex UG (haftungsbeschraenkt)
|
|||
// All rights reserved. Licensed under the MIT license.
|
|||
// ==========================================================================
|
|||
|
|||
using System; |
|||
using System.Collections.Generic; |
|||
using System.Linq; |
|||
using System.Threading; |
|||
using System.Threading.Tasks; |
|||
using MongoDB.Driver; |
|||
using Xunit; |
|||
|
|||
namespace Squidex.Infrastructure.MongoDb |
|||
{ |
|||
public class MongoExtensionsTests |
|||
{ |
|||
public sealed class Cursor<T> : IAsyncCursor<T> where T : notnull |
|||
{ |
|||
private readonly List<object> items = new List<object>(); |
|||
private int index = -1; |
|||
|
|||
public IEnumerable<T> Current |
|||
{ |
|||
get |
|||
{ |
|||
if (items[index] is Exception ex) |
|||
{ |
|||
throw ex; |
|||
} |
|||
|
|||
return Enumerable.Repeat((T)items[index], 1); |
|||
} |
|||
} |
|||
|
|||
public Cursor<T> Add(params T[] newItems) |
|||
{ |
|||
foreach (var item in newItems) |
|||
{ |
|||
items.Add(item); |
|||
} |
|||
|
|||
return this; |
|||
} |
|||
|
|||
public Cursor<T> Add(Exception ex) |
|||
{ |
|||
items.Add(ex); |
|||
|
|||
return this; |
|||
} |
|||
|
|||
public void Dispose() |
|||
{ |
|||
} |
|||
|
|||
public bool MoveNext(CancellationToken cancellationToken = default) |
|||
{ |
|||
index++; |
|||
|
|||
return index < items.Count; |
|||
} |
|||
|
|||
public async Task<bool> MoveNextAsync(CancellationToken cancellationToken = default) |
|||
{ |
|||
await Task.Delay(1, cancellationToken); |
|||
|
|||
return MoveNext(cancellationToken); |
|||
} |
|||
} |
|||
|
|||
[Fact] |
|||
public async Task Should_enumerate_over_items() |
|||
{ |
|||
var result = new List<int>(); |
|||
|
|||
var cursor = new Cursor<int>().Add(0, 1, 2, 3, 4, 5); |
|||
|
|||
await cursor.ForEachPipedAsync(x => |
|||
{ |
|||
result.Add(x); |
|||
return Task.CompletedTask; |
|||
}); |
|||
|
|||
Assert.Equal(new List<int> { 0, 1, 2, 3, 4, 5 }, result); |
|||
} |
|||
|
|||
[Fact] |
|||
public async Task Should_break_if_cursor_failed() |
|||
{ |
|||
var ex = new InvalidOperationException(); |
|||
|
|||
var result = new List<int>(); |
|||
|
|||
using (var cursor = new Cursor<int>().Add(0, 1, 2).Add(ex).Add(3, 4, 5)) |
|||
{ |
|||
await Assert.ThrowsAsync<InvalidOperationException>(() => |
|||
{ |
|||
return cursor.ForEachPipedAsync(x => |
|||
{ |
|||
result.Add(x); |
|||
return Task.CompletedTask; |
|||
}); |
|||
}); |
|||
} |
|||
|
|||
Assert.Equal(new List<int> { 0, 1, 2 }, result); |
|||
} |
|||
|
|||
[Fact] |
|||
public async Task Should_break_if_handler_failed() |
|||
{ |
|||
var ex = new InvalidOperationException(); |
|||
|
|||
var result = new List<int>(); |
|||
|
|||
using (var cursor = new Cursor<int>().Add(0, 1, 2, 3, 4, 5)) |
|||
{ |
|||
await Assert.ThrowsAsync<InvalidOperationException>(() => |
|||
{ |
|||
return cursor.ForEachPipedAsync(x => |
|||
{ |
|||
if (x == 2) |
|||
{ |
|||
throw ex; |
|||
} |
|||
|
|||
result.Add(x); |
|||
return Task.CompletedTask; |
|||
}); |
|||
}); |
|||
} |
|||
|
|||
Assert.Equal(new List<int> { 0, 1 }, result); |
|||
} |
|||
|
|||
[Fact] |
|||
public async Task Should_stop_if_cancelled1() |
|||
{ |
|||
using (var cts = new CancellationTokenSource()) |
|||
{ |
|||
var result = new List<int>(); |
|||
|
|||
using (var cursor = new Cursor<int>().Add(0, 1, 2, 3, 4, 5)) |
|||
{ |
|||
await Assert.ThrowsAnyAsync<OperationCanceledException>(() => |
|||
{ |
|||
return cursor.ForEachPipedAsync(x => |
|||
{ |
|||
if (x == 2) |
|||
{ |
|||
cts.Cancel(); |
|||
} |
|||
|
|||
result.Add(x); |
|||
|
|||
return Task.CompletedTask; |
|||
}, cts.Token); |
|||
}); |
|||
} |
|||
|
|||
Assert.Equal(new List<int> { 0, 1, 2 }, result); |
|||
} |
|||
} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue