Skip to content

Add session filter support for DML statement #1931

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 12 commits into from
Dec 14, 2018
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .editorconfig
Original file line number Diff line number Diff line change
Expand Up @@ -30,3 +30,6 @@ indent_size = 2
[*.cshtml]
indent_style = space
indent_size = 4

[*.g]
indent_style = tab
216 changes: 216 additions & 0 deletions src/NHibernate.Test/Async/NHSpecificTest/GH1921/Fixture.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,216 @@
//------------------------------------------------------------------------------
// <auto-generated>
// This code was generated by AsyncGenerator.
//
// Changes to this file may cause incorrect behavior and will be lost if
// the code is regenerated.
// </auto-generated>
//------------------------------------------------------------------------------


using System.Linq;
using NUnit.Framework;
using NHibernate.Linq;

namespace NHibernate.Test.NHSpecificTest.GH1921
{
using System.Threading.Tasks;
using System.Threading;
[TestFixture]
public class FixtureAsync : BugTestCase
{
protected override bool AppliesTo(Dialect.Dialect dialect)
{
return TestDialect.NativeGeneratorSupportsBulkInsertion;
}

protected override void OnSetUp()
{
using (var session = OpenSession())
using (var transaction = session.BeginTransaction())
{
var e1 = new Entity { Name = "Bob" };
session.Save(e1);

var e2 = new Entity { Name = "Sally" };
session.Save(e2);

var me1 = new MultiTableEntity { Name = "Bob", OtherName = "Bob" };
session.Save(me1);

var me2 = new MultiTableEntity { Name = "Sally", OtherName = "Sally" };
session.Save(me2);

transaction.Commit();
}
}

protected override void OnTearDown()
{
using (var session = OpenSession())
using (var transaction = session.BeginTransaction())
{
session.CreateQuery("delete from System.Object").ExecuteUpdate();

transaction.Commit();
}
}

[Theory]
public async Task DmlInsertAsync(bool filtered)
{
using (var session = OpenSession())
using (var transaction = session.BeginTransaction())
{
if (filtered)
session.EnableFilter("NameFilter").SetParameter("name", "Bob");
var rowCount = await (session.CreateQuery("insert into Entity (Name) select e.Name from Entity e")
.ExecuteUpdateAsync());
await (transaction.CommitAsync());

Assert.That(rowCount, Is.EqualTo(filtered ? 1 : 2));
}
}

[Theory]
public async Task DmlUpdateAsync(bool filtered)
{
using (var session = OpenSession())
using (var transaction = session.BeginTransaction())
{
if (filtered)
session.EnableFilter("NameFilter").SetParameter("name", "Bob");
var rowCount = await (session.CreateQuery("update Entity e set Name = 'newName'").ExecuteUpdateAsync());
await (transaction.CommitAsync());

Assert.That(rowCount, Is.EqualTo(filtered ? 1 : 2));
}
}

[Theory]
public async Task DmlDeleteAsync(bool filtered)
{
using (var session = OpenSession())
using (var transaction = session.BeginTransaction())
{
if (filtered)
session.EnableFilter("NameFilter").SetParameter("name", "Bob");
var rowCount = await (session.CreateQuery("delete Entity").ExecuteUpdateAsync());
await (transaction.CommitAsync());

Assert.That(rowCount, Is.EqualTo(filtered ? 1 : 2));
}
}

[TestCase(null)]
[TestCase("NameFilter")]
[TestCase("OtherNameFilter")]
public async Task MultiTableDmlInsertAsync(string filter, CancellationToken cancellationToken = default(CancellationToken))
{
var isFiltered = !string.IsNullOrEmpty(filter);

using (var session = OpenSession())
using (var transaction = session.BeginTransaction())
{
if (isFiltered)
session.EnableFilter(filter).SetParameter("name", "Bob");
var rowCount =
await (session
.CreateQuery(
// No insert of OtherName: not supported (INSERT statements cannot refer to superclass/joined properties)
"insert into MultiTableEntity (Name) select e.Name from MultiTableEntity e")
.ExecuteUpdateAsync(cancellationToken));
await (transaction.CommitAsync(cancellationToken));

Assert.That(rowCount, Is.EqualTo(isFiltered ? 1 : 2), "ResultCount");
}

using (var session = OpenSession())
using (var transaction = session.BeginTransaction())
{
Assert.That(
await (session.Query<MultiTableEntity>().CountAsync(e => e.Name != "Bob", cancellationToken)),
Is.EqualTo(isFiltered ? 1 : 2), "Name != \"Bob\"");
Assert.That(
await (session.Query<MultiTableEntity>().CountAsync(e => e.OtherName == null, cancellationToken)),
Is.EqualTo(isFiltered ? 1 : 2), "OtherName is null");
await (transaction.CommitAsync(cancellationToken));
}
}

[TestCase(null)]
[TestCase("NameFilter")]
[TestCase("OtherNameFilter")]
public async Task MultiTableDmlUpdateAsync(string filter, CancellationToken cancellationToken = default(CancellationToken))
{
var isFiltered = !string.IsNullOrEmpty(filter);

using (var session = OpenSession())
using (var transaction = session.BeginTransaction())
{
if (isFiltered)
session.EnableFilter(filter).SetParameter("name", "Bob");
var rowCount =
await (session
.CreateQuery(
"update MultiTableEntity e" +
" set Name = 'newName', OtherName = 'newOtherName'" +
// Check referencing columns is supported
" where e.Name is not null and e.OtherName is not null")
.ExecuteUpdateAsync(cancellationToken));
await (transaction.CommitAsync(cancellationToken));

Assert.That(rowCount, Is.EqualTo(isFiltered ? 1 : 2), "ResultCount");
}

using (var session = OpenSession())
using (var transaction = session.BeginTransaction())
{
Assert.That(
await (session.Query<MultiTableEntity>().CountAsync(e => e.Name == "newName", cancellationToken)),
Is.EqualTo(isFiltered ? 1 : 2), "Name == \"newName\"");
Assert.That(
await (session.Query<MultiTableEntity>().CountAsync(e => e.OtherName == "newOtherName", cancellationToken)),
Is.EqualTo(isFiltered ? 1 : 2), "Name == \"newOtherName\"");
await (transaction.CommitAsync(cancellationToken));
}
}

[TestCase(null)]
[TestCase("NameFilter")]
[TestCase("OtherNameFilter")]
public async Task MultiTableDmlDeleteAsync(string filter, CancellationToken cancellationToken = default(CancellationToken))
{
var isFiltered = !string.IsNullOrEmpty(filter);

using (var session = OpenSession())
using (var transaction = session.BeginTransaction())
{
if (isFiltered)
session.EnableFilter(filter).SetParameter("name", "Bob");
var rowCount =
await (session
.CreateQuery(
"delete MultiTableEntity e" +
// Check referencing columns is supported
" where e.Name is not null and e.OtherName is not null")
.ExecuteUpdateAsync(cancellationToken));
await (transaction.CommitAsync(cancellationToken));

Assert.That(rowCount, Is.EqualTo(isFiltered ? 1 : 2), "ResultCount");
}

using (var session = OpenSession())
using (var transaction = session.BeginTransaction())
{
Assert.That(
await (session.Query<MultiTableEntity>().CountAsync(e => e.Name != "Bob", cancellationToken)),
Is.EqualTo(isFiltered ? 1 : 0), "Name != \"Bob\"");
Assert.That(
await (session.Query<MultiTableEntity>().CountAsync(e => e.OtherName != "Bob", cancellationToken)),
Is.EqualTo(isFiltered ? 1 : 0), "OtherName != \"Bob\"");
await (transaction.CommitAsync(cancellationToken));
}
}
}
}
17 changes: 17 additions & 0 deletions src/NHibernate.Test/NHSpecificTest/GH1921/Entity.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
using System;

namespace NHibernate.Test.NHSpecificTest.GH1921
{
class Entity
{
public virtual int Id { get; set; }
public virtual string Name { get; set; }
}

class MultiTableEntity
{
public virtual int Id { get; set; }
public virtual string Name { get; set; }
public virtual string OtherName { get; set; }
}
}
Loading