rows_test.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785
  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. 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. f.GetRowVisible("Sheet3", 2)
  145. f.GetRowVisible("Sheet3", 25)
  146. assert.EqualError(t, f.SetRowVisible("Sheet3", 0, true), "invalid row number 0")
  147. assert.EqualError(t, f.SetRowVisible("SheetN", 2, false), "sheet SheetN is not exist")
  148. visible, err := f.GetRowVisible("Sheet3", 0)
  149. assert.Equal(t, false, visible)
  150. assert.EqualError(t, err, "invalid row number 0")
  151. _, err = f.GetRowVisible("SheetN", 1)
  152. assert.EqualError(t, err, "sheet SheetN is not exist")
  153. assert.NoError(t, f.SaveAs(filepath.Join("test", "TestRowVisibility.xlsx")))
  154. }
  155. func TestRemoveRow(t *testing.T) {
  156. f := NewFile()
  157. sheet1 := f.GetSheetName(1)
  158. r, err := f.workSheetReader(sheet1)
  159. assert.NoError(t, err)
  160. const (
  161. colCount = 10
  162. rowCount = 10
  163. )
  164. fillCells(f, sheet1, colCount, rowCount)
  165. f.SetCellHyperLink(sheet1, "A5", "https://github.com/360EntSecGroup-Skylar/excelize", "External")
  166. assert.EqualError(t, f.RemoveRow(sheet1, -1), "invalid row number -1")
  167. assert.EqualError(t, f.RemoveRow(sheet1, 0), "invalid row number 0")
  168. assert.NoError(t, f.RemoveRow(sheet1, 4))
  169. if !assert.Len(t, r.SheetData.Row, rowCount-1) {
  170. t.FailNow()
  171. }
  172. f.MergeCell(sheet1, "B3", "B5")
  173. assert.NoError(t, f.RemoveRow(sheet1, 2))
  174. if !assert.Len(t, r.SheetData.Row, rowCount-2) {
  175. t.FailNow()
  176. }
  177. assert.NoError(t, f.RemoveRow(sheet1, 4))
  178. if !assert.Len(t, r.SheetData.Row, rowCount-3) {
  179. t.FailNow()
  180. }
  181. err = f.AutoFilter(sheet1, "A2", "A2", `{"column":"A","expression":"x != blanks"}`)
  182. if !assert.NoError(t, err) {
  183. t.FailNow()
  184. }
  185. assert.NoError(t, f.RemoveRow(sheet1, 1))
  186. if !assert.Len(t, r.SheetData.Row, rowCount-4) {
  187. t.FailNow()
  188. }
  189. assert.NoError(t, f.RemoveRow(sheet1, 2))
  190. if !assert.Len(t, r.SheetData.Row, rowCount-5) {
  191. t.FailNow()
  192. }
  193. assert.NoError(t, f.RemoveRow(sheet1, 1))
  194. if !assert.Len(t, r.SheetData.Row, rowCount-6) {
  195. t.FailNow()
  196. }
  197. assert.NoError(t, f.RemoveRow(sheet1, 10))
  198. assert.NoError(t, f.SaveAs(filepath.Join("test", "TestRemoveRow.xlsx")))
  199. // Test remove row on not exist worksheet
  200. assert.EqualError(t, f.RemoveRow("SheetN", 1), `sheet SheetN is not exist`)
  201. }
  202. func TestInsertRow(t *testing.T) {
  203. xlsx := NewFile()
  204. sheet1 := xlsx.GetSheetName(1)
  205. r, err := xlsx.workSheetReader(sheet1)
  206. assert.NoError(t, err)
  207. const (
  208. colCount = 10
  209. rowCount = 10
  210. )
  211. fillCells(xlsx, sheet1, colCount, rowCount)
  212. xlsx.SetCellHyperLink(sheet1, "A5", "https://github.com/360EntSecGroup-Skylar/excelize", "External")
  213. assert.EqualError(t, xlsx.InsertRow(sheet1, -1), "invalid row number -1")
  214. assert.EqualError(t, xlsx.InsertRow(sheet1, 0), "invalid row number 0")
  215. assert.NoError(t, xlsx.InsertRow(sheet1, 1))
  216. if !assert.Len(t, r.SheetData.Row, rowCount+1) {
  217. t.FailNow()
  218. }
  219. assert.NoError(t, xlsx.InsertRow(sheet1, 4))
  220. if !assert.Len(t, r.SheetData.Row, rowCount+2) {
  221. t.FailNow()
  222. }
  223. assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestInsertRow.xlsx")))
  224. }
  225. // Testing internal sructure state after insert operations.
  226. // It is important for insert workflow to be constant to avoid side effect with functions related to internal structure.
  227. func TestInsertRowInEmptyFile(t *testing.T) {
  228. xlsx := NewFile()
  229. sheet1 := xlsx.GetSheetName(1)
  230. r, err := xlsx.workSheetReader(sheet1)
  231. assert.NoError(t, err)
  232. assert.NoError(t, xlsx.InsertRow(sheet1, 1))
  233. assert.Len(t, r.SheetData.Row, 0)
  234. assert.NoError(t, xlsx.InsertRow(sheet1, 2))
  235. assert.Len(t, r.SheetData.Row, 0)
  236. assert.NoError(t, xlsx.InsertRow(sheet1, 99))
  237. assert.Len(t, r.SheetData.Row, 0)
  238. assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestInsertRowInEmptyFile.xlsx")))
  239. }
  240. func TestDuplicateRowFromSingleRow(t *testing.T) {
  241. const sheet = "Sheet1"
  242. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  243. cells := map[string]string{
  244. "A1": "A1 Value",
  245. "A2": "A2 Value",
  246. "A3": "A3 Value",
  247. "B1": "B1 Value",
  248. "B2": "B2 Value",
  249. "B3": "B3 Value",
  250. }
  251. t.Run("FromSingleRow", func(t *testing.T) {
  252. xlsx := NewFile()
  253. xlsx.SetCellStr(sheet, "A1", cells["A1"])
  254. xlsx.SetCellStr(sheet, "B1", cells["B1"])
  255. assert.NoError(t, xlsx.DuplicateRow(sheet, 1))
  256. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.FromSingleRow_1"))) {
  257. t.FailNow()
  258. }
  259. expect := map[string]string{
  260. "A1": cells["A1"], "B1": cells["B1"],
  261. "A2": cells["A1"], "B2": cells["B1"],
  262. }
  263. for cell, val := range expect {
  264. v, err := xlsx.GetCellValue(sheet, cell)
  265. assert.NoError(t, err)
  266. if !assert.Equal(t, val, v, cell) {
  267. t.FailNow()
  268. }
  269. }
  270. assert.NoError(t, xlsx.DuplicateRow(sheet, 2))
  271. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.FromSingleRow_2"))) {
  272. t.FailNow()
  273. }
  274. expect = map[string]string{
  275. "A1": cells["A1"], "B1": cells["B1"],
  276. "A2": cells["A1"], "B2": cells["B1"],
  277. "A3": cells["A1"], "B3": cells["B1"],
  278. }
  279. for cell, val := range expect {
  280. v, err := xlsx.GetCellValue(sheet, cell)
  281. assert.NoError(t, err)
  282. if !assert.Equal(t, val, v, cell) {
  283. t.FailNow()
  284. }
  285. }
  286. })
  287. }
  288. func TestDuplicateRowUpdateDuplicatedRows(t *testing.T) {
  289. const sheet = "Sheet1"
  290. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  291. cells := map[string]string{
  292. "A1": "A1 Value",
  293. "A2": "A2 Value",
  294. "A3": "A3 Value",
  295. "B1": "B1 Value",
  296. "B2": "B2 Value",
  297. "B3": "B3 Value",
  298. }
  299. t.Run("UpdateDuplicatedRows", func(t *testing.T) {
  300. xlsx := NewFile()
  301. xlsx.SetCellStr(sheet, "A1", cells["A1"])
  302. xlsx.SetCellStr(sheet, "B1", cells["B1"])
  303. assert.NoError(t, xlsx.DuplicateRow(sheet, 1))
  304. xlsx.SetCellStr(sheet, "A2", cells["A2"])
  305. xlsx.SetCellStr(sheet, "B2", cells["B2"])
  306. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.UpdateDuplicatedRows"))) {
  307. t.FailNow()
  308. }
  309. expect := map[string]string{
  310. "A1": cells["A1"], "B1": cells["B1"],
  311. "A2": cells["A2"], "B2": cells["B2"],
  312. }
  313. for cell, val := range expect {
  314. v, err := xlsx.GetCellValue(sheet, cell)
  315. assert.NoError(t, err)
  316. if !assert.Equal(t, val, v, cell) {
  317. t.FailNow()
  318. }
  319. }
  320. })
  321. }
  322. func TestDuplicateRowFirstOfMultipleRows(t *testing.T) {
  323. const sheet = "Sheet1"
  324. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  325. cells := map[string]string{
  326. "A1": "A1 Value",
  327. "A2": "A2 Value",
  328. "A3": "A3 Value",
  329. "B1": "B1 Value",
  330. "B2": "B2 Value",
  331. "B3": "B3 Value",
  332. }
  333. newFileWithDefaults := func() *File {
  334. f := NewFile()
  335. for cell, val := range cells {
  336. f.SetCellStr(sheet, cell, val)
  337. }
  338. return f
  339. }
  340. t.Run("FirstOfMultipleRows", func(t *testing.T) {
  341. xlsx := newFileWithDefaults()
  342. assert.NoError(t, xlsx.DuplicateRow(sheet, 1))
  343. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.FirstOfMultipleRows"))) {
  344. t.FailNow()
  345. }
  346. expect := map[string]string{
  347. "A1": cells["A1"], "B1": cells["B1"],
  348. "A2": cells["A1"], "B2": cells["B1"],
  349. "A3": cells["A2"], "B3": cells["B2"],
  350. "A4": cells["A3"], "B4": cells["B3"],
  351. }
  352. for cell, val := range expect {
  353. v, err := xlsx.GetCellValue(sheet, cell)
  354. assert.NoError(t, err)
  355. if !assert.Equal(t, val, v, cell) {
  356. t.FailNow()
  357. }
  358. }
  359. })
  360. }
  361. func TestDuplicateRowZeroWithNoRows(t *testing.T) {
  362. const sheet = "Sheet1"
  363. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  364. t.Run("ZeroWithNoRows", func(t *testing.T) {
  365. xlsx := NewFile()
  366. assert.EqualError(t, xlsx.DuplicateRow(sheet, 0), "invalid row number 0")
  367. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.ZeroWithNoRows"))) {
  368. t.FailNow()
  369. }
  370. val, err := xlsx.GetCellValue(sheet, "A1")
  371. assert.NoError(t, err)
  372. assert.Equal(t, "", val)
  373. val, err = xlsx.GetCellValue(sheet, "B1")
  374. assert.NoError(t, err)
  375. assert.Equal(t, "", val)
  376. val, err = xlsx.GetCellValue(sheet, "A2")
  377. assert.NoError(t, err)
  378. assert.Equal(t, "", val)
  379. val, err = xlsx.GetCellValue(sheet, "B2")
  380. assert.NoError(t, err)
  381. assert.Equal(t, "", val)
  382. assert.NoError(t, err)
  383. expect := map[string]string{
  384. "A1": "", "B1": "",
  385. "A2": "", "B2": "",
  386. }
  387. for cell, val := range expect {
  388. v, err := xlsx.GetCellValue(sheet, cell)
  389. assert.NoError(t, err)
  390. if !assert.Equal(t, val, v, cell) {
  391. t.FailNow()
  392. }
  393. }
  394. })
  395. }
  396. func TestDuplicateRowMiddleRowOfEmptyFile(t *testing.T) {
  397. const sheet = "Sheet1"
  398. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  399. t.Run("MiddleRowOfEmptyFile", func(t *testing.T) {
  400. xlsx := NewFile()
  401. assert.NoError(t, xlsx.DuplicateRow(sheet, 99))
  402. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.MiddleRowOfEmptyFile"))) {
  403. t.FailNow()
  404. }
  405. expect := map[string]string{
  406. "A98": "",
  407. "A99": "",
  408. "A100": "",
  409. }
  410. for cell, val := range expect {
  411. v, err := xlsx.GetCellValue(sheet, cell)
  412. assert.NoError(t, err)
  413. if !assert.Equal(t, val, v, cell) {
  414. t.FailNow()
  415. }
  416. }
  417. })
  418. }
  419. func TestDuplicateRowWithLargeOffsetToMiddleOfData(t *testing.T) {
  420. const sheet = "Sheet1"
  421. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  422. cells := map[string]string{
  423. "A1": "A1 Value",
  424. "A2": "A2 Value",
  425. "A3": "A3 Value",
  426. "B1": "B1 Value",
  427. "B2": "B2 Value",
  428. "B3": "B3 Value",
  429. }
  430. newFileWithDefaults := func() *File {
  431. f := NewFile()
  432. for cell, val := range cells {
  433. f.SetCellStr(sheet, cell, val)
  434. }
  435. return f
  436. }
  437. t.Run("WithLargeOffsetToMiddleOfData", func(t *testing.T) {
  438. xlsx := newFileWithDefaults()
  439. assert.NoError(t, xlsx.DuplicateRowTo(sheet, 1, 3))
  440. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.WithLargeOffsetToMiddleOfData"))) {
  441. t.FailNow()
  442. }
  443. expect := map[string]string{
  444. "A1": cells["A1"], "B1": cells["B1"],
  445. "A2": cells["A2"], "B2": cells["B2"],
  446. "A3": cells["A1"], "B3": cells["B1"],
  447. "A4": cells["A3"], "B4": cells["B3"],
  448. }
  449. for cell, val := range expect {
  450. v, err := xlsx.GetCellValue(sheet, cell)
  451. assert.NoError(t, err)
  452. if !assert.Equal(t, val, v, cell) {
  453. t.FailNow()
  454. }
  455. }
  456. })
  457. }
  458. func TestDuplicateRowWithLargeOffsetToEmptyRows(t *testing.T) {
  459. const sheet = "Sheet1"
  460. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  461. cells := map[string]string{
  462. "A1": "A1 Value",
  463. "A2": "A2 Value",
  464. "A3": "A3 Value",
  465. "B1": "B1 Value",
  466. "B2": "B2 Value",
  467. "B3": "B3 Value",
  468. }
  469. newFileWithDefaults := func() *File {
  470. f := NewFile()
  471. for cell, val := range cells {
  472. f.SetCellStr(sheet, cell, val)
  473. }
  474. return f
  475. }
  476. t.Run("WithLargeOffsetToEmptyRows", func(t *testing.T) {
  477. xlsx := newFileWithDefaults()
  478. assert.NoError(t, xlsx.DuplicateRowTo(sheet, 1, 7))
  479. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.WithLargeOffsetToEmptyRows"))) {
  480. t.FailNow()
  481. }
  482. expect := map[string]string{
  483. "A1": cells["A1"], "B1": cells["B1"],
  484. "A2": cells["A2"], "B2": cells["B2"],
  485. "A3": cells["A3"], "B3": cells["B3"],
  486. "A7": cells["A1"], "B7": cells["B1"],
  487. }
  488. for cell, val := range expect {
  489. v, err := xlsx.GetCellValue(sheet, cell)
  490. assert.NoError(t, err)
  491. if !assert.Equal(t, val, v, cell) {
  492. t.FailNow()
  493. }
  494. }
  495. })
  496. }
  497. func TestDuplicateRowInsertBefore(t *testing.T) {
  498. const sheet = "Sheet1"
  499. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  500. cells := map[string]string{
  501. "A1": "A1 Value",
  502. "A2": "A2 Value",
  503. "A3": "A3 Value",
  504. "B1": "B1 Value",
  505. "B2": "B2 Value",
  506. "B3": "B3 Value",
  507. }
  508. newFileWithDefaults := func() *File {
  509. f := NewFile()
  510. for cell, val := range cells {
  511. f.SetCellStr(sheet, cell, val)
  512. }
  513. return f
  514. }
  515. t.Run("InsertBefore", func(t *testing.T) {
  516. xlsx := newFileWithDefaults()
  517. assert.NoError(t, xlsx.DuplicateRowTo(sheet, 2, 1))
  518. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.InsertBefore"))) {
  519. t.FailNow()
  520. }
  521. expect := map[string]string{
  522. "A1": cells["A2"], "B1": cells["B2"],
  523. "A2": cells["A1"], "B2": cells["B1"],
  524. "A3": cells["A2"], "B3": cells["B2"],
  525. "A4": cells["A3"], "B4": cells["B3"],
  526. }
  527. for cell, val := range expect {
  528. v, err := xlsx.GetCellValue(sheet, cell)
  529. assert.NoError(t, err)
  530. if !assert.Equal(t, val, v, cell) {
  531. t.FailNow()
  532. }
  533. }
  534. })
  535. }
  536. func TestDuplicateRowInsertBeforeWithLargeOffset(t *testing.T) {
  537. const sheet = "Sheet1"
  538. outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
  539. cells := map[string]string{
  540. "A1": "A1 Value",
  541. "A2": "A2 Value",
  542. "A3": "A3 Value",
  543. "B1": "B1 Value",
  544. "B2": "B2 Value",
  545. "B3": "B3 Value",
  546. }
  547. newFileWithDefaults := func() *File {
  548. f := NewFile()
  549. for cell, val := range cells {
  550. f.SetCellStr(sheet, cell, val)
  551. }
  552. return f
  553. }
  554. t.Run("InsertBeforeWithLargeOffset", func(t *testing.T) {
  555. xlsx := newFileWithDefaults()
  556. assert.NoError(t, xlsx.DuplicateRowTo(sheet, 3, 1))
  557. if !assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, "TestDuplicateRow.InsertBeforeWithLargeOffset"))) {
  558. t.FailNow()
  559. }
  560. expect := map[string]string{
  561. "A1": cells["A3"], "B1": cells["B3"],
  562. "A2": cells["A1"], "B2": cells["B1"],
  563. "A3": cells["A2"], "B3": cells["B2"],
  564. "A4": cells["A3"], "B4": cells["B3"],
  565. }
  566. for cell, val := range expect {
  567. v, err := xlsx.GetCellValue(sheet, cell)
  568. assert.NoError(t, err)
  569. if !assert.Equal(t, val, v) {
  570. t.FailNow()
  571. }
  572. }
  573. })
  574. }
  575. func TestDuplicateRowInvalidRownum(t *testing.T) {
  576. const sheet = "Sheet1"
  577. outFile := filepath.Join("test", "TestDuplicateRowInvalidRownum.%s.xlsx")
  578. cells := map[string]string{
  579. "A1": "A1 Value",
  580. "A2": "A2 Value",
  581. "A3": "A3 Value",
  582. "B1": "B1 Value",
  583. "B2": "B2 Value",
  584. "B3": "B3 Value",
  585. }
  586. invalidIndexes := []int{-100, -2, -1, 0}
  587. for _, row := range invalidIndexes {
  588. name := fmt.Sprintf("%d", row)
  589. t.Run(name, func(t *testing.T) {
  590. xlsx := NewFile()
  591. for col, val := range cells {
  592. xlsx.SetCellStr(sheet, col, val)
  593. }
  594. assert.EqualError(t, xlsx.DuplicateRow(sheet, row), fmt.Sprintf("invalid row number %d", row))
  595. for col, val := range cells {
  596. v, err := xlsx.GetCellValue(sheet, col)
  597. assert.NoError(t, err)
  598. if !assert.Equal(t, val, v) {
  599. t.FailNow()
  600. }
  601. }
  602. assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, name)))
  603. })
  604. }
  605. for _, row1 := range invalidIndexes {
  606. for _, row2 := range invalidIndexes {
  607. name := fmt.Sprintf("[%d,%d]", row1, row2)
  608. t.Run(name, func(t *testing.T) {
  609. xlsx := NewFile()
  610. for col, val := range cells {
  611. xlsx.SetCellStr(sheet, col, val)
  612. }
  613. assert.EqualError(t, xlsx.DuplicateRowTo(sheet, row1, row2), fmt.Sprintf("invalid row number %d", row1))
  614. for col, val := range cells {
  615. v, err := xlsx.GetCellValue(sheet, col)
  616. assert.NoError(t, err)
  617. if !assert.Equal(t, val, v) {
  618. t.FailNow()
  619. }
  620. }
  621. assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, name)))
  622. })
  623. }
  624. }
  625. }
  626. func TestGetValueFrom(t *testing.T) {
  627. c := &xlsxC{T: "inlineStr"}
  628. f := NewFile()
  629. d := &xlsxSST{}
  630. val, err := c.getValueFrom(f, d)
  631. assert.NoError(t, err)
  632. assert.Equal(t, "", val)
  633. }
  634. func TestErrSheetNotExistError(t *testing.T) {
  635. err := ErrSheetNotExist{SheetName: "Sheet1"}
  636. assert.EqualValues(t, err.Error(), "sheet Sheet1 is not exist")
  637. }
  638. func BenchmarkRows(b *testing.B) {
  639. f, _ := OpenFile(filepath.Join("test", "Book1.xlsx"))
  640. for i := 0; i < b.N; i++ {
  641. rows, _ := f.Rows("Sheet2")
  642. for rows.Next() {
  643. row, _ := rows.Columns()
  644. for i := range row {
  645. if i >= 0 {
  646. continue
  647. }
  648. }
  649. }
  650. }
  651. }
  652. func trimSliceSpace(s []string) []string {
  653. for {
  654. if len(s) > 0 && s[len(s)-1] == "" {
  655. s = s[:len(s)-1]
  656. } else {
  657. break
  658. }
  659. }
  660. return s
  661. }