gremlin-python/src/main/python/tests/process/test_strategies.py

#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
# 
# http://www.apache.org/licenses/LICENSE-2.0
# 
# 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.
#

__author__ = 'Marko A. Rodriguez (http://markorodriguez.com)'

from gremlin_python.process.anonymous_traversal import traversal
from gremlin_python.process.strategies import *
from gremlin_python.process.graph_traversal import __


class TestTraversalStrategies(object):
    def test_singletons(self):
        g = traversal().with_(None)
        gremlin_script = g.with_strategies(ReadOnlyStrategy()).gremlin_lang
        gremlin_instr = gremlin_script.gremlin
        assert "withStrategies" in str(gremlin_script)
        assert "ReadOnlyStrategy" in str(gremlin_script)
        assert "withStrategies(ReadOnlyStrategy)" == gremlin_instr[1]
        assert 0 == len(g.traversal_strategies.traversal_strategies)  # these strategies are proxies
        ##
        g = g.with_strategies(ReadOnlyStrategy(), IncidentToAdjacentStrategy())
        gremlin_script = g.gremlin_lang
        gremlin_instr = gremlin_script.gremlin
        assert "withStrategies" in str(gremlin_script)
        assert "ReadOnlyStrategy" in str(gremlin_script)
        assert "withStrategies(ReadOnlyStrategy,IncidentToAdjacentStrategy)" == gremlin_instr[1]
        ##
        gremlin_script = g.V().gremlin_lang
        gremlin_instr = gremlin_script.gremlin
        assert "withStrategies" in str(gremlin_script)
        assert "ReadOnlyStrategy" in str(gremlin_script)
        assert "IncidentToAdjacentStrategy" in str(gremlin_script)
        assert "V()" in str(gremlin_script)
        assert "withStrategies(ReadOnlyStrategy,IncidentToAdjacentStrategy)" == gremlin_instr[1]
        assert 'V' == gremlin_instr[3]
        ##
        gremlin_script = g.without_strategies(ReadOnlyStrategy()).V().gremlin_lang
        gremlin_instr = gremlin_script.gremlin
        gremlin_params = gremlin_script.parameters
        assert "withStrategies" in str(gremlin_script)
        assert "ReadOnlyStrategy" in str(gremlin_script)
        assert "IncidentToAdjacentStrategy" in str(gremlin_script)
        assert "withoutStrategies" in str(gremlin_script)
        assert "V()" in str(gremlin_script)
        assert "withStrategies(ReadOnlyStrategy,IncidentToAdjacentStrategy)" == gremlin_instr[1]
        assert ReadOnlyStrategy() == gremlin_params["_0"]
        ##
        gremlin_script = g.without_strategies(ReadOnlyStrategy(), LazyBarrierStrategy()).V().gremlin_lang
        gremlin_params = gremlin_script.parameters
        assert "withStrategies" in str(gremlin_script)
        assert "ReadOnlyStrategy" in str(gremlin_script)
        assert "IncidentToAdjacentStrategy" in str(gremlin_script)
        assert "withoutStrategies" in str(gremlin_script)
        assert "V()" in str(gremlin_script)
        assert ReadOnlyStrategy() == gremlin_params["_1"]
        assert LazyBarrierStrategy() == gremlin_params["_2"]
        ###
        g = traversal().with_(None)
        gremlin_script = g.with_("x", "test").with_("y").gremlin_lang
        options = gremlin_script.options_strategies
        assert "test" == options[0].configuration["x"]
        assert options[1].configuration["y"]

    def test_configurable(self):
        g = traversal().with_(None)
        gremlin_script = g.with_strategies(MatchAlgorithmStrategy("greedy")).gremlin_lang
        assert "withStrategies" in str(gremlin_script)
        assert "MatchAlgorithmStrategy" in str(gremlin_script)
        assert "greedy" in str(gremlin_script)
        ###
        gremlin_script = g.with_strategies(SubgraphStrategy(vertices=__.has("name", "marko"))).gremlin_lang
        assert "withStrategies" in str(gremlin_script)
        assert "SubgraphStrategy" in str(gremlin_script)
        assert "__.has('name','marko')" in str(gremlin_script)
        ###
        gremlin_script = g.with_strategies(OptionsStrategy(x="test", y=True)).gremlin_lang
        options = gremlin_script.options_strategies
        assert "test" == options[0].configuration["x"]
        assert options[0].configuration["y"]

    def test_custom_strategies(self):
        g = traversal().with_(None)

        gremlin_script = g.with_strategies(TraversalStrategy(strategy_name='CustomSingletonStrategy')).gremlin_lang
        assert "withStrategies(CustomSingletonStrategy)" in str(gremlin_script)

        gremlin_script = g.with_strategies(TraversalStrategy(
            strategy_name='CustomConfigurableStrategy',
            stringKey='string value',
            intKey=5,
            booleanKey=True
        )).gremlin_lang
        assert "withStrategies(new CustomConfigurableStrategy(stringKey:'string value',intKey:5,booleanKey:true))" in str(gremlin_script)