analysis_excel.test.js 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. 'use strict';
  2. /**
  3. *
  4. *
  5. * @author Mai
  6. * @date
  7. * @version
  8. */
  9. const { app, assert } = require('egg-mock/bootstrap');
  10. const AnalysisExcel = require('../../../app/lib/analysis_excel');
  11. const Xlsx = require('js-xlsx');
  12. const _ = require('lodash');
  13. describe('test/app/lib/analysis_excel.test.js', () => {
  14. it('analysis Test Data', function* () {
  15. const analysisExcel = new AnalysisExcel();
  16. const testData = {
  17. rows: [
  18. ['项目节编号', '清单子目号', '部位明细', '名称', '单位', '清单数量', '设计数量1', '设计数量2', '单价', '合价', '图号', '备注'],
  19. ['1', null, null, '第一部分 建筑安装工程费', '公路公里'],
  20. ['1-1', null, null, '临时工程', '公路公里'],
  21. ['1-1-6', null, null, '临时道路修建、养护与拆除(包括原道路的养护费)'],
  22. [ null, '103-1', null, '临时道路修建、养护与拆除(包括原道路的养护费)', '总额', 1, null, null, 50000 ],
  23. [ '1-1-7', null, null, '临时占地' ],
  24. [ null, '103-2', null, '临时占地', '总额', 1, null, null, 57350 ],
  25. [ '1-1-8', null, null, '临时供电设施' ],
  26. [ null, '103-3-a', null, '设施架设、拆除', '总额', 1, null, null, 638406 ],
  27. [ null, '103-3-b', null, '设施维修', '总额', 1, null, null, 50000 ],
  28. [ '1-1-9', null, null, '电讯设施的提供、维修与拆除' ],
  29. [ null, '103-4', null, '电讯设施的提供、维修与拆除', '总额', 1, null, null, 50000 ],
  30. [ '1-1-10', null, null, '承包人驻地建设' ],
  31. [ null, '104-1', null, '承包人驻地建设', '总额', 1, null, null, 200000 ],
  32. [ '1-2', null, null, '路基工程', 'km' ],
  33. [ '1-2-2', null, null, '挖方', 'm3' ],
  34. [ '1-2-2-1', null, null, '挖土方', 'm3' ],
  35. [ '1-2-2-1-1', null, null, '挖路基土方', 'm3' ],
  36. [ null, '203-1-a', null, ' ', 'm3', null, null, null, 7.53 ],
  37. [ null, null, 1, 'K0+000-K1+000', null, 11619.34375, null, null, null, null, '第二册S-2-1' ],
  38. [ null, null, 2, 'K1+000-K1+800', null, 11619.34375, null, null, null, null, '第二册S-2-1' ],
  39. [ null, null, 3, 'K2+800-K3+004', null, 11619.34375, null, null, null, null, '第二册S-2-1' ],
  40. [ null, null, 4, 'K3+004-K4+040', null, 11619.34375, null, null, null, null, '第二册S-2-1' ],
  41. [ null, null, 5, 'K4+040-K5+000', null, 11619.34375, null, null, null, null, '第二册S-2-1' ],
  42. [ null, null, 6, 'K5+000-K6+000', null, 11619.34375, null, null, null, null, '第二册S-2-1' ],
  43. [ null, null, 7, 'K6+000-K7+000', null, 11619.34375, null, null, null, null, '第二册S-2-1' ],
  44. [ null, null, 8, 'K7+000-K7+700', null, 11619.34375, null, null, null, null, '第二册S-2-1' ],
  45. [ '1-2-2-2', null, null, '挖石方', 'm3' ],
  46. [ '1-2-2-2-1', null, null, '挖软石', 'm3' ],
  47. [ null, '203-1-b', null, '挖石方', 'm3', 105.36, null, null, 16.11 ],
  48. [ '1-3', null, null, '路面工程', 'km' ],
  49. [ '1-4', null, null, '桥梁涵洞工程', 'km' ],
  50. [ '1-6', null, null, '隧道工程', 'km' ],
  51. ],
  52. range: 'A1:L29',
  53. };
  54. const ctx = app.mockContext();
  55. const templateData = yield ctx.service.tenderNodeTemplate.getData(true);
  56. const result = analysisExcel.analysisData(testData, templateData);
  57. // 检查插入总量
  58. assert(result.items.length === 72);
  59. assert(result.roots.length === 6);
  60. assert(result.pos.length === 8);
  61. // 检查第一部分子节点排序
  62. const nodes = result.roots[0].children;
  63. assert(nodes[nodes.length - 2].code === '1-6');
  64. assert(nodes[nodes.length - 1].code === '1-10');
  65. // 检查第203-1-a的数量
  66. const xmj = result.codeNodes['1-2-2-1-1'];
  67. const gcl = xmj.children[0];
  68. assert(gcl.quantity === 92954.75);
  69. });
  70. it('Analysis Excel Test Data And Import', function* () {
  71. const excelFile = app.baseDir + '/app/public/files/template/ledger/导入分项清单Excel格式.xls';
  72. const wb = Xlsx.readFile(excelFile);
  73. const name = wb.SheetNames[0];
  74. const sheetData = {
  75. rows: Xlsx.utils.sheet_to_json(wb.Sheets[name], {header: 1}),
  76. merge: wb.Sheets[name]["!merges"],
  77. };
  78. const ctx = app.mockContext();
  79. const analysisExcel = new AnalysisExcel();
  80. const templateData = yield ctx.service.tenderNodeTemplate.getData(true);
  81. const result = analysisExcel.analysisData(sheetData, templateData);
  82. assert(result.items.length === 216);
  83. const xmj = result.codeNodes['1-2-3-1-2'];
  84. const gcl = xmj.children[0];
  85. assert(gcl.quantity === 0);
  86. ctx.tender = { id: 2 };
  87. // pos需要记录createUserId
  88. ctx.session = {
  89. sessionUser: {
  90. accountId: 2,
  91. },
  92. };
  93. yield ctx.service.ledger.importExcel(sheetData);
  94. });
  95. it('analysis 渭武12标 Excel', function* () {
  96. const excelFile = app.baseDir + '/app/public/files/template/ledger/渭武12标.xls';
  97. const wb = Xlsx.readFile(excelFile);
  98. const name = wb.SheetNames[0];
  99. const sheetData = {
  100. rows: Xlsx.utils.sheet_to_json(wb.Sheets[name], {header: 1}),
  101. merge: wb.Sheets[name]["!merges"],
  102. };
  103. const ctx = app.mockContext();
  104. ctx.tender = { id: 2 };
  105. // pos需要记录createUserId
  106. ctx.session = {
  107. sessionUser: {
  108. accountId: 2,
  109. },
  110. };
  111. yield ctx.service.ledger.importExcel(sheetData);
  112. });
  113. });