Skip to content

Commit ec0b25a

Browse files
committed
Added tests for ToArrayAsync and ToListAsync
1 parent e889db9 commit ec0b25a

File tree

1 file changed

+36
-0
lines changed

1 file changed

+36
-0
lines changed

tests/MongoFramework.Tests/Linq/QueryableAsyncExtensionsTests.cs

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,42 @@ public async Task AsyncEnumeration()
4949
}
5050
}
5151

52+
[TestMethod]
53+
public async Task ToArrayAsync()
54+
{
55+
EntityMapping.RegisterType(typeof(MongoFrameworkQueryableModel));
56+
57+
var connection = TestConfiguration.GetConnection();
58+
var context = new MongoDbContext(connection);
59+
var provider = new MongoFrameworkQueryProvider<MongoFrameworkQueryableModel>(connection);
60+
var queryable = new MongoFrameworkQueryable<MongoFrameworkQueryableModel>(provider);
61+
62+
context.ChangeTracker.SetEntityState(new MongoFrameworkQueryableModel { Title = "ToArrayAsync" }, EntityEntryState.Added);
63+
context.SaveChanges();
64+
65+
var result = await queryable.ToArrayAsync();
66+
Assert.AreEqual(1, result.Length);
67+
Assert.AreEqual("ToArrayAsync", result[0].Title);
68+
}
69+
70+
[TestMethod]
71+
public async Task ToListAsync()
72+
{
73+
EntityMapping.RegisterType(typeof(MongoFrameworkQueryableModel));
74+
75+
var connection = TestConfiguration.GetConnection();
76+
var context = new MongoDbContext(connection);
77+
var provider = new MongoFrameworkQueryProvider<MongoFrameworkQueryableModel>(connection);
78+
var queryable = new MongoFrameworkQueryable<MongoFrameworkQueryableModel>(provider);
79+
80+
context.ChangeTracker.SetEntityState(new MongoFrameworkQueryableModel { Title = "ToListAsync" }, EntityEntryState.Added);
81+
context.SaveChanges();
82+
83+
var result = await queryable.ToListAsync();
84+
Assert.AreEqual(1, result.Count);
85+
Assert.AreEqual("ToListAsync", result[0].Title);
86+
}
87+
5288
[TestMethod]
5389
public async Task FirstAsync_NoValue()
5490
{

0 commit comments

Comments
 (0)