public void GenerateToObject_WhenCalledWithVariableTableType_UsesCorrectType()
{
IMapping mapping = new Mapping();
Func<Result, BaseScore, bool> canJoin = ( r, b ) => r.Id == b.ResultId;
Action<Result, List<BaseScore>> listSet = ( r, b ) => r.Scores = b;
Func<IDataRecord, Type> variableTableType = dataRecord =>
{
switch ( dataRecord.GetInt32( 4 ) )
{
case 1:
return typeof( IntScore );
default:
return typeof( StringScore );
}
};
mapping.TableJoin<Result, BaseScore>( canJoin, listSet );
mapping.VariableTableType<BaseScore>( variableTableType );
mapping.Table<Result>( 0 );
mapping.Table<BaseScore>( 1 );
var classMapResult = new ClassMapResult
{
Mappings = (Mapping)mapping
};
var mockDataReader = new MockDataReader();
var results = mockDataReader.AddTable( "Id", "Name" );
var scores = mockDataReader.AddTable( "Id", "ResultId", "Value", "ValueString", "ResultType" );
mockDataReader.AddRow( results, 1, "Result 1" );
mockDataReader.AddRow( results, 2, "Result 2" );
mockDataReader.AddRow( scores, 1, 1, 10, "", 1 );
mockDataReader.AddRow( scores, 1, 2, DBNull.Value, "10", 2 );
var func = MappingGenerator.Generate<List<Result>>( new MockDataReader(), classMapResult );
var dictionary = new Dictionary<Tuple<int, Type>, Func<IDataRecord, Dictionary<MappingEnum, object>, object>>();
var result = func( mockDataReader, new Dictionary<MappingEnum, object>
{
{
MappingEnum.TableJoin, new List<TableJoinMap>
{
new TableJoinMap
{
CanJoin = canJoin,
ListSet = listSet
}
}
},
{
MappingEnum.VariableTableType, new Dictionary<Type, object>
{
{ typeof( BaseScore ), variableTableType }
}
}
}, dictionary );
Assert.AreEqual( 1, result[ 0 ].Id );
Assert.AreEqual( 2, result[ 1 ].Id );
Assert.AreEqual( 1, result[ 0 ].Scores.Count );
Assert.AreEqual( 1, result[ 1 ].Scores.Count );
Assert.IsInstanceOf<IntScore>( result[ 0 ].Scores[ 0 ] );
Assert.IsInstanceOf<StringScore>( result[ 1 ].Scores[ 0 ] );
Assert.AreEqual( 10, ( result[ 0 ].Scores[ 0 ] as IntScore ).Value );
Assert.AreEqual( "10", ( result[ 1 ].Scores[ 0 ] as StringScore ).ValueString );
}
public void GenerateToObject_WhenCalledWithVariableTableTypeAndPropertyMap_UsesPropertyMap()
{
IMapping mapping = new Mapping();
Func<IDataRecord, Type> variableTableType = dataRecord => typeof( StringScore );
mapping.VariableTableType<BaseScore>( variableTableType );
Func<IDataRecord, string> action = dataRecord => "The Score Is: " + dataRecord.GetString( 2 );
mapping.PropertyMap<StringScore, String>( x => x.ValueString, action );
mapping.Table<BaseScore>( 0 );
var classMapResult = new ClassMapResult
{
Mappings = (Mapping)mapping
};
var mockDataReader = new MockDataReader();
var scores = mockDataReader.AddTable( "Id", "ResultId", "ValueString", "ResultType" );
mockDataReader.AddRow( scores, 1, 1, "10", 2 );
var func = MappingGenerator.Generate<List<BaseScore>>( new MockDataReader(), classMapResult );
var dictionary = new Dictionary<Tuple<int, Type>, Func<IDataRecord, Dictionary<MappingEnum, object>, object>>();
var result = func( mockDataReader, new Dictionary<MappingEnum, object>
{
{
MappingEnum.VariableTableType, new Dictionary<Type, object>
{
{ typeof( BaseScore ), variableTableType }
}
},
{
MappingEnum.PropertyMap, new Dictionary<PropertyInfo, PropertyMap>
{
{
typeof( StringScore ).GetProperty( "ValueString" ), new PropertyMap( action )
}
}
}
}, dictionary );
Assert.AreEqual( 1, result[ 0 ].Id );
Assert.IsInstanceOf<StringScore>( result[ 0 ] );
Assert.AreEqual( "The Score Is: 10", ( result[ 0 ] as StringScore ).ValueString );
}
