make sure Columns() work with row styles

qax-os · Aug 8, 2022 · dd1fb8b · dd1fb8b
1 parent c8e5b9c
commit dd1fb8b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/rows.go b/rows.go
@@ -183,6 +183,8 @@ func (rows *Rows) Columns(opts ...Options) ([]string, error) {
 		case xml.EndElement:
 			if xmlElement.Name.Local == "sheetData" {
 				return rowIterator.columns, rowIterator.err
+			} else if xmlElement.Name.Local == "row" {
+				return rowIterator.columns, rowIterator.err
 			}
 		}
 	}

diff --git a/rows_test.go b/rows_test.go
@@ -108,12 +108,15 @@ func TestRowsGetRowOpts(t *testing.T) {
 	require.NoError(t, err)
 
 	rows.Next()
+	rows.Columns()
 	got := rows.GetRowOpts()
 	assert.Equal(t, expectedRowStyleID1, got)
 	rows.Next()
+	rows.Columns()
 	got = rows.GetRowOpts()
 	assert.Equal(t, expectedRowStyleID2, got)
 	rows.Next()
+	rows.Columns()
 	got = rows.GetRowOpts()
 	assert.Equal(t, expectedRowStyleID3, got)
 }