@@ -362,22 +362,35 @@ func TestStreamSetRowWithStyle(t *testing.T) {
362362 blueStyleID , err := file .NewStyle (& Style {Font : & Font {Color : "0000FF" }})
363363 assert .NoError (t , err )
364364
365- streamWriter , err := file .NewStreamWriter ("Sheet1" )
365+ sheetName := "Sheet1"
366+ streamWriter , err := file .NewStreamWriter (sheetName )
366367 assert .NoError (t , err )
368+ assert .NoError (t , streamWriter .SetColStyle (1 , 1 , grayStyleID ))
369+ assert .NoError (t , streamWriter .SetColStyle (3 , 3 , blueStyleID ))
367370 assert .NoError (t , streamWriter .SetRow ("A1" , []interface {}{
368- "value1 " ,
369- Cell {Value : "value2 " },
370- & Cell {Value : "value2 " },
371- Cell {StyleID : blueStyleID , Value : "value3 " },
372- & Cell {StyleID : blueStyleID , Value : "value3 " },
371+ "A1 " ,
372+ Cell {Value : "B1 " },
373+ & Cell {Value : "C1 " },
374+ Cell {StyleID : blueStyleID , Value : "D1 " },
375+ & Cell {StyleID : blueStyleID , Value : "E1 " },
373376 }, RowOpts {StyleID : grayStyleID }))
377+ assert .NoError (t , streamWriter .SetRow ("A2" , []interface {}{
378+ "A2" ,
379+ Cell {Value : "B2" },
380+ & Cell {Value : "C2" },
381+ Cell {StyleID : grayStyleID , Value : "D2" },
382+ & Cell {StyleID : blueStyleID , Value : "E2" },
383+ }))
374384 assert .NoError (t , streamWriter .Flush ())
375385
376- ws , err := file .workSheetReader ("Sheet1" )
386+ ws , err := file .workSheetReader (sheetName )
377387 assert .NoError (t , err )
378388 for colIdx , expected := range []int {grayStyleID , grayStyleID , grayStyleID , blueStyleID , blueStyleID } {
379389 assert .Equal (t , expected , ws .SheetData .Row [0 ].C [colIdx ].S )
380390 }
391+ for colIdx , expected := range []int {grayStyleID , 0 , blueStyleID , grayStyleID , blueStyleID } {
392+ assert .Equal (t , expected , ws .SheetData .Row [1 ].C [colIdx ].S )
393+ }
381394}
382395
383396func TestStreamSetCellValFunc (t * testing.T ) {
0 commit comments