Dapper.ColumnMapper 的使用

using System;

using System.Collections.Generic;

using System.Data;

using System.Data.SqlClient;

using System.Linq;

using System.Text;

using System.Threading.Tasks;

using NUnit.Framework;

using FluentAssertions;



namespace Dapper.ColumnMapper.Tests

{

    [TestFixture]

    public class ColumnMapperTests

    {

        private SqlConnection connection;



        public static readonly string connectionString = "Data Source=.\\SQLEXPRESS;Initial Catalog=tempdb;Integrated Security=True";



        public class ColumnMappingObject

        {

            public string DefaultColumn { get; set; }

            public string NonMatchingColumn { get; set; }

 [ColumnMapping("MappedCol")]
            public string MappedColumn { get; set; } public string MiscasedColumn { get; set; }

        }



        [TestFixtureSetUp]

        public void Setup()

        {

            connection = new SqlConnection(connectionString);

            connection.Open();



            CreateTestTable();



            SqlMapper.SetTypeMap(typeof(ColumnMappingObject), new ColumnTypeMapper(typeof(ColumnMappingObject)));

        }



        private void CreateTestTable()

        {

            const string createSql = @"

                create table #Test (Id int, DefaultColumn varchar(20), BadCol varchar(20), MappedCol varchar(20), miscasedColumn varchar(20))

                insert #Test values(1, 'DefaultColumn1', 'BadColumn', 'MappedColumn1', 'MiscasedColumn1')";



            connection.Execute(createSql);

        }



        [TestFixtureTearDown]

        public void TearDown()

        {

            if (connection != null && connection.State == ConnectionState.Open)

            {

                const string dropSql = "drop table #Test";

                connection.Execute(dropSql);



                connection.Close();

            }



            connection = null;

        }



        [Test]

        public void Property_Matching_Column_Name_Exactly_Is_Mapped_Correctly()

        {

            var selectedObject = SelectObjects();



            selectedObject.Should().NotBeNull();

            selectedObject.DefaultColumn.Should().NotBeNullOrEmpty();

            selectedObject.DefaultColumn.Should().Be("DefaultColumn1");

        }



        [Test]

        public void Property_Not_Matching_Coumn_Name_Exactly_And_No_Attribute_Is_Not_Mapped()

        {

            var selectedObject = SelectObjects();



            selectedObject.Should().NotBeNull();

            selectedObject.NonMatchingColumn.Should().BeNullOrEmpty();

        }



        [Test]

        public void Property_Using_Column_Mapping_Attribute_Is_Mapped_Correctly()

        {

            var selectedObject = SelectObjects();



            selectedObject.Should().NotBeNull();

            selectedObject.MappedColumn.Should().NotBeNullOrEmpty();

            selectedObject.MappedColumn.Should().Be("MappedColumn1");

        }



        [Test]

        public void Miscased_Property_Is_Mapped_Correctly()

        {

            var selectedObject = SelectObjects();



            selectedObject.Should().NotBeNull();

            selectedObject.MiscasedColumn.Should().NotBeNullOrEmpty();

            selectedObject.MiscasedColumn.Should().Be("MiscasedColumn1");



        }



        private ColumnMappingObject SelectObjects()

        {

            const string selectSql = "select * from #Test";



            return connection.Query<ColumnMappingObject>(selectSql).FirstOrDefault();

        }



        [Test]

        public void Can_Register_Single_Type()

        {

            ColumnTypeMapper.RegisterForTypes(typeof (DateTime));



            var map = SqlMapper.GetTypeMap(typeof(DateTime));



            map.Should().NotBeNull();

            map.Should().BeOfType<ColumnTypeMapper>();

        }



        [Test]

        public void Can_Register_Multiple_Types()

        {

            ColumnTypeMapper.RegisterForTypes(typeof(DateTime), typeof(TimeSpan));



            var dateTimeMap = SqlMapper.GetTypeMap(typeof (DateTime));



            dateTimeMap.Should().NotBeNull();

            dateTimeMap.Should().BeOfType<ColumnTypeMapper>();



            var timeSpanMap = SqlMapper.GetTypeMap(typeof (TimeSpan));



            timeSpanMap.Should().NotBeNull();

            timeSpanMap.Should().BeOfType<ColumnTypeMapper>();

        }



    }

}