Correct error writing struct (#961)

* Correct error writing struct

* Update src/write.jl

* Update src/write.jl

* Update test/write.jl

* Delete jcunwin.jl

Co-authored-by: John Unwin <[email protected]>
Co-authored-by: Jacob Quinn <[email protected]>

Author: 3 people

src/write.jl

@@ -261,8 +261,8 @@ function write(::Nothing, rows, file, opts;
         return file
     end
     row, st = state
-    names = header isa Bool || isempty(header) ? propertynames(row) : header
-    sch = Tables.Schema(names, nothing)
+    names = header isa Bool || isempty(header) ? Tables.columnnames(row) : header
+    sch = Tables.Schema(Tables.columnnames(row), nothing)
     cols = length(names)
     with(file, append, compress) do io
         if !append && opts.bom

test/write.jl

@@ -17,6 +17,13 @@ const table_types = (
     col8=weakrefs,
 )
 
+struct StructType
+    adate::Date
+    astring::Union{String, Nothing}
+    aumber::Union{Real, Nothing}
+end
+
+
 @testset "CSV.write" begin
 
     testcases = [
@@ -172,6 +179,12 @@ const table_types = (
             Any[(a=1,)],
             NamedTuple(),
             "a\n1\n"
+        ),
+        # jcunwin
+        (
+            [StructType(Date("2021-12-01"), "string 1", 123.45), StructType(Date("2021-12-02"), "string 2", 456.78)],
+            (header=["Date Column", "String Column", "Number Column"],),
+            "Date Column,String Column,Number Column\n2021-12-01,string 1,123.45\n2021-12-02,string 2,456.78\n"
         )
     ]