Skip to content

Commit

Permalink
Fix value generation with converter between different numeric types (#…
Browse files Browse the repository at this point in the history 
…29601)
  • Loading branch information
@ajcvickers
ajcvickers authored Nov 19, 2022
1 parent e64fda2 commit 3579d25
Show file tree
Hide file tree
Showing 3 changed files with 377 additions and 55 deletions.
131 changes: 76 additions & 55 deletions src/EFCore/ValueGeneration/TemporaryNumberValueGeneratorFactory.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -27,70 +27,91 @@ public class TemporaryNumberValueGeneratorFactory : ValueGeneratorFactory
/// <returns>The newly created value generator.</returns>
public override ValueGenerator Create(IProperty property, IEntityType entityType)
{
var type = (property.GetValueConverter()?.ProviderClrType ?? property.GetTypeMapping().ClrType).UnwrapEnumType();
var type = property.GetTypeMapping().ClrType.UnwrapEnumType();

if (type == typeof(int))
var generator = TryCreate();
if (generator != null)
{
return new TemporaryIntValueGenerator();
return generator;
}

if (type == typeof(long))
type = property.GetValueConverter()?.ProviderClrType.UnwrapEnumType();
if (type != null)
{
return new TemporaryLongValueGenerator();
}

if (type == typeof(short))
{
return new TemporaryShortValueGenerator();
}

if (type == typeof(byte))
{
return new TemporaryByteValueGenerator();
}

if (type == typeof(char))
{
return new TemporaryCharValueGenerator();
}

if (type == typeof(ulong))
{
return new TemporaryULongValueGenerator();
}

if (type == typeof(uint))
{
return new TemporaryUIntValueGenerator();
}

if (type == typeof(ushort))
{
return new TemporaryUShortValueGenerator();
}

if (type == typeof(sbyte))
{
return new TemporarySByteValueGenerator();
}

if (type == typeof(decimal))
{
return new TemporaryDecimalValueGenerator();
}

if (type == typeof(float))
{
return new TemporaryFloatValueGenerator();
}

if (type == typeof(double))
{
return new TemporaryDoubleValueGenerator();
generator = TryCreate();
if (generator != null)
{
return generator;
}
}

throw new ArgumentException(
CoreStrings.InvalidValueGeneratorFactoryProperty(
nameof(TemporaryNumberValueGeneratorFactory), property.Name, property.DeclaringEntityType.DisplayName()));

ValueGenerator? TryCreate()
{
if (type == typeof(int))
{
return new TemporaryIntValueGenerator();
}

if (type == typeof(long))
{
return new TemporaryLongValueGenerator();
}

if (type == typeof(short))
{
return new TemporaryShortValueGenerator();
}

if (type == typeof(byte))
{
return new TemporaryByteValueGenerator();
}

if (type == typeof(char))
{
return new TemporaryCharValueGenerator();
}

if (type == typeof(ulong))
{
return new TemporaryULongValueGenerator();
}

if (type == typeof(uint))
{
return new TemporaryUIntValueGenerator();
}

if (type == typeof(ushort))
{
return new TemporaryUShortValueGenerator();
}

if (type == typeof(sbyte))
{
return new TemporarySByteValueGenerator();
}

if (type == typeof(decimal))
{
return new TemporaryDecimalValueGenerator();
}

if (type == typeof(float))
{
return new TemporaryFloatValueGenerator();
}

if (type == typeof(double))
{
return new TemporaryDoubleValueGenerator();
}

return null;
}
}
}
149 changes: 149 additions & 0 deletions test/EFCore.Specification.Tests/StoreGeneratedTestBase.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2198,6 +2198,145 @@ public virtual void Insert_update_and_delete_with_wrapped_int_key()
});
}

protected class LongToIntPrincipal
{
[DatabaseGenerated(DatabaseGeneratedOption.Identity)]
public long Id { get; set; }

public ICollection<LongToIntDependentShadow> Dependents { get; } = new List<LongToIntDependentShadow>();
public ICollection<LongToIntDependentRequired> RequiredDependents { get; } = new List<LongToIntDependentRequired>();
public ICollection<LongToIntDependentOptional> OptionalDependents { get; } = new List<LongToIntDependentOptional>();
}

protected class LongToIntDependentShadow
{
[DatabaseGenerated(DatabaseGeneratedOption.Identity)]
public long Id { get; set; }

public LongToIntPrincipal? Principal { get; set; }
}

protected class LongToIntDependentRequired
{
[DatabaseGenerated(DatabaseGeneratedOption.Identity)]
public long Id { get; set; }

public long PrincipalId { get; set; }
public LongToIntPrincipal Principal { get; set; } = null!;
}

