rows_test.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784
  1. package excelize
  2. import (
  3. "bytes"
  4. "fmt"
  5. "path/filepath"
  6. "testing"
  7. "github.com/stretchr/testify/assert"
  8. "github.com/stretchr/testify/require"
  9. )
  10. func TestRows(t *testing.T) {
  11. const sheet2 = "Sheet2"
  12. f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
  13. if !assert.NoError(t, err) {
  14. t.FailNow()
  15. }
  16. rows, err := f.Rows(sheet2)
  17. if !assert.NoError(t, err) {
  18. t.FailNow()
  19. }
  20. var collectedRows [][]string
  21. for rows.Next() {
  22. columns, err := rows.Columns()
  23. assert.NoError(t, err)
  24. collectedRows = append(collectedRows, trimSliceSpace(columns))
  25. }
  26. if !assert.NoError(t, rows.Error()) {
  27. t.FailNow()
  28. }
  29. returnedRows, err := f.GetRows(sheet2)
  30. assert.NoError(t, err)
  31. for i := range returnedRows {
  32. returnedRows[i] = trimSliceSpace(returnedRows[i])
  33. }
  34. if !assert.Equal(t, collectedRows, returnedRows) {
  35. t.FailNow()
  36. }
  37. f = NewFile()
  38. f.XLSX["xl/worksheets/sheet1.xml"] = []byte(`<worksheet><sheetData><row r="1"><c r="A1" t="s"><v>1</v></c></row><row r="A"><c r="2" t="str"><v>B</v></c></row></sheetData></worksheet>`)
  39. _, err = f.Rows("Sheet1")
  40. assert.EqualError(t, err, `strconv.Atoi: parsing "A": invalid syntax`)
  41. }
  42. func TestRowsIterator(t *testing.T) {
  43. const (
  44. sheet2 = "Sheet2"
  45. expectedNumRow = 11
  46. )
  47. f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
  48. require.NoError(t, err)
  49. rows, err := f.Rows(sheet2)
  50. require.NoError(t, err)
  51. var rowCount int
  52. for rows.Next() {
  53. rowCount++
  54. require.True(t, rowCount <= expectedNumRow, "rowCount is greater than expected")
  55. }
  56. assert.Equal(t, expectedNumRow, rowCount)
  57. // Valued cell sparse distribution test
  58. f = NewFile()
  59. cells := []string{"C1", "E1", "A3", "B3", "C3", "D3", "E3"}
  60. for _, cell := range cells {
  61. assert.NoError(t, f.SetCellValue("Sheet1", cell, 1))
  62. }
  63. rows, err = f.Rows("Sheet1")
  64. require.NoError(t, err)
  65. rowCount = 0
  66. for rows.Next() {
  67. rowCount++
  68. require.True(t, rowCount <= 3, "rowCount is greater than expected")
  69. }
  70. assert.Equal(t, 3, rowCount)
  71. }
  72. func TestRowsError(t *testing.T) {
  73. xlsx, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
  74. if !assert.NoError(t, err) {
  75. t.FailNow()
  76. }
  77. _, err = xlsx.Rows("SheetN")
  78. assert.EqualError(t, err, "sheet SheetN is not exist")
  79. }
  80. func TestRowHeight(t *testing.T) {
  81. xlsx := NewFile()
  82. sheet1 := xlsx.GetSheetName(1)
  83. assert.EqualError(t, xlsx.SetRowHeight(sheet1, 0, defaultRowHeightPixels+1.0), "invalid row number 0")
  84. _, err := xlsx.GetRowHeight("Sheet1", 0)
  85. assert.EqualError(t, err, "invalid row number 0")
  86. assert.NoError(t, xlsx.SetRowHeight(sheet1, 1, 111.0))
  87. height, err := xlsx.GetRowHeight(sheet1, 1)
  88. assert.NoError(t, err)
  89. assert.Equal(t, 111.0, height)
  90. assert.NoError(t, xlsx.SetRowHeight(sheet1, 4, 444.0))
  91. height, err = xlsx.GetRowHeight(sheet1, 4)
  92. assert.NoError(t, err)
  93. assert.Equal(t, 444.0, height)
  94. // Test get row height that rows index over exists rows.
  95. height, err = xlsx.GetRowHeight(sheet1, 5)
  96. assert.NoError(t, err)
  97. assert.Equal(t, defaultRowHeightPixels, height)
  98. // Test get row height that rows heights haven't changed.
  99. height, err = xlsx.GetRowHeight(sheet1, 3)
  100. assert.NoError(t, err)
  101. assert.Equal(t, defaultRowHeightPixels, height)
  102. // Test set and get row height on not exists worksheet.
  103. assert.EqualError(t, xlsx.SetRowHeight("SheetN", 1, 111.0), "sheet SheetN is not exist")
  104. _, err = xlsx.GetRowHeight("SheetN", 3)
  105. assert.EqualError(t, err, "sheet SheetN is not exist")
  106. err = xlsx.SaveAs(filepath.Join("test", "TestRowHeight.xlsx"))
  107. if !assert.NoError(t, err) {
  108. t.FailNow()
  109. }
  110. convertColWidthToPixels(0)
  111. }
  112. func TestColumns(t *testing.T) {
  113. f := NewFile()
  114. rows, err := f.Rows("Sheet1")
  115. assert.NoError(t, err)
  116. rows.decoder = f.xmlNewDecoder(bytes.NewReader([]byte(`<worksheet><sheetData><row r="A"><c r="A1" t="s"><v>1</v></c></row><row r="A"><c r="2" t="str"><v>B</v></c></row></sheetData></worksheet>`)))
  117. _, err = rows.Columns()
  118. assert.EqualError(t, err, `strconv.Atoi: parsing "A": invalid syntax`)
  119. rows.decoder = f.xmlNewDecoder(bytes.NewReader([]byte(`<worksheet><sheetData><row r="1"><c r="A1" t="s"><v>1</v></c></row><row r="A"><c r="2" t="str"><v>B</v></c></row></sheetData></worksheet>`)))
  120. _, err = rows.Columns()
  121. assert.NoError(t, err)
  122. rows.curRow = 3
  123. rows.decoder = f.xmlNewDecoder(bytes.NewReader([]byte(`<worksheet><sheetData><row r="1"><c r="A" t="s"><v>1</v></c></row></sheetData></worksheet>`)))
  124. _, err = rows.Columns()
  125. assert.EqualError(t, err, `cannot convert cell "A" to coordinates: invalid cell name "A"`)
  126. // Test token is nil
  127. rows.decoder = f.xmlNewDecoder(bytes.NewReader(nil))
  128. _, err = rows.Columns()
  129. assert.NoError(t, err)
  130. }
  131. func TestSharedStringsReader(t *testing.T) {
  132. f := NewFile()
  133. f.XLSX["xl/sharedStrings.xml"] = MacintoshCyrillicCharset
  134. f.sharedStringsReader()
  135. }
  136. func TestRowVisibility(t *testing.T) {
  137. f, err := prepareTestBook1()
  138. if !assert.NoError(t, err) {
  139. t.FailNow()
  140. }
  141. f.NewSheet("Sheet3")
  142. assert.NoError(t, f.SetRowVisible("Sheet3", 2, false))
  143. assert.NoError(t, f.SetRowVisible("Sheet3", 2, true))
  144. visiable, err := f.GetRowVisible("Sheet3", 2)
  145. assert.Equal(t, true, visiable)
  146. assert.NoError(t, err)
  147. visiable, err = f.GetRowVisible("Sheet3", 25)
  148. assert.Equal(t, false, visiable)
  149. assert.NoError(t, err)
  150. assert.EqualError(t, f.SetRowVisible("Sheet3", 0, true), "invalid row number 0")
  151. assert.EqualError(t, f.SetRowVisible("SheetN", 2, false), "sheet SheetN is not exist")
  152. visible, err := f.GetRowVisible("Sheet3", 0)
  153. assert.Equal(t, false, visible)
  154. assert.EqualError(t, err, "invalid row number 0")
  155. _, err = f.GetRowVisible("SheetN", 1)
  156. assert.EqualError(t, err, "sheet SheetN is not exist")
  157. assert.NoError(t, f.SaveAs(filepath.Join("test", "TestRowVisibility.xlsx")))
  158. }
  159. func TestRemoveRow(t *testing.T) {
  160. f := NewFile()
  161. sheet1 := f.GetSheetName(1)
  162. r, err := f.workSheetReader(sheet1)
  163. assert.NoError(t, err)
  164. const (
  165. colCount = 10
  166. rowCount = 10
  167. )
  168. fillCells(f, sheet1, colCount, rowCount)
  169. assert.NoError(t, f.SetCellHyperLink(sheet1, "A5", "https://github.com/360EntSecGroup-Skylar/excelize", "External"))
  170. assert.EqualError(t, f.RemoveRow(sheet1, -1), "invalid row number -1")
  171. assert.EqualError(t, f.RemoveRow(sheet1, 0), "invalid row number 0")
  172. assert.NoError(t, f.RemoveRow(sheet1, 4))
  173. if !assert.Len(t, r.SheetData.Row, rowCount-1) {
  174. t.FailNow()
  175. }
  176. assert.NoError(t, f.MergeCell(sheet1, "B3", "B5"))
  177. assert.NoError(t, f.RemoveRow(sheet1, 2))
  178. if !assert.Len(t, r.SheetData.Row, rowCount-2) {
  179. t.FailNow()
  180. }
  181. assert.NoError(t, f.RemoveRow(sheet1, 4))
  182. if !assert.Len(t, r.SheetData.Row, rowCount-3) {
  183. t.FailNow()
  184. }
  185. err = f.AutoFilter(sheet1, "A2", "A2", `{"column":"A","expression":"x != blanks"}`)
  186. if !assert.NoError(t, err) {
  187. t.FailNow()
  188. }
  189. assert.NoError(t, f.RemoveRow(sheet1, 1))
  190. if !assert.Len(t, r.SheetData.Row, rowCount-4) {
  191. t.FailNow()
  192. }
  193. assert.NoError(t, f.RemoveRow(sheet1, 2))
  194. if !assert.Len(t, r.SheetData.Row, rowCount-5) {
  195. t.FailNow()
  196. }
  197. assert.NoError(t, f.RemoveRow(sheet1, 1))
  198. if !assert.Len(t, r.SheetData.Row, rowCount-6) {
  199. t.FailNow()
  200. }
  201. assert.NoError(t, f.RemoveRow(sheet1, 10))
  202. assert.NoError(t, f.SaveAs(filepath.Join("test", "TestRemoveRow.xlsx")))
  203. // Test remove row on not exist worksheet
  204. assert.EqualError(t, f.RemoveRow("SheetN", 1), `sheet SheetN is not exist`)
  205. }
  206. func TestInsertRow(t *testing.T) {
  207. xlsx := NewFile()
  208. sheet1 := xlsx.GetSheetName(1)
  209. r, err := xlsx.workSheetReader(sheet1)
  210. assert.NoError(t, err)
  211. const (
  212. colCount = 10
  213. rowCount = 10
  214. )
  215. fillCells(xlsx, sheet1, colCount, rowCount)
  216. assert.NoError(t, xlsx.SetCellHyperLink(sheet1, "A5", "https://github.com/360EntSecGroup-Skylar/excelize", "External"))
  217. assert.EqualError(t, xlsx.InsertRow(sheet1, -1), "invalid row number -1")
  218. assert.EqualError(t, xlsx.InsertRow(sheet1, 0), "invalid row number 0")
  219. assert.NoError(t, xlsx.InsertRow(sheet1, 1))
  220. if !assert.Len(t, r.SheetData.Row, rowCount+1) {
  221. t.FailNow()
  222. }
  223. assert.NoError(t, xlsx.InsertRow(sheet1, 4))
  224. if !assert.Len(t, r.SheetData.Row, rowCount+2) {
  225. t.FailNow()
  226. }
  227. assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestInsertRow.xlsx")))
  228. }
  229. // Testing internal sructure state after insert operations.
  230. // It is important for insert workflow to be constant to avoid side effect with functions related to internal structure.
  231. func TestInsertRowInEmptyFile(t *testing.T) {
  232. xlsx := NewFile()
  233. sheet1 := xlsx.GetSheetName(1)
  234. r, err := xlsx.workSheetReader(sheet1)
  235. assert.NoError(t, err)
  236. assert.NoError(t, xlsx.InsertRow(sheet1, 1))
  237. assert.Len(t, r.SheetData.Row, 0)
  238. assert.NoError(t, xlsx.InsertRow(sheet1, 2))
  239. assert.Len(t, r.SheetData.Row, 0)
  240. assert.NoError(t, xlsx.InsertRow(sheet1, 99))
  241. assert.Len(t, r.SheetData.Row, 0)
  242. assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestInsertRowInEmptyFile.xlsx")))
  243. }
  244. func TestDuplicateRowFromSingleRow(t *testing.T) {
  245. const sheet = "Sheet1"
  246. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  247. cells := map[string]string{
  248. "A1": "A1 Value",
  249. "A2": "A2 Value",
  250. "A3": "A3 Value",
  251. "B1": "B1 Value",
  252. "B2": "B2 Value",
  253. "B3": "B3 Value",
  254. }
  255. t.Run("FromSingleRow", func(t *testing.T) {
  256. xlsx := NewFile()
  257. assert.NoError(t, xlsx.SetCellStr(sheet, "A1", cells["A1"]))
  258. assert.NoError(t, xlsx.SetCellStr(sheet, "B1", cells["B1"]))
  259. assert.NoError(t, xlsx.DuplicateRow(sheet, 1))
  260. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.FromSingleRow_1"))) {
  261. t.FailNow()
  262. }
  263. expect := map[string]string{
  264. "A1": cells["A1"], "B1": cells["B1"],
  265. "A2": cells["A1"], "B2": cells["B1"],
  266. }
  267. for cell, val := range expect {
  268. v, err := xlsx.GetCellValue(sheet, cell)
  269. assert.NoError(t, err)
  270. if !assert.Equal(t, val, v, cell) {
  271. t.FailNow()
  272. }
  273. }
  274. assert.NoError(t, xlsx.DuplicateRow(sheet, 2))
  275. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.FromSingleRow_2"))) {
  276. t.FailNow()
  277. }
  278. expect = map[string]string{
  279. "A1": cells["A1"], "B1": cells["B1"],
  280. "A2": cells["A1"], "B2": cells["B1"],
  281. "A3": cells["A1"], "B3": cells["B1"],
  282. }
  283. for cell, val := range expect {
  284. v, err := xlsx.GetCellValue(sheet, cell)
  285. assert.NoError(t, err)
  286. if !assert.Equal(t, val, v, cell) {
  287. t.FailNow()
  288. }
  289. }
  290. })
  291. }
  292. func TestDuplicateRowUpdateDuplicatedRows(t *testing.T) {
  293. const sheet = "Sheet1"
  294. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  295. cells := map[string]string{
  296. "A1": "A1 Value",
  297. "A2": "A2 Value",
  298. "A3": "A3 Value",
  299. "B1": "B1 Value",
  300. "B2": "B2 Value",
  301. "B3": "B3 Value",
  302. }
  303. t.Run("UpdateDuplicatedRows", func(t *testing.T) {
  304. xlsx := NewFile()
  305. assert.NoError(t, xlsx.SetCellStr(sheet, "A1", cells["A1"]))
  306. assert.NoError(t, xlsx.SetCellStr(sheet, "B1", cells["B1"]))
  307. assert.NoError(t, xlsx.DuplicateRow(sheet, 1))
  308. assert.NoError(t, xlsx.SetCellStr(sheet, "A2", cells["A2"]))
  309. assert.NoError(t, xlsx.SetCellStr(sheet, "B2", cells["B2"]))
  310. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.UpdateDuplicatedRows"))) {
  311. t.FailNow()
  312. }
  313. expect := map[string]string{
  314. "A1": cells["A1"], "B1": cells["B1"],
  315. "A2": cells["A2"], "B2": cells["B2"],
  316. }
  317. for cell, val := range expect {
  318. v, err := xlsx.GetCellValue(sheet, cell)
  319. assert.NoError(t, err)
  320. if !assert.Equal(t, val, v, cell) {
  321. t.FailNow()
  322. }
  323. }
  324. })
  325. }
  326. func TestDuplicateRowFirstOfMultipleRows(t *testing.T) {
  327. const sheet = "Sheet1"
  328. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  329. cells := map[string]string{
  330. "A1": "A1 Value",
  331. "A2": "A2 Value",
  332. "A3": "A3 Value",
  333. "B1": "B1 Value",
  334. "B2": "B2 Value",
  335. "B3": "B3 Value",
  336. }
  337. newFileWithDefaults := func() *File {
  338. f := NewFile()
  339. for cell, val := range cells {
  340. assert.NoError(t, f.SetCellStr(sheet, cell, val))
  341. }
  342. return f
  343. }
  344. t.Run("FirstOfMultipleRows", func(t *testing.T) {
  345. xlsx := newFileWithDefaults()
  346. assert.NoError(t, xlsx.DuplicateRow(sheet, 1))
  347. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.FirstOfMultipleRows"))) {
  348. t.FailNow()
  349. }
  350. expect := map[string]string{
  351. "A1": cells["A1"], "B1": cells["B1"],
  352. "A2": cells["A1"], "B2": cells["B1"],
  353. "A3": cells["A2"], "B3": cells["B2"],
  354. "A4": cells["A3"], "B4": cells["B3"],
  355. }
  356. for cell, val := range expect {
  357. v, err := xlsx.GetCellValue(sheet, cell)
  358. assert.NoError(t, err)
  359. if !assert.Equal(t, val, v, cell) {
  360. t.FailNow()
  361. }
  362. }
  363. })
  364. }
  365. func TestDuplicateRowZeroWithNoRows(t *testing.T) {
  366. const sheet = "Sheet1"
  367. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  368. t.Run("ZeroWithNoRows", func(t *testing.T) {
  369. xlsx := NewFile()
  370. assert.EqualError(t, xlsx.DuplicateRow(sheet, 0), "invalid row number 0")
  371. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.ZeroWithNoRows"))) {
  372. t.FailNow()
  373. }
  374. val, err := xlsx.GetCellValue(sheet, "A1")
  375. assert.NoError(t, err)
  376. assert.Equal(t, "", val)
  377. val, err = xlsx.GetCellValue(sheet, "B1")
  378. assert.NoError(t, err)
  379. assert.Equal(t, "", val)
  380. val, err = xlsx.GetCellValue(sheet, "A2")
  381. assert.NoError(t, err)
  382. assert.Equal(t, "", val)
  383. val, err = xlsx.GetCellValue(sheet, "B2")
  384. assert.NoError(t, err)
  385. assert.Equal(t, "", val)
  386. assert.NoError(t, err)
  387. expect := map[string]string{
  388. "A1": "", "B1": "",
  389. "A2": "", "B2": "",
  390. }
  391. for cell, val := range expect {
  392. v, err := xlsx.GetCellValue(sheet, cell)
  393. assert.NoError(t, err)
  394. if !assert.Equal(t, val, v, cell) {
  395. t.FailNow()
  396. }
  397. }
  398. })
  399. }
  400. func TestDuplicateRowMiddleRowOfEmptyFile(t *testing.T) {
  401. const sheet = "Sheet1"
  402. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  403. t.Run("MiddleRowOfEmptyFile", func(t *testing.T) {
  404. xlsx := NewFile()
  405. assert.NoError(t, xlsx.DuplicateRow(sheet, 99))
  406. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.MiddleRowOfEmptyFile"))) {
  407. t.FailNow()
  408. }
  409. expect := map[string]string{
  410. "A98": "",
  411. "A99": "",
  412. "A100": "",
  413. }
  414. for cell, val := range expect {
  415. v, err := xlsx.GetCellValue(sheet, cell)
  416. assert.NoError(t, err)
  417. if !assert.Equal(t, val, v, cell) {
  418. t.FailNow()
  419. }
  420. }
  421. })
  422. }
  423. func TestDuplicateRowWithLargeOffsetToMiddleOfData(t *testing.T) {
  424. const sheet = "Sheet1"
  425. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  426. cells := map[string]string{
  427. "A1": "A1 Value",
  428. "A2": "A2 Value",
  429. "A3": "A3 Value",
  430. "B1": "B1 Value",
  431. "B2": "B2 Value",
  432. "B3": "B3 Value",
  433. }
  434. newFileWithDefaults := func() *File {
  435. f := NewFile()
  436. for cell, val := range cells {
  437. assert.NoError(t, f.SetCellStr(sheet, cell, val))
  438. }
  439. return f
  440. }
  441. t.Run("WithLargeOffsetToMiddleOfData", func(t *testing.T) {
  442. xlsx := newFileWithDefaults()
  443. assert.NoError(t, xlsx.DuplicateRowTo(sheet, 1, 3))
  444. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.WithLargeOffsetToMiddleOfData"))) {
  445. t.FailNow()
  446. }
  447. expect := map[string]string{
  448. "A1": cells["A1"], "B1": cells["B1"],
  449. "A2": cells["A2"], "B2": cells["B2"],
  450. "A3": cells["A1"], "B3": cells["B1"],
  451. "A4": cells["A3"], "B4": cells["B3"],
  452. }
  453. for cell, val := range expect {
  454. v, err := xlsx.GetCellValue(sheet, cell)
  455. assert.NoError(t, err)
  456. if !assert.Equal(t, val, v, cell) {
  457. t.FailNow()
  458. }
  459. }
  460. })
  461. }
  462. func TestDuplicateRowWithLargeOffsetToEmptyRows(t *testing.T) {
  463. const sheet = "Sheet1"
  464. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  465. cells := map[string]string{
  466. "A1": "A1 Value",
  467. "A2": "A2 Value",
  468. "A3": "A3 Value",
  469. "B1": "B1 Value",
  470. "B2": "B2 Value",
  471. "B3": "B3 Value",
  472. }
  473. newFileWithDefaults := func() *File {
  474. f := NewFile()
  475. for cell, val := range cells {
  476. assert.NoError(t, f.SetCellStr(sheet, cell, val))
  477. }
  478. return f
  479. }
  480. t.Run("WithLargeOffsetToEmptyRows", func(t *testing.T) {
  481. xlsx := newFileWithDefaults()
  482. assert.NoError(t, xlsx.DuplicateRowTo(sheet, 1, 7))
  483. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.WithLargeOffsetToEmptyRows"))) {
  484. t.FailNow()
  485. }
  486. expect := map[string]string{
  487. "A1": cells["A1"], "B1": cells["B1"],
  488. "A2": cells["A2"], "B2": cells["B2"],
  489. "A3": cells["A3"], "B3": cells["B3"],
  490. "A7": cells["A1"], "B7": cells["B1"],
  491. }
  492. for cell, val := range expect {
  493. v, err := xlsx.GetCellValue(sheet, cell)
  494. assert.NoError(t, err)
  495. if !assert.Equal(t, val, v, cell) {
  496. t.FailNow()
  497. }
  498. }
  499. })
  500. }
  501. func TestDuplicateRowInsertBefore(t *testing.T) {
  502. const sheet = "Sheet1"
  503. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  504. cells := map[string]string{
  505. "A1": "A1 Value",
  506. "A2": "A2 Value",
  507. "A3": "A3 Value",
  508. "B1": "B1 Value",
  509. "B2": "B2 Value",
  510. "B3": "B3 Value",
  511. }
  512. newFileWithDefaults := func() *File {
  513. f := NewFile()
  514. for cell, val := range cells {
  515. assert.NoError(t, f.SetCellStr(sheet, cell, val))
  516. }
  517. return f
  518. }
  519. t.Run("InsertBefore", func(t *testing.T) {
  520. xlsx := newFileWithDefaults()
  521. assert.NoError(t, xlsx.DuplicateRowTo(sheet, 2, 1))
  522. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.InsertBefore"))) {
  523. t.FailNow()
  524. }
  525. expect := map[string]string{
  526. "A1": cells["A2"], "B1": cells["B2"],
  527. "A2": cells["A1"], "B2": cells["B1"],
  528. "A3": cells["A2"], "B3": cells["B2"],
  529. "A4": cells["A3"], "B4": cells["B3"],
  530. }
  531. for cell, val := range expect {
  532. v, err := xlsx.GetCellValue(sheet, cell)
  533. assert.NoError(t, err)
  534. if !assert.Equal(t, val, v, cell) {
  535. t.FailNow()
  536. }
  537. }
  538. })
  539. }
  540. func TestDuplicateRowInsertBeforeWithLargeOffset(t *testing.T) {
  541. const sheet = "Sheet1"
  542. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  543. cells := map[string]string{
  544. "A1": "A1 Value",
  545. "A2": "A2 Value",
  546. "A3": "A3 Value",
  547. "B1": "B1 Value",
  548. "B2": "B2 Value",
  549. "B3": "B3 Value",
  550. }
  551. newFileWithDefaults := func() *File {
  552. f := NewFile()
  553. for cell, val := range cells {
  554. assert.NoError(t, f.SetCellStr(sheet, cell, val))
  555. }
  556. return f
  557. }
  558. t.Run("InsertBeforeWithLargeOffset", func(t *testing.T) {
  559. xlsx := newFileWithDefaults()
  560. assert.NoError(t, xlsx.DuplicateRowTo(sheet, 3, 1))
  561. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.InsertBeforeWithLargeOffset"))) {
  562. t.FailNow()
  563. }
  564. expect := map[string]string{
  565. "A1": cells["A3"], "B1": cells["B3"],
  566. "A2": cells["A1"], "B2": cells["B1"],
  567. "A3": cells["A2"], "B3": cells["B2"],
  568. "A4": cells["A3"], "B4": cells["B3"],
  569. }
  570. for cell, val := range expect {
  571. v, err := xlsx.GetCellValue(sheet, cell)
  572. assert.NoError(t, err)
  573. if !assert.Equal(t, val, v) {
  574. t.FailNow()
  575. }
  576. }
  577. })
  578. }
  579. func TestDuplicateRowInvalidRownum(t *testing.T) {
  580. const sheet = "Sheet1"
  581. outFile := filepath.Join("test", "TestDuplicateRowInvalidRownum.%s.xlsx")
  582. cells := map[string]string{
  583. "A1": "A1 Value",
  584. "A2": "A2 Value",
  585. "A3": "A3 Value",
  586. "B1": "B1 Value",
  587. "B2": "B2 Value",
  588. "B3": "B3 Value",
  589. }
  590. invalidIndexes := []int{-100, -2, -1, 0}
  591. for _, row := range invalidIndexes {
  592. name := fmt.Sprintf("%d", row)
  593. t.Run(name, func(t *testing.T) {
  594. xlsx := NewFile()
  595. for col, val := range cells {
  596. assert.NoError(t, xlsx.SetCellStr(sheet, col, val))
  597. }
  598. assert.EqualError(t, xlsx.DuplicateRow(sheet, row), fmt.Sprintf("invalid row number %d", row))
  599. for col, val := range cells {
  600. v, err := xlsx.GetCellValue(sheet, col)
  601. assert.NoError(t, err)
  602. if !assert.Equal(t, val, v) {
  603. t.FailNow()
  604. }
  605. }
  606. assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, name)))
  607. })
  608. }
  609. for _, row1 := range invalidIndexes {
  610. for _, row2 := range invalidIndexes {
  611. name := fmt.Sprintf("[%d,%d]", row1, row2)
  612. t.Run(name, func(t *testing.T) {
  613. xlsx := NewFile()
  614. for col, val := range cells {
  615. assert.NoError(t, xlsx.SetCellStr(sheet, col, val))
  616. }
  617. assert.EqualError(t, xlsx.DuplicateRowTo(sheet, row1, row2), fmt.Sprintf("invalid row number %d", row1))
  618. for col, val := range cells {
  619. v, err := xlsx.GetCellValue(sheet, col)
  620. assert.NoError(t, err)
  621. if !assert.Equal(t, val, v) {
  622. t.FailNow()
  623. }
  624. }
  625. assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, name)))
  626. })
  627. }
  628. }
  629. }
  630. func TestGetValueFrom(t *testing.T) {
  631. c := &xlsxC{T: "inlineStr"}
  632. f := NewFile()
  633. d := &xlsxSST{}
  634. val, err := c.getValueFrom(f, d)
  635. assert.NoError(t, err)
  636. assert.Equal(t, "", val)
  637. }
  638. func TestErrSheetNotExistError(t *testing.T) {
  639. err := ErrSheetNotExist{SheetName: "Sheet1"}
  640. assert.EqualValues(t, err.Error(), "sheet Sheet1 is not exist")
  641. }
  642. func BenchmarkRows(b *testing.B) {
  643. f, _ := OpenFile(filepath.Join("test", "Book1.xlsx"))
  644. for i := 0; i < b.N; i++ {
  645. rows, _ := f.Rows("Sheet2")
  646. for rows.Next() {
  647. row, _ := rows.Columns()
  648. for i := range row {
  649. if i >= 0 {
  650. continue
  651. }
  652. }
  653. }
  654. }
  655. }
  656. func trimSliceSpace(s []string) []string {
  657. for {
  658. if len(s) > 0 && s[len(s)-1] == "" {
  659. s = s[:len(s)-1]
  660. } else {
  661. break
  662. }
  663. }
  664. return s
  665. }