#######################################################################
#
# pgAdmin 4 - PostgreSQL Tools
#
# Copyright (C) 2013 - 2018, The pgAdmin Development Team
# This software is released under the PostgreSQL Licence
#
##########################################################################
from pgadmin.utils.compile_template_name import compile_template_name
from pgadmin.utils.route import BaseTestGenerator


class StartRunningQueryTest(BaseTestGenerator):
    """
    Check that the apply_explain_plan_weapper_if_needed method works as
    intended
    """
    scenarios = [
        (
            'When server is Postgres and version is 10, it returns the path '
            'to the postgres template',
            dict(
                server_type='pg',
                version=100000,
                expected_return_value='some/prefix/#100000#/some_file.sql'
            )
        ),
        (
            'When server is GreenPlum and version is 5, it returns the path '
            'to the GreenPlum template',
            dict(
                server_type='gpdb',
                version=80323,
                expected_return_value='some/prefix/#gpdb#80323#/some_file.sql'
            )
        ),
    ]

    def runTest(self):
        result = compile_template_name(
            'some/prefix', 'some_file.sql', self.server_type, self.version
        )
        self.assertEquals(result, self.expected_return_value)
