wisemapping-open-source/mindplot/src/main/javascript/nlayout/TestSuite.js

/*
 *    Copyright [2011] [wisemapping]
 *
 *   Licensed under WiseMapping Public License, Version 1.0 (the "License").
 *   It is basically the Apache License, Version 2.0 (the "License") plus the
 *   "powered by wisemapping" text requirement on every single page;
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the license at
 *
 *       http://www.wisemapping.org/license
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 */
mindplot.nlayout.TestSuite = new Class({
    Extends: mindplot.nlayout.ChildrenSorterStrategy,

    initialize:function() {
        this.testAligned();
        this.testSymmetry();
        this.testBalanced();
        this.testEvents();
        this.testEventsComplex();
        this.testDisconnect();
        this.testReconnect();
        this.testRemoveNode();
        this.testFreePosition();
    },

    testAligned: function() {

        var position = {x:0,y:0};
        var manager = new mindplot.nlayout.LayoutManager(0, mindplot.nlayout.TestSuite.ROOT_NODE_SIZE);

        manager.addNode(1, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(2, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(3, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(4, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.connectNode(0, 1, 0);
        manager.connectNode(1, 2, 0);
        manager.connectNode(2, 3, 0);
        manager.connectNode(3, 4, 0);

        manager.layout();
        manager.plot("testAligned", {width:1200,height:200});

        // All nodes should be vertically aligned
        $assert(manager.find(0).getPosition().y == manager.find(1).getPosition().y, "Nodes are not aligned");
        $assert(manager.find(0).getPosition().y == manager.find(2).getPosition().y, "Nodes are not aligned");
        $assert(manager.find(0).getPosition().y == manager.find(3).getPosition().y, "Nodes are not aligned");
        $assert(manager.find(0).getPosition().y == manager.find(4).getPosition().y, "Nodes are not aligned");
    },

    testSymmetry: function() {
        var position = {x:0,y:0};
        var manager = new mindplot.nlayout.LayoutManager(0, mindplot.nlayout.TestSuite.ROOT_NODE_SIZE);

        manager.addNode(1, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(2, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(3, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(4, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(5, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(6, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(7, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(8, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(9, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(10, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(11, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(12, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(13, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(14, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.connectNode(0, 14, 0);
        manager.connectNode(14, 13, 0);
        manager.connectNode(13, 1, 0);
        manager.connectNode(13, 2, 1);
        manager.connectNode(13, 3, 2);
        manager.connectNode(13, 4, 3);
        manager.connectNode(13, 5, 4);
        manager.connectNode(1, 6, 0);
        manager.connectNode(1, 7, 1);
        manager.connectNode(7, 8, 0);
        manager.connectNode(8, 9, 0);
        manager.connectNode(5, 10, 0);
        manager.connectNode(6, 11, 0);
        manager.connectNode(6, 12, 1);

        manager.layout();
        manager.plot("testSymmetry",{width:1200, height:400});

        //TODO(gb): make asserts
    },

    testBalanced: function() {
        var position = {x:0, y:0};
        var plotsize = {width:800, height:400};
        var manager = new mindplot.nlayout.LayoutManager(0, mindplot.nlayout.TestSuite.ROOT_NODE_SIZE);

        manager.addNode(1, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.connectNode(0, 1, 0);
        manager.layout();
        manager.plot("testBalanced1", plotsize);

        manager.addNode(2, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.connectNode(0, 2, 1);
        manager.layout();
        manager.plot("testBalanced2", plotsize);

        manager.addNode(3, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.connectNode(0, 3, 2);
        manager.layout();
        manager.plot("testBalanced3", plotsize);

        manager.addNode(4, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.connectNode(0, 4, 3);
        manager.layout();
        manager.plot("testBalanced4", plotsize);

        manager.addNode(5, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.connectNode(0, 5, 4);
        manager.layout();
        manager.plot("testBalanced5", plotsize);

        manager.addNode(6, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.connectNode(0, 6, 5);
        manager.layout();
        manager.plot("testBalanced6", plotsize);

        manager.addNode(7, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(8, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(9, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.connectNode(3, 7, 0)
        manager.connectNode(7, 8, 0)
        manager.connectNode(7, 9, 1);
        manager.layout();
        manager.plot("testBalanced7", plotsize);

        manager.addNode(10, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(11, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(12, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.connectNode(6, 10, 0)
        manager.connectNode(10, 11, 0)
        manager.connectNode(10, 12, 1);
        manager.layout();
        manager.plot("testBalanced8", plotsize);

        //TODO(gb): make asserts
    },

    testEvents: function() {
        console.log("testEvents:");        //TODO(gb): Remove trace!!!
        var position = {x:0,y:0};
        var manager = new mindplot.nlayout.LayoutManager(0, mindplot.nlayout.TestSuite.ROOT_NODE_SIZE);

        // Add 3 nodes...
        manager.addNode(1, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(2, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(3, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(4, mindplot.nlayout.TestSuite.NODE_SIZE, position);

        // Now connect one with two....
        manager.connectNode(0, 1, 0);
        manager.connectNode(0, 2, 0);
        manager.connectNode(1, 3, 0);

        // Basic layout repositioning ...
        console.log("\t--- Updated tree ---");
        var events = [];
        manager.addEvent('change', function(event) {
            console.log("\tUpdated nodes: {id:" + event.getId() + ", order: " + event.getOrder() + ",position: {" + event.getPosition().x + "," + event.getPosition().y + "}");
            events.push(event);
        });
        manager.layout(true);
        manager.plot("testEvents1", {width:800, height:400});

        // Ok, if a new node is added, this an event should be fired  ...
        console.log("\t---- Layout without changes should not affect the tree  ---");
        events.empty();
        manager.layout(true);
        manager.plot("testEvents2", {width:800, height:400});

        $assert(events.length == 0, "Unnecessary tree updated.");

        //TODO(gb): make asserts
        console.log("\n");        //TODO(gb): Remove trace!!!
    },

    testEventsComplex: function() {
        console.log("testEventsComplex:");        //TODO(gb): Remove trace!!!
        var position = {x:0,y:0};
        var manager = new mindplot.nlayout.LayoutManager(0, mindplot.nlayout.TestSuite.ROOT_NODE_SIZE);

        // Add 3 nodes...
        manager.addNode(1, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(2, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(3, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(4, mindplot.nlayout.TestSuite.NODE_SIZE, position);

        // Now connect one with two....
        manager.connectNode(0, 1, 0);
        manager.connectNode(1, 2, 0);
        manager.connectNode(1, 3, 1);

        var events = [];
        manager.addEvent('change', function(event) {
            console.log("\tUpdated nodes: {id:" + event.getId() + ", order: " + event.getOrder() + ",position: {" + event.getPosition().x + "," + event.getPosition().y + "}");
            events.push(event);
        });

        // Reposition ...
        manager.layout(true);
        manager.plot("testEventsComplex1", {width:800, height:400});

        // Add a new node and connect. Only children nodes should be affected.
        console.log("\t---- Connect a new node  ---");

        events.empty();
        manager.connectNode(1, 4, 2);
        manager.layout(true);
        manager.plot("testEventsComplex2", {width:800, height:400});

        // @todo: This seems no to be ok...
        $assert(events.length == 4, "Only 3 nodes should be repositioned.");

        //TODO(gb): make asserts
        console.log("\n");        //TODO(gb): Remove trace!!!
    },

    testDisconnect: function() {
        console.log("testDisconnect:");        //TODO(gb): Remove trace!!!
        var position = {x:0,y:0};
        var manager = new mindplot.nlayout.LayoutManager(0, mindplot.nlayout.TestSuite.ROOT_NODE_SIZE);

        // Prepare a sample graph ...
        manager.addNode(1, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(2, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(3, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(4, mindplot.nlayout.TestSuite.NODE_SIZE, position);

        manager.connectNode(0, 1, 0);
        manager.connectNode(1, 2, 0);
        manager.connectNode(1, 3, 1);
        manager.connectNode(3, 4, 0);

        var events = [];
        manager.addEvent('change', function(event) {

            var pos = event.getPosition();
            var posStr = pos ? ",position: {" + pos.x + "," + pos.y : "";
            var node = manager.find(event.getId());
            console.log("\tUpdated nodes: {id:" + event.getId() + ", order: " + event.getOrder() + posStr + "}");
            events.push(event);
        });
        manager.layout(true);
        manager.plot("testDisconnect1", {width:800, height:400});

        // Now, disconnect one node ...
        console.log("\t--- Disconnect a single node ---");
        events.empty();
        manager.disconnectNode(2);
        manager.layout(true);
        manager.plot("testDisconnect2", {width:800, height:400});

        $assert(events.some(
            function(event) {
                return event.getId() == 2;
            }), "Event for disconnected node seems not to be propagated");

        // Great, let's disconnect a not with children.
        console.log("\t--- Disconnect a node with children ---");
        manager.disconnectNode(3);
        manager.layout(true);
        manager.plot("testDisconnect3", {width:800, height:400});

        $assert(events.some(
            function(event) {
                return event.getId() == 2;
            }), "Event for disconnected node seems not to be propagated");

        //TODO(gb): make asserts
        console.log("\n");        //TODO(gb): Remove trace!!!
    },

    testReconnect: function() {
        var position = {x:0,y:0};
        var manager = new mindplot.nlayout.LayoutManager(0, mindplot.nlayout.TestSuite.ROOT_NODE_SIZE);

        manager.addNode(1, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(2, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(3, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(4, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(5, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(6, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(7, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(8, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(9, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(10, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(11, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(12, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.connectNode(0, 1, 0);
        manager.connectNode(0, 2, 1);
        manager.connectNode(0, 3, 2);
        manager.connectNode(0, 4, 3);
        manager.connectNode(0, 5, 4);
        manager.connectNode(1, 6, 0);
        manager.connectNode(1, 7, 1);
        manager.connectNode(7, 8, 0);
        manager.connectNode(8, 9, 0);
        manager.connectNode(5, 10, 0);
        manager.connectNode(6, 11, 0);
        manager.connectNode(6, 12, 1);

        manager.layout();
        manager.plot("testReconnect1",{width:1000, height:400});

        // Reconnect node 6 to node 4
        manager.disconnectNode(6);
        manager.connectNode(4,6,0);
        manager.layout();
        manager.plot("testReconnect2",{width:1000, height:400});

        //TODO(gb): make asserts
    },

    testRemoveNode: function() {
        console.log("testRemoveNode:");        //TODO(gb): Remove trace!!!
        var position = {x:0,y:0};
        var manager = new mindplot.nlayout.LayoutManager(0, mindplot.nlayout.TestSuite.ROOT_NODE_SIZE);

        // Prepare a sample graph ...
        manager.addNode(1, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(2, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(3, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(4, mindplot.nlayout.TestSuite.NODE_SIZE, position);

        manager.connectNode(0, 1, 0);
        manager.connectNode(1, 2, 0);
        manager.connectNode(1, 3, 1);
        manager.connectNode(3, 4, 0);

        var events = [];
        manager.addEvent('change', function(event) {
            var pos = event.getPosition();
            var posStr = pos ? ",position: {" + pos.x + "," + event.getPosition().y : "";
            console.log("\tUpdated nodes: {id:" + event.getId() + ", order: " + event.getOrder() + posStr + "}");
            events.push(event);
        });
        manager.layout(true);
        manager.plot("testRemoveNode1", {width:800, height:200});

        // Test removal of a connected node ...
        console.log("\t--- Remove node 3  ---");
        manager.removeNode(3);
        manager.layout(true);
        manager.plot("testRemoveNode2", {width:800, height:200});

        //TODO(gb): make asserts
        console.log("\n");        //TODO(gb): Remove trace!!!
    },

    testFreePosition: function() {
        var position = {x:0,y:0};
        var manager = new mindplot.nlayout.LayoutManager(0, mindplot.nlayout.TestSuite.ROOT_NODE_SIZE);

        // Prepare a sample graph ...
        manager.addNode(1, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(2, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(3, mindplot.nlayout.TestSuite.NODE_SIZE, position);
        manager.addNode(4, mindplot.nlayout.TestSuite.NODE_SIZE, position);

        manager.connectNode(0, 4, 0);
        manager.connectNode(4, 1, 0);
        manager.connectNode(4, 2, 1);
        manager.connectNode(4, 3, 2);

        manager.layout();
        manager.plot("testFreePosition", {width:800, height:400});

        //TODO(gb): make asserts
    }
});

mindplot.nlayout.TestSuite.NODE_SIZE = {width:80, height:30},
mindplot.nlayout.TestSuite.ROOT_NODE_SIZE = {width:120, height:40}