protected class LongToIntDependentOptional
{
[DatabaseGenerated(DatabaseGeneratedOption.Identity)]
public long Id { get; set; }

public long? PrincipalId { get; set; }
public LongToIntPrincipal? Principal { get; set; }
}

[ConditionalFact]
public virtual void Insert_update_and_delete_with_long_to_int_conversion()
{
var id1 = 0L;
ExecuteWithStrategyInTransaction(
context =>
{
var principal1 = context.Add(
new LongToIntPrincipal
{
Dependents = { new LongToIntDependentShadow(), new LongToIntDependentShadow() },
OptionalDependents = { new LongToIntDependentOptional(), new LongToIntDependentOptional() },
RequiredDependents = { new LongToIntDependentRequired(), new LongToIntDependentRequired() }
}).Entity;

context.SaveChanges();

id1 = principal1.Id;
Assert.NotEqual(0L, id1);
foreach (var dependent in principal1.Dependents)
{
Assert.NotEqual(0L, dependent.Id);
Assert.Same(principal1, dependent.Principal);
Assert.Equal(id1, context.Entry(dependent).Property<long?>("PrincipalId").CurrentValue!.Value);
}

foreach (var dependent in principal1.OptionalDependents)
{
Assert.NotEqual(0L, dependent.Id);
Assert.Same(principal1, dependent.Principal);
Assert.Equal(id1, dependent.PrincipalId);
}

foreach (var dependent in principal1.RequiredDependents)
{
Assert.NotEqual(0L, dependent.Id);
Assert.Same(principal1, dependent.Principal);
Assert.Equal(id1, dependent.PrincipalId);
}
},
context =>
{
var principal1 = context.Set<LongToIntPrincipal>()
.Include(e => e.Dependents)
.Include(e => e.OptionalDependents)
.Include(e => e.RequiredDependents)
.Single();

Assert.Equal(principal1.Id, id1);
foreach (var dependent in principal1.Dependents)
{
Assert.Same(principal1, dependent.Principal);
Assert.Equal(id1, context.Entry(dependent).Property<long?>("PrincipalId").CurrentValue!.Value);
}

foreach (var dependent in principal1.OptionalDependents)
{
Assert.Same(principal1, dependent.Principal);
Assert.Equal(id1, dependent.PrincipalId!.Value);
}

foreach (var dependent in principal1.RequiredDependents)
{
Assert.Same(principal1, dependent.Principal);
Assert.Equal(id1, dependent.PrincipalId);
}

principal1.Dependents.Remove(principal1.Dependents.First());
principal1.OptionalDependents.Remove(principal1.OptionalDependents.First());
principal1.RequiredDependents.Remove(principal1.RequiredDependents.First());

context.SaveChanges();
},
context =>
{
var dependents1 = context.Set<LongToIntDependentShadow>().Include(e => e.Principal).ToList();
Assert.Equal(2, dependents1.Count);
Assert.Null(
context.Entry(dependents1.Single(e => e.Principal == null))
.Property<long?>("PrincipalId").CurrentValue);

var optionalDependents1 = context.Set<LongToIntDependentOptional>().Include(e => e.Principal).ToList();
Assert.Equal(2, optionalDependents1.Count);
Assert.Null(optionalDependents1.Single(e => e.Principal == null).PrincipalId);

var requiredDependents1 = context.Set<LongToIntDependentRequired>().Include(e => e.Principal).ToList();
Assert.Single(requiredDependents1);

context.Remove(dependents1.Single(e => e.Principal != null));
context.Remove(optionalDependents1.Single(e => e.Principal != null));
context.Remove(requiredDependents1.Single());
context.Remove(requiredDependents1.Single().Principal);

context.SaveChanges();
},
context =>
{
Assert.Equal(1, context.Set<LongToIntDependentShadow>().Count());
Assert.Equal(1, context.Set<LongToIntDependentOptional>().Count());
Assert.Equal(0, context.Set<LongToIntDependentRequired>().Count());
});
}

protected class WrappedStringClass
{
public string? Value { get; set; }
Expand Down Expand Up @@ -4579,6 +4718,16 @@ protected override void OnModelCreating(ModelBuilder modelBuilder, DbContext con
entity.Property(e => e.NonKey).HasValueGenerator<WrappedIntRecordValueGenerator>();
});

modelBuilder.Entity<LongToIntPrincipal>(
entity =>
{
var keyConverter = new ValueConverter<long, int>(
v => (int)v,
v => (long)v);

entity.Property(e => e.Id).HasConversion(keyConverter);
});

modelBuilder.Entity<WrappedGuidClassPrincipal>(
entity =>
{
Expand Down
Loading

0 comments on commit 3579d25

Please sign in to comment.