Calculation/test/app/service/ledger.test.js

/**
 * 标段--台账 模型 单元测试
 *
 * @author Mai
 * @date 2017/12/1
 * @version
 */
'use strict';

/* const testNodeData = [
    { ledger_id: 1, ledger_pid: -1, order: 1, level: 1, full_path: '1', code: '1' },
    { ledger_id: 2, ledger_pid: 1, order: 1, level: 2, full_path: '1.2', code: '1-1' },
    { ledger_id: 6, ledger_pid: 2, order: 1, level: 3, full_path: '1.2.6', code: '1-1-1' },
    { ledger_id: 7, ledger_pid: 6, order: 1, level: 4, full_path: '1.2.6.7', code: '202-1' },
    { ledger_id: 10, ledger_pid: 7, order: 2, level: 5, full_path: '1.2.6.7.10', code: '202-1-a' },
    { ledger_id: 9, ledger_pid: 7, order: 1, level: 5, full_path: '1.2.6.7.9', code: '202-1-b' },
    { ledger_id: 8, ledger_pid: 6, order: 2, level: 4, full_path: '1.2.6.8', code: '202-2' },
    { ledger_id: 11, ledger_pid: 8, order: 1, level: 5, full_path: '1.2.6.8.11', code: '202-2-c' },
    { ledger_id: 12, ledger_pid: 8, order: 2, level: 5, full_path: '1.2.6.8.12', code: '202-2-e' },
    { ledger_id: 13, ledger_pid: 2, order: 2, level: 3, full_path: '1.2.13', code: '1-1-2' },
    { ledger_id: 14, ledger_pid: 2, order: 3, level: 3, full_path: '1.2.14', code: '1-1-3' },
    { ledger_id: 3, ledger_pid: 1, order: 2, level: 2, full_path: '1.3', code: '1-2' },
    { ledger_id: 4, ledger_pid: 1, order: 3, level: 2, full_path: '1.4', code: '1-3' },
    { ledger_id: 5, ledger_pid: 1, order: 4, level: 2, full_path: '1.5', code: '1-4' },
];*/
const testNodeData = [
    { ledger_id: 1, ledger_pid: -1, order: 1, level: 1, full_path: '1', code: '1', is_leaf: false },
    { ledger_id: 2, ledger_pid: 1, order: 1, level: 2, full_path: '1.2', code: '1-1', is_leaf: false },
    { ledger_id: 6, ledger_pid: 2, order: 1, level: 3, full_path: '1.2.6', code: '1-1-1', is_leaf: false },
    { ledger_id: 7, ledger_pid: 6, order: 1, level: 4, full_path: '1.2.6.7', code: '202-1', is_leaf: false },
    { ledger_id: 10, ledger_pid: 7, order: 2, level: 5, full_path: '1.2.6.7.10', code: '202-1-a', is_leaf: true },
    { ledger_id: 9, ledger_pid: 7, order: 1, level: 5, full_path: '1.2.6.7.9', code: '202-1-b', is_leaf: true },
    { ledger_id: 8, ledger_pid: 6, order: 2, level: 4, full_path: '1.2.6.8', code: '202-2', is_leaf: false },
    { ledger_id: 11, ledger_pid: 8, order: 1, level: 5, full_path: '1.2.6.8.11', code: '202-2-c', is_leaf: true },
    { ledger_id: 12, ledger_pid: 8, order: 2, level: 5, full_path: '1.2.6.8.12', code: '202-2-e', is_leaf: true },
    { ledger_id: 13, ledger_pid: 2, order: 2, level: 3, full_path: '1.2.13', code: '1-1-2', is_leaf: true },
    { ledger_id: 14, ledger_pid: 2, order: 3, level: 3, full_path: '1.2.14', code: '1-1-3', is_leaf: true },
    { ledger_id: 3, ledger_pid: 1, order: 2, level: 2, full_path: '1.3', code: '1-2', is_leaf: false },
    { ledger_id: 15, ledger_pid: 3, order: 1, level: 3, full_path: '1.3.15', code: '1-2-1', is_leaf: true },
    { ledger_id: 4, ledger_pid: 1, order: 3, level: 2, full_path: '1.4', code: '1-3', is_leaf: false },
    { ledger_id: 16, ledger_pid: 4, order: 1, level: 3, full_path: '1.4.16', code: '1-3-1', is_leaf: true },
    { ledger_id: 5, ledger_pid: 1, order: 4, level: 2, full_path: '1.5', code: '1-4', is_leaf: true },
];
const testTenderId = 3;

