Read List and Enum with Dapper

Giorgi · Oct 4, 2023 · 037e20b · 037e20b
1 parent a31dfbc
commit 037e20b
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/DuckDB.NET.Test/DuckDBDataReaderEnumTests.cs b/DuckDB.NET.Test/DuckDBDataReaderEnumTests.cs
@@ -85,7 +85,7 @@ public override void Dispose()
         base.Dispose();
     }
 
-    enum Mood
+    internal enum Mood
     {
         Sad, Ok, Happy
     }

diff --git a/DuckDB.NET.Test/DuckDBDataReaderListTests.cs b/DuckDB.NET.Test/DuckDBDataReaderListTests.cs
@@ -1,6 +1,7 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
+using Dapper;
 using FluentAssertions;
 using Xunit;
 
@@ -154,4 +155,22 @@ public void ReadListOfTimes()
         var list = reader.GetFieldValue<List<TimeOnly>>(0);
         list.Should().BeEquivalentTo(new List<TimeOnly> { new(12, 14, 16), new(18, 10, 12) });
     }
+
+    [Fact]
+    public void ReadListWithDapper()
+    {
+        Command.CommandText = "CREATE TYPE mood AS ENUM ('sad', 'ok', 'happy');";
+        Command.ExecuteNonQuery();
+
+        var person = Connection.QueryFirst<Person>("SELECT [1, 2, 3] as Ids, 'happy' as Mood");
+
+        person.Ids.Should().BeEquivalentTo(new List<int> { 1, 2, 3 });
+        person.Mood.Should().Be(DuckDBDataReaderEnumTests.Mood.Happy);
+    }
+
+    class Person
+    {
+        public List<int> Ids { get; set; }
+        public DuckDBDataReaderEnumTests.Mood Mood { get; set; }
+    }
 }