1
0
mirror of https://github.com/bitwarden/server.git synced 2024-12-05 14:23:30 +01:00
bitwarden-server/test/Infrastructure.Dapper.Test/DataTableBuilderTests.cs

181 lines
5.8 KiB
C#
Raw Normal View History

using System.Data;
namespace Bit.Infrastructure.Dapper.Test;
public class DataTableBuilderTests
{
public class TestItem
{
// Normal value type
public int Id { get; set; }
// Normal reference type
public string? Name { get; set; }
// Nullable value type
public DateTime? DeletedDate { get; set; }
public object? ObjectProp { get; set; }
public DefaultEnum DefaultEnum { get; set; }
public DefaultEnum? NullableDefaultEnum { get; set; }
public ByteEnum ByteEnum { get; set; }
public ByteEnum? NullableByteEnum { get; set; }
public int Method()
{
throw new NotImplementedException();
}
}
public enum DefaultEnum
{
Zero,
One,
}
public enum ByteEnum : byte
{
Zero,
One,
}
[Fact]
public void DataTableBuilder_Works()
{
var dtb = new DataTableBuilder<TestItem>(
[
i => i.Id,
i => i.Name,
i => i.DeletedDate,
i => i.ObjectProp,
i => i.DefaultEnum,
i => i.NullableDefaultEnum,
i => i.ByteEnum,
i => i.NullableByteEnum,
]
);
var table = dtb.Build(
[
new TestItem
{
Id = 4,
Name = "Test",
DeletedDate = new DateTime(2024, 8, 8),
ObjectProp = 1,
DefaultEnum = DefaultEnum.One,
NullableDefaultEnum = DefaultEnum.Zero,
ByteEnum = ByteEnum.One,
NullableByteEnum = ByteEnum.Zero,
},
new TestItem
{
Id = int.MaxValue,
Name = null,
DeletedDate = null,
ObjectProp = "Hi",
DefaultEnum = DefaultEnum.Zero,
NullableDefaultEnum = null,
ByteEnum = ByteEnum.Zero,
NullableByteEnum = null,
},
]
);
Assert.Collection(
table.Columns.Cast<DataColumn>(),
column =>
{
Assert.Equal("Id", column.ColumnName);
Assert.Equal(typeof(int), column.DataType);
},
column =>
{
Assert.Equal("Name", column.ColumnName);
Assert.Equal(typeof(string), column.DataType);
},
column =>
{
Assert.Equal("DeletedDate", column.ColumnName);
// Checking that it will unwrap the `Nullable<T>`
Assert.Equal(typeof(DateTime), column.DataType);
},
column =>
{
Assert.Equal("ObjectProp", column.ColumnName);
Assert.Equal(typeof(object), column.DataType);
},
column =>
{
Assert.Equal("DefaultEnum", column.ColumnName);
Assert.Equal(typeof(int), column.DataType);
},
column =>
{
Assert.Equal("NullableDefaultEnum", column.ColumnName);
Assert.Equal(typeof(int), column.DataType);
},
column =>
{
Assert.Equal("ByteEnum", column.ColumnName);
Assert.Equal(typeof(byte), column.DataType);
},
column =>
{
Assert.Equal("NullableByteEnum", column.ColumnName);
Assert.Equal(typeof(byte), column.DataType);
}
);
Assert.Collection(
table.Rows.Cast<DataRow>(),
row =>
{
Assert.Collection(
row.ItemArray,
item => Assert.Equal(4, item),
item => Assert.Equal("Test", item),
item => Assert.Equal(new DateTime(2024, 8, 8), item),
item => Assert.Equal(1, item),
item => Assert.Equal((int)DefaultEnum.One, item),
item => Assert.Equal((int)DefaultEnum.Zero, item),
item => Assert.Equal((byte)ByteEnum.One, item),
item => Assert.Equal((byte)ByteEnum.Zero, item)
);
},
row =>
{
Assert.Collection(
row.ItemArray,
item => Assert.Equal(int.MaxValue, item),
item => Assert.Equal(DBNull.Value, item),
item => Assert.Equal(DBNull.Value, item),
item => Assert.Equal("Hi", item),
item => Assert.Equal((int)DefaultEnum.Zero, item),
item => Assert.Equal(DBNull.Value, item),
item => Assert.Equal((byte)ByteEnum.Zero, item),
item => Assert.Equal(DBNull.Value, item)
);
}
);
}
[Fact]
public void DataTableBuilder_ThrowsOnInvalidExpression()
{
var argException = Assert.Throws<ArgumentException>(() => new DataTableBuilder<TestItem>([i => i.Method()]));
Assert.Equal(
"Could not determine the property info from the given expression 'i => Convert(i.Method(), Object)'.",
argException.Message
);
}
[Fact]
public void DataTableBuilder_ThrowsOnRepeatExpression()
{
var argException = Assert.Throws<ArgumentException>(() => new DataTableBuilder<TestItem>([i => i.Id, i => i.Id]));
Assert.Equal(
"Property with name 'Id' was already added, properties can only be added once.",
argException.Message
);
}
}