diff options
Diffstat (limited to 'stream_test.go')
| -rw-r--r-- | stream_test.go | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/stream_test.go b/stream_test.go index cf133f1..f911ccc 100644 --- a/stream_test.go +++ b/stream_test.go @@ -99,8 +99,8 @@ func TestStreamWriter(t *testing.T) { // Test unsupported charset file = NewFile() - delete(file.Sheet, "xl/worksheets/sheet1.xml") - file.XLSX["xl/worksheets/sheet1.xml"] = MacintoshCyrillicCharset + file.Sheet.Delete("xl/worksheets/sheet1.xml") + file.Pkg.Store("xl/worksheets/sheet1.xml", MacintoshCyrillicCharset) _, err = file.NewStreamWriter("Sheet1") assert.EqualError(t, err, "xml decode error: XML syntax error on line 1: invalid UTF-8") @@ -145,7 +145,9 @@ func TestStreamTable(t *testing.T) { // Verify the table has names. var table xlsxTable - assert.NoError(t, xml.Unmarshal(file.XLSX["xl/tables/table1.xml"], &table)) + val, ok := file.Pkg.Load("xl/tables/table1.xml") + assert.True(t, ok) + assert.NoError(t, xml.Unmarshal(val.([]byte), &table)) assert.Equal(t, "A", table.TableColumns.TableColumn[0].Name) assert.Equal(t, "B", table.TableColumns.TableColumn[1].Name) assert.Equal(t, "C", table.TableColumns.TableColumn[2].Name) |