rows_test.go 18 KB

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