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>();
        }

    }
}