完善标段相关单元测试

test/app/service/tender.test.js

@@ -14,8 +14,14 @@ const addData = {
     name: 'test_addTender',
     category: null,
 };
+const renderbody = {
+    name : '测试保存项目信息',
+    type: 1,
+};
 const mockData = {};
 let testTenderId;
+const _ = require('lodash');
+const PayConst = require('../../../app/const/deal_pay.js');
 
 describe('test/app/service/tender.test.js', () => {
 
@@ -33,10 +39,24 @@ describe('test/app/service/tender.test.js', () => {
         assert(loginResult);
         mockData.session = ctx.session;
         // 移除以往测试脏数据
-        yield ctx.service.tender.db.delete(ctx.service.tender.tableName, {
+        const tender = yield ctx.service.tender.getDataByCondition({
+            name: renderbody.name,
             project_id: ctx.session.sessionProject.id,
-            name: addData.name,
         });
+        if (tender) {
+            const tRst = yield ctx.service.tender.db.delete(ctx.service.tender.tableName, {
+                id: tender.id
+            });
+            assert(tRst.affectedRows === 1);
+            const lRst = yield ctx.service.ledger.db.delete(ctx.service.ledger.tableName, {
+                tender_id: tender.id
+            });
+            assert(lRst.affectedRows === 53);
+            const pRst = yield ctx.service.pay.db.delete(ctx.service.pay.tableName, {
+                tid: tender.id
+            });
+            assert(pRst.affectedRows === 5);
+        }
     });
 
     it('test addTender', function* () {
@@ -49,6 +69,18 @@ describe('test/app/service/tender.test.js', () => {
         });
         assert(data);
         testTenderId = data.id;
+
+        const ledger = yield ctx.service.ledger.getDataByTenderId(testTenderId, -1);
+        assert(ledger.length === 53);
+
+        const payNodes = yield ctx.service.pay.getAllDataByCondition({where: {tid: testTenderId}});
+        assert(payNodes.length === 5);
+        let payNode = _.find(payNodes, {ptype: PayConst.payType.wc});
+        assert(payNode.name === '本期完成计量');
+        payNode = _.find(payNodes, {ptype: PayConst.payType.yf});
+        assert(payNode.name === '本期应付');
+        payNode = _.find(payNodes, {ptype: PayConst.payType.sf});
+        assert(payNode.name === '本期实付');
     });
 
     it('test getinfo', function* () {
@@ -60,10 +92,6 @@ describe('test/app/service/tender.test.js', () => {
 
     it('save test',function* () {
         const ctx = app.mockContext(mockData);
-        const renderbody = {
-            name : '测试标段5',
-            type: 1,
-        };
         const result = yield ctx.service.tender.save(renderbody, testTenderId);
         assert(result);
         const data = yield ctx.service.tender.getTender(testTenderId);