@biothings-explorer/query_graph_handler/__test__/unittest/QueryEdge.test.js

A nodejs module to query TRAPI Query Graph

const qEdge = require("../../src/query_edge");


describe("Test QEdge class", () => {
    describe("Test getPredicate function", () => {

        test("Non reversed edge should return predicates itself", () => {
            const edge = new qEdge('e01', {
                predicates: 'biolink:treats',
                object: {
                    getCurie() {
                        return undefined;
                    }
                },
                subject: {
                    getCurie() {
                        return 'uye'
                    }
                }
            })
            const res = edge.getPredicate();
            expect(res).toContain("treats");
        })

        test("Undefined predicate should return itself", () => {
            const edge = new qEdge('e01', {
            })
            const res = edge.getPredicate();
            expect(res).toBeUndefined;
        })

        test("An array of non-undefined predicates should return itself", () => {
            const edge = new qEdge('e01', {
                predicates: ['biolink:treats', 'biolink:targets'],
                object: {
                    getCurie() {
                        return undefined
                    }
                },
                subject: {
                    getCurie() {
                        return 'yes';
                    }
                }
            })
            const res = edge.getPredicate();
            expect(res).toContain("treats");
            expect(res).toContain("targets");
        })

        test("An array of non-undefined predicates with reverse edge should exclude return value if undefined", () => {

            const edge = new qEdge('e01', {
                predicates: ['biolink:treats', 'biolink:targets'],
                object: {
                    getCurie() {
                        return 'yes'
                    }
                },
                subject: {
                    getCurie() {
                        return undefined;
                    }
                }
            })
            const res = edge.getPredicate();
            expect(res).toContain("treated_by")
        })

        test("An array of non-undefined predicates with reverse edge should return reversed predicates if not undefined", () => {
            const edge = new qEdge('e01', {
                predicates: ['biolink:treats', 'biolink:targets'],
                object: {
                    getCurie() {
                        return 'yes'
                    }
                },
                subject: {
                    getCurie() {
                        return undefined;
                    }
                }
            })
            const res = edge.getPredicate();
            expect(res).toContain("treated_by");
        })
    })

    describe("Test getOutputNode function", () => {
        test("reversed edge should return the subject", () => {

            const edge = new qEdge('e01', {
                predicates: ['biolink:treats', 'biolink:targets'],
                object: {
                    getCurie() {
                        return 'yes'
                    },
                    id() {
                        return 1
                    }
                },
                subject: {
                    getCurie() {
                        return undefined;
                    },
                    id() {
                        return 2
                    }
                }
            })
            const res = edge.getOutputNode();
            expect(res.id()).toEqual(2);
        })

        test("non reversed edge should return the object", () => {
            const edge = new qEdge('e01', {
                predicates: ['biolink:treats', 'biolink:targets'],
                object: {
                    getCurie() {
                        return undefined
                    },
                    id() {
                        return 1
                    }
                },
                subject: {
                    getCurie() {
                        return 'aa';
                    },
                    id() {
                        return 2
                    }
                }
            })
            const res = edge.getOutputNode();
            expect(res.id()).toEqual(1);
        })
    })
})