const { app, assert } = require('egg-mock/bootstrap');
const findById = function(nodes, Id) {
    const filters = nodes.filter(function(x) {
        return x.ledger_id === Id;
    });
    return filters.length > 0 ? filters[0] : undefined;
};

describe('test/app/service/ledger.test.js', () => {
    // 准备测试数据
    it('clear history test data', function* () {
        const ctx = app.mockContext();
        const result = yield ctx.service.ledger.db.delete(ctx.service.ledger.tableName, { tender_id: testTenderId });
        assert(result.affectedRows >= 0);
    });
    it('add test data', function* () {
        const ctx = app.mockContext();
        for (const data of testNodeData) {
            data.tender_id = testTenderId;
        }
        const result = yield ctx.service.ledger.db.insert(ctx.service.ledger.tableName, testNodeData);
        assert(result.affectedRows === testNodeData.length);
        ctx.service.ledger.cache.set('tender_node_maxId:' + testTenderId, 16, 'EX', ctx.app.config.cacheTime);
    });
    /* 期望运行结果:
        1
        ├── 1-1
        │   ├── 1-1-1
        │   │   ├── 202-1
        │   │   │   ├── 202-1-a
        │   │   │   └── 202-1-b
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-1-3
        │   ├── 1-1-2
        │   └── 1-1-3
        ├── 1-2
        │   └── 1-2-1
        ├── 1-3
        │   └── 1-3-1
        └── 1-4
     */

    // 测试R类方法
    it('test getDataByTenderId', function* () {
        const ctx = app.mockContext();

        // 查询前4层节点
        const result1 = yield ctx.service.ledger.getDataByTenderId(testTenderId);
        assert(result1.length === 12);
        // 查询前3层节点
        const result2 = yield ctx.service.ledger.getDataByTenderId(testTenderId, 3);
        assert(result2.length === 10);
    });
    it('test getDataByNodeId', function* () {
        const ctx = app.mockContext();

        // 查询节点202-1
        const node = yield ctx.service.ledger.getDataByNodeId(testTenderId, 7);
        assert(node);
        assert(node.code === '202-1');
        assert(node.full_path === '1.2.6.7');
    });
    it('test getDataByNodeIds', function* () {
        const ctx = app.mockContext();

        // 查询节点202-1-a与201-1-b
        const result = yield ctx.service.ledger.getDataByNodeIds(testTenderId, [10, 9]);
        assert(result.length === 2);

        let node = findById(result, 10);
        assert(node.code === '202-1-a');

        node = findById(result, 9);
        assert(node.full_path === '1.2.6.7.9');
    });
    it('test getLastChildData', function* () {
        const ctx = app.mockContext();

        // 查询节点202-1最后一个子节点
        const result = yield ctx.service.ledger.getLastChildData(testTenderId, 7);
        assert(result.ledger_id === 10);
        assert(result.full_path === '1.2.6.7.10');
    });
    it('test getDataByParentAndOrder', function* () {
        const ctx = app.mockContext();

        // 查询节点202-1 第1子节点
        const result1 = yield ctx.service.ledger.getDataByParentAndOrder(testTenderId, 8, 1);
        assert(result1.ledger_id === 11);
        assert(result1.full_path === '1.2.6.8.11');

        // 查询节点1-1 第2/3子节点
        const result2 = yield ctx.service.ledger.getDataByParentAndOrder(testTenderId, 2, [2, 3]);
        assert(result2.length === 2);

        let node = findById(result2, 13);
        assert(node);
        assert(node.code === '1-1-2');

        node = findById(result2, 14);
        assert(node);
        assert(node.code === '1-1-3');
    });
    it('test getChildrenByParentId', function* () {
        const ctx = app.mockContext();

        // 查询节点202-1最后一个子节点
        const result = yield ctx.service.ledger.getChildrenByParentId(testTenderId, 8);
        assert(result.length === 2);

        let node = findById(result, 11);
        assert(node);
        assert(node.code === '202-2-c');

        node = findById(result, 12);
        assert(node);
        assert(node.full_path === '1.2.6.8.12');
    });
    it('test getDataByFullPath', function* () {
        const ctx = app.mockContext();

        // 查询节点202-1最后一个子节点
        const result = yield ctx.service.ledger.getDataByFullPath(testTenderId, '1.2.6.8%');
        assert(result.length === 3);

        let node = findById(result, 8);
        assert(node);
        assert(node.code === '202-2');

        node = findById(result, 11);
        assert(node);
        assert(node.code === '202-2-c');

        node = findById(result, 12);
        assert(node);
        assert(node.full_path === '1.2.6.8.12');
    });
    it('test getNextsData', function* () {
        const ctx = app.mockContext();

        // 查询节点202-1最后一个子节点
        const result = yield ctx.service.ledger.getNextsData(testTenderId, 2, 1);
        assert(result.length === 2);

        let node = findById(result, 13);
        assert(node);
        assert(node.code === '1-1-2');

        node = findById(result, 14);
        assert(node);
        assert(node.code === '1-1-3');
    });

    // 测试CUD类方法
    it('test addNode', function* () {
        const ctx = app.mockContext();
        // 选中1-1-1，插入节点
        const resultData = yield ctx.service.ledger.addNode(testTenderId, 6);
        assert(resultData.create.length === 1);
        assert(resultData.update.length === 2);
        assert(resultData.create[0].is_leaf === 1);
        assert(resultData.create[0].ledger_id === 17);
    });
    /* 期望运行结果:
        1
        ├── 1-1
        │   ├── 1-1-1
        │   │   ├── 202-1
        │   │   │   ├── 202-1-a
        │   │   │   └── 202-1-b
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-1-3
        │   ├── new
        │   ├── 1-1-2
        │   └── 1-1-3
        ├── 1-2
        │   └── 1-2-1
        ├── 1-3
        │   └── 1-3-1
        └── 1-4
     */
    it('test deleteNode', function* () {
        const ctx = app.mockContext();
        // 选中202-1，删除节点
        const resultData = yield ctx.service.ledger.deleteNode(testTenderId, 7);
        assert(resultData.delete.length === 3);
        assert(resultData.update.length === 1);
    });
    /* 期望运行结果:
        1
        ├── 1-1
        │   ├── 1-1-1
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-2-e
        │   ├── new
        │   ├── 1-1-2
        │   └── 1-1-3
        ├── 1-2
        │   └── 1-2-1
        ├── 1-3
        │   └── 1-3-1
        └── 1-4
     */
    it('test upMoveNode', function* () {
        const ctx = app.mockContext();
        // 选中202-2-e上移
        const resultData = yield ctx.service.ledger.upMoveNode(testTenderId, 12);
        resultData.update.sort(function(x, y) {
            return x.order - y.order;
        });
        assert(resultData.update.length === 2);
        assert(resultData.update[0].code === '202-2-e');
    });
    /* 期望运行结果:
        1
        ├── 1-1
        │   ├── 1-1-1
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-2-e
        │   ├── new
        │   ├── 1-1-2
        │   └── 1-1-3
        ├── 1-2
        │   └── 1-2-1
        ├── 1-3
        │   └── 1-3-1
        └── 1-4
     */
    it('test downMoveNode', function* () {
        const ctx = app.mockContext();
        // 选中202-2-e下移
        const resultData = yield ctx.service.ledger.downMoveNode(testTenderId, 12);
        resultData.update.sort(function(x, y) {
            return x.order - y.order;
        });
        assert(resultData.update.length === 2);
        assert(resultData.update[0].code === '202-2-c');
    });
    /* 期望运行结果:
        1
        ├── 1-1
        │   ├── 1-1-1
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-2-e
        │   ├── new
        │   ├── 1-1-2
        │   └── 1-1-3
        ├── 1-2
        │   └── 1-2-1
        ├── 1-3
        │   └── 1-3-1
        └── 1-4
     */
    it('test upLevelNode', function* () {
        const ctx = app.mockContext();
        // 选中 1-1-2 升级
        const resultData = yield ctx.service.ledger.upLevelNode(testTenderId, 13);
        assert(resultData);
        assert(resultData.update.length === 5);

        let node = findById(resultData.update, 13);
        assert(node.full_path === '1.13');
        assert(node.ledger_pid === 1);
        assert(!node.is_leaf);

        node = findById(resultData.update, 14);
        assert(node.ledger_pid === 13);
        assert(node.full_path === '1.13.14');

        node = findById(resultData.update, 3);
        assert(node.order === 3);

        node = findById(resultData.update, 4);
        assert(node.order === 4);

        node = findById(resultData.update, 5);
        assert(node.order === 5);
    });
    /* 期望运行结果:
        1
        ├── 1-1
        │   ├── 1-1-1
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-2-e
        │   └── new
        ├── 1-1-2
        │   └── 1-1-3
        ├── 1-2
        │   └── 1-2-1
        ├── 1-3
        │   └── 1-3-1
        └── 1-4
     */
    it('test downLevelNode', function* () {
        const ctx = app.mockContext();
        // 选中1-3 降级
        const resultData = yield ctx.service.ledger.downLevelNode(testTenderId, 4);
        assert(resultData.update.length === 3);

        let node = findById(resultData.update, 4);
        assert(node.full_path === '1.3.4');
        assert(node.ledger_pid === 3);
        assert(node.level === 3);
        assert(node.order === 2);

        node = findById(resultData.update, 16);
        assert(node.level === 4);
        assert(node.full_path === '1.3.4.16');

        node = findById(resultData.update, 5);
        assert(node.order === 4);
    });
    /* 期望运行结果:
        1
        ├── 1-1
        │   ├── 1-1-1
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-2-e
        │   └── new
        ├── 1-1-2
        │   └── 1-1-3
        ├── 1-2
        │   ├── 1-2-1
        │   └── 1-3
        │       └── 1-3-1
        └── 1-4
     */
    it('test pasteBlock', function* () {
        const ctx = app.mockContext();
        // 选中1-2-1, 粘贴1-1-1和new
        const resultData = yield ctx.service.ledger.pasteBlock(testTenderId, 15, [6, 17]);

        assert(resultData.create.length === 2);
        assert(resultData.create[0].order = 2);
        assert(resultData.create[0].level = 3);
        assert(resultData.create[0].full_path = '1.3.18');

        assert(resultData.create[1].order = 3);
        assert(resultData.create[1].level = 3);
        assert(resultData.create[1].full_path = '1.3.22');

        assert(resultData.update.length === 1);
        assert(resultData.update[0].order = 3);
    });
    /* 期望运行结果:
        1
        ├── 1-1
        │   ├── 1-1-1
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-2-e
        │   └── new
        ├── 1-1-2
        │   └── 1-1-3
        ├── 1-2
        │   ├── 1-2-1
        │   ├── 1-1-1
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-2-e
        │   ├── new
        │   └── 1-3
        │       └── 1-3-1
        └── 1-4
     */
    it('test updateInfo', function* () {
        const ctx = app.mockContext();

        // 修改new(id=17)的code 为 1-1-4
        const node = yield ctx.service.ledger.getDataByNodeId(testTenderId, 17);
        assert(node);

        const resultData = yield ctx.service.ledger.updateInfo(testTenderId, {
            id: node.id,
            tender_id: node.tender_id,
            ledger_id: node.ledger_id,
            code: '1-1-4',
        });
        assert(resultData.code === '1-1-4');
    });
    /* 期望运行结果:
        1
        ├── 1-1
        │   ├── 1-1-1
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-2-e
        │   └── 1-1-3
        ├── 1-1-2
        │   └── 1-1-3
        ├── 1-2
        │   ├── 1-2-1
        │   ├── 1-1-1
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-2-e
        │   ├── new
        │   └── 1-3
        │       └── 1-3-1
        └── 1-4
     */
    it('test updateInfos', function* () {
        const ctx = app.mockContext();

        // 修改new(id=17)的code 为 1-1-4
        const node1 = yield ctx.service.ledger.getDataByNodeId(testTenderId, 18);
        assert(node1);
        const node2 = yield ctx.service.ledger.getDataByNodeId(testTenderId, 22);
        assert(node2);

        const resultData = yield ctx.service.ledger.updateInfos(testTenderId, [{
            id: node1.id,
            tender_id: node1.tender_id,
            ledger_id: node1.ledger_id,
            code: '1-2-2',
        }, {
            id: node2.id,
            tender_id: node2.tender_id,
            ledger_id: node2.ledger_id,
            code: '1-2-3',
        }]);
        assert(resultData.length === 2);
        assert(resultData[0].code === '1-2-2');
        assert(resultData[1].code === '1-2-3');
    });
    /* 期望运行结果:
        1
        ├── 1-1
        │   ├── 1-1-1
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-2-e
        │   └── 1-1-3
        ├── 1-1-2
        │   └── 1-1-3
        ├── 1-2
        │   ├── 1-2-1
        │   ├── 1-2-2
        │   │   └── 202-2
        │   │       ├── 202-2-c
        │   │       └── 202-2-e
        │   ├── 1-2-3
        │   └── 1-3
        │       └── 1-3-1
        └── 1-4
     */